test parsing of schema with both imports and entities #6

statebox · Sep 17, 2018 · 5d2b59a · 5d2b59a
1 parent 89944ec
commit 5d2b59a
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/src/Language/Parser/Schema.hs b/src/Language/Parser/Schema.hs
@@ -38,4 +38,13 @@ schemaLiteralSectionParser = do
  maybeImports <- optional $ do
  constant "imports"
  many typesideImportParser
- pure $ SchemaLiteralSection (fromMaybe [] maybeImports) [] [] [] [] []
+ maybeEntities <- optional $ do
+ constant "entities"
+ many identifier
+ pure $ SchemaLiteralSection
+ (fromMaybe [] maybeImports)
+ (fromMaybe [] maybeEntities)
+ []
+ []
+ []
+ []
diff --git a/test/Parser/SchemaSpec.hs b/test/Parser/SchemaSpec.hs
@@ -37,3 +37,12 @@ spec = do
  forAll (listOf typesideImportGen) $
  \typesideImports -> parse schemaLiteralSectionParser "" ("imports " ++ (unwords $ map show typesideImports))
  == Right (SchemaLiteralSection typesideImports [] [] [] [] [])
+ specify "parses correctly a SchemaLiteralSection with entities" $
+ forAll (listOf identifierGen) $
+ \identifiers -> parse schemaLiteralSectionParser "" ("entities " ++ (unwords $ identifiers))
+ == Right (SchemaLiteralSection [] identifiers [] [] [] [])
+ specify "parses correctly a SchemaLiteralSection with every piece" $
+ forAll ((\a b -> (a, b)) <$> listOf typesideImportGen <*> listOf identifierGen) $
+ \(typesideImports, identifiers) ->
+ parse schemaLiteralSectionParser "" ("imports " ++ (unwords $ map show typesideImports) ++ " entities " ++ (unwords $ identifiers))
+ == Right (SchemaLiteralSection typesideImports identifiers [] [] [] [])