mirror of
https://github.com/github/semantic.git
synced 2024-12-30 02:14:20 +03:00
116 lines
4.0 KiB
Haskell
116 lines
4.0 KiB
Haskell
{-# LANGUAGE TypeApplications, ScopedTypeVariables #-}
|
|
|
|
module Proto3.Roundtrip (spec) where
|
|
|
|
import SpecHelpers
|
|
|
|
import Data.Blob
|
|
import Data.Span
|
|
import qualified Data.ByteString.Lazy as L
|
|
import Data.Source
|
|
import Data.Functor.Foldable
|
|
import Proto3.Suite
|
|
import qualified Proto3.Wire.Encode as E
|
|
import qualified Data.Syntax.Literal as Literal
|
|
import qualified Data.Syntax.Statement as Statement
|
|
import qualified Data.Syntax.Declaration as Declaration
|
|
import Data.Term (Term)
|
|
import Data.Diff (Diff)
|
|
import Data.Sum
|
|
import Language.Ruby.Assignment (Syntax)
|
|
import qualified Language.Ruby.Assignment as Ruby
|
|
import qualified Language.Python.Syntax as Python
|
|
import Data.Functor.Classes
|
|
import qualified Data.Syntax.Expression as Expression
|
|
|
|
shouldRoundtrip :: (Eq a, Show a, Message a) => a -> Expectation
|
|
shouldRoundtrip a = go a `shouldBe` Right a
|
|
where go = fromByteString . L.toStrict . toLazyByteString
|
|
|
|
shouldRoundtrip1 :: forall f a. (Show (f a), Eq (f a), Show1 f, Eq1 f, Eq a, Show a, Message1 f, Message a) => f a -> Expectation
|
|
shouldRoundtrip1 a = go a `shouldBe` Right a
|
|
where go = fromByteString1 . L.toStrict . toLazyByteString1
|
|
|
|
instance Named1 (Sum '[Literal.Null]) where nameOf1 _ = "NullSyntax"
|
|
|
|
spec :: Spec
|
|
spec = parallel $ do
|
|
describe "blobs" $
|
|
prop "roundtrips" $
|
|
\sp -> shouldRoundtrip @Blob sp
|
|
describe "blob pairs" $
|
|
prop "roundtrips" $
|
|
\sp -> shouldRoundtrip @BlobPair sp
|
|
describe "spans" $
|
|
prop "roundtrips" $
|
|
\x -> shouldRoundtrip @Span x
|
|
describe "pos" $
|
|
prop "roundtrips" $
|
|
\x -> shouldRoundtrip @Pos x
|
|
|
|
describe "nulls" $
|
|
prop "roundtrips" $
|
|
\sp -> shouldRoundtrip1 @Literal.Null @(Term (Sum Syntax) ()) (unListableF sp)
|
|
|
|
describe "text elements" $
|
|
prop "roundtrips" $
|
|
\sp -> shouldRoundtrip1 @Literal.TextElement @(Term (Sum Syntax) ()) (unListableF sp)
|
|
|
|
describe "floats" $
|
|
prop "roundtrips" $
|
|
\sp -> shouldRoundtrip1 @Literal.Float @(Term (Sum Syntax) ()) (unListableF sp)
|
|
|
|
describe "negate" $
|
|
prop "roundtrips" $
|
|
\sp -> shouldRoundtrip1 @Expression.Negate @(Term (Sum '[Literal.Null]) ()) (unListableF sp)
|
|
|
|
describe "booleans" $
|
|
prop "roundtrips" $
|
|
\sp -> shouldRoundtrip1 @Literal.Boolean @(Term (Sum Syntax) ()) (unListableF sp)
|
|
|
|
describe "terms of syntax" $
|
|
prop "roundtrips" $
|
|
\sp -> shouldRoundtrip @(Term (Sum Syntax) ()) (unListableF sp)
|
|
|
|
-- describe "diffs of syntax" $
|
|
-- prop "roundtrips" $
|
|
-- \sp -> do
|
|
-- shouldRoundtrip @(Diff (Sum Syntax) () ()) (unListableF2 sp)
|
|
|
|
describe "arrays" $
|
|
prop "roundtrips" $
|
|
\sp -> shouldRoundtrip1 @Literal.Array @(Term (Sum Syntax) ()) (unListableF sp)
|
|
|
|
describe "key values" $
|
|
prop "roundtrips" $
|
|
\sp -> shouldRoundtrip1 @Literal.KeyValue @(Term (Sum Syntax) ()) (unListableF sp)
|
|
|
|
describe "statements" $
|
|
prop "roundtrips" $
|
|
\sp -> shouldRoundtrip @(Term (Sum Syntax) ()) (unListableF sp)
|
|
|
|
describe "statements1" $
|
|
prop "roundtrips" $
|
|
\sp -> shouldRoundtrip1 @Statement.Statements @(Term (Sum Syntax) ()) (unListableF sp)
|
|
|
|
describe "methods" $
|
|
prop "roundtrips" $
|
|
\sp -> shouldRoundtrip1 @Declaration.Method @(Term (Sum Syntax) ()) (unListableF sp)
|
|
|
|
describe "blobs" $ do
|
|
it "should roundtrip given a Message instance" $ do
|
|
let bl = Blob (fromUTF8 "puts 'hi'") "example.rb" Ruby
|
|
shouldRoundtrip bl
|
|
|
|
describe "languages" $ do
|
|
-- If this test broke, it means you've probably added another 'Language'.
|
|
-- Add it to the list of languages below and everything should be good,
|
|
-- as long as you added it in a way that doesn't break prior Enum encodings.
|
|
it "should match up with Enum declarations" $ do
|
|
let go :: (Primitive f, MessageField f) => [f] -> [L.ByteString]
|
|
go x = E.toLazyByteString . encodePrimitive (fieldNumber 0) <$> x
|
|
let ints = [0..fromEnum (maxBound @Language)]
|
|
let langs = [Unknown, Go, Haskell, Java, JavaScript, JSON,
|
|
JSX, Markdown, Python, Ruby, TypeScript, PHP]
|
|
go ints `shouldBe` go langs
|