1 files changed, 105 insertions, 0 deletions

diff --git a/core/test/Parsing/PrimitivesTest.hs b/core/test/Parsing/PrimitivesTest.hs
new file mode 100644
index 0000000..95bc8c8
--- /dev/null
+++ b/core/test/Parsing/PrimitivesTest.hs
@@ -0,0 +1,105 @@
+{-# LANGUAGE OverloadedStrings #-}
+
+module Parsing.PrimitivesTest
+  ( suite
+  ) where
+
+import           Text.Edifact.Parsing
+
+import           Data.Text                      (Text, unpack)
+import           Text.Parsec                    (eof)
+
+import           Test.Framework
+import           Test.Framework.Providers.HUnit (testCase)
+import           Test.HUnit                     ((@?=))
+
+suite :: Test
+suite =
+  testGroup "Primitives"
+    [ test_an
+    , test_an_3
+    , test_an3
+    , test_n
+    ]
+
+test_an :: Test
+test_an =
+  let parser = alphaNumeric
+  in testGroup "an"
+       [ expectSuccess parser "A"  'A'
+       , expectSuccess parser "?:" ':'
+       , expectSuccess parser "?+" '+'
+       , expectSuccess parser "??" '?'
+       , expectSuccess parser "?'" '\''
+       ]
+
+test_an_3 :: Test
+test_an_3 =
+  let parser = alphaNumeric `upTo` 3
+  in testGroup "an..3"
+       [ testGroup "valid"
+         [ expectSuccess parser ""     ""
+         , expectSuccess parser "A"    "A"
+         , expectSuccess parser "AB"   "AB"
+         , expectSuccess parser "ABC"  "ABC"
+         , expectSuccess parser "AB??" "AB?"
+         ]
+       , testGroup "invalid"
+         [ expectFailure parser "ABCD"
+         ]
+       ]
+
+test_an3 :: Test
+test_an3 =
+  let parser = alphaNumeric `exactly` 3
+  in testGroup "an3"
+       [ testGroup "valid"
+         [ expectSuccess parser "ABC"  "ABC"
+         , expectSuccess parser "AB??" "AB?"
+         ]
+       , testGroup "invalid"
+        [ expectFailure parser ""
+        , expectFailure parser "A"
+        , expectFailure parser "AB"
+        , expectFailure parser "ABCD"
+        ]
+      ]
+
+test_n :: Test
+test_n =
+  let parser = numeric
+  in testGroup "n"
+       [ testGroup "valid"
+         [ expectSuccess parser "0" '0'
+         , expectSuccess parser "1" '1'
+         , expectSuccess parser "2" '2'
+         , expectSuccess parser "3" '3'
+         , expectSuccess parser "4" '4'
+         , expectSuccess parser "5" '5'
+         , expectSuccess parser "6" '6'
+         , expectSuccess parser "7" '7'
+         , expectSuccess parser "8" '8'
+         , expectSuccess parser "9" '9'
+         , expectSuccess parser "-" '-'
+         , expectSuccess parser "." '.'
+         ]
+       , testGroup "invalid"
+         [ expectFailure parser "A"
+         , expectFailure parser "?:"
+         , expectFailure parser "?+"
+         , expectFailure parser "??"
+         , expectFailure parser "?'"
+         ]
+       ]
+
+expectSuccess :: (Show a, Eq a) => Parser a -> Text -> a -> Test
+expectSuccess p t = expectParse p t . pure
+
+expectFailure :: (Show a, Eq a) => Parser a -> Text -> Test
+expectFailure p t = expectParse p t Nothing
+
+expectParse :: (Show a, Eq a) => Parser a -> Text -> Maybe a -> Test
+expectParse p t e =
+  let either2Maybe = either (const Nothing) Just
+      title = "\"" <> unpack t <> "\""
+  in testCase title (either2Maybe (parse (p <* eof) t) @?= e)