Add some tests

2 files changed, 48 insertions, 1 deletions

diff --git a/tests/StylishHaskell/Parse/Tests.hs b/tests/StylishHaskell/Parse/Tests.hs
new file mode 100644
index 0000000..5b3fbe1
--- /dev/null
+++ b/tests/StylishHaskell/Parse/Tests.hs
@@ -0,0 +1,45 @@
+--------------------------------------------------------------------------------
+module StylishHaskell.Parse.Tests
+    ( tests
+    ) where
+
+
+--------------------------------------------------------------------------------
+import           Test.Framework                 (Test, testGroup)
+import           Test.Framework.Providers.HUnit (testCase)
+import           Test.HUnit                     (assert)
+
+
+--------------------------------------------------------------------------------
+import           StylishHaskell.Parse
+
+
+--------------------------------------------------------------------------------
+tests :: Test
+tests = testGroup "StylishHaskell.Parse"
+    [ testBom
+    , testExtraExtensions
+    ]
+
+
+--------------------------------------------------------------------------------
+testBom :: Test
+testBom = testCase "UTF-8 Byte Order Mark" $ assert $ isRight $
+    parseModule [] Nothing input
+  where
+    input = unlines
+        [ '\xfeff' : "foo :: Int"
+        , "foo = 3"
+        ]
+
+
+--------------------------------------------------------------------------------
+testExtraExtensions :: Test
+testExtraExtensions = testCase "Extra extensions" $ assert $ isRight $
+    parseModule ["TemplateHaskell"] Nothing "$(foo)"
+
+
+--------------------------------------------------------------------------------
+isRight :: Either a b -> Bool
+isRight (Right _) = True
+isRight _         = False
diff --git a/tests/TestSuite.hs b/tests/TestSuite.hs
index f157b8c..e1c8acb 100644
--- a/tests/TestSuite.hs
+++ b/tests/TestSuite.hs
@@ -9,6 +9,7 @@ import           Test.Framework                               (defaultMain)
 
 
 --------------------------------------------------------------------------------
+import qualified StylishHaskell.Parse.Tests
 import qualified StylishHaskell.Step.Imports.Tests
 import qualified StylishHaskell.Step.LanguagePragmas.Tests
 import qualified StylishHaskell.Step.Tabs.Tests
@@ -19,7 +20,8 @@ import qualified StylishHaskell.Step.UnicodeSyntax.Tests
 --------------------------------------------------------------------------------
 main :: IO ()
 main = defaultMain
-    [ StylishHaskell.Step.Imports.Tests.tests
+    [ StylishHaskell.Parse.Tests.tests
+    , StylishHaskell.Step.Imports.Tests.tests
     , StylishHaskell.Step.LanguagePragmas.Tests.tests
     , StylishHaskell.Step.Tabs.Tests.tests
     , StylishHaskell.Step.TrailingWhitespace.Tests.tests