Add a test for byExtension

1 files changed, 35 insertions, 0 deletions

diff --git a/tests/Hakyll/Core/Compiler/Tests.hs b/tests/Hakyll/Core/Compiler/Tests.hs
new file mode 100644
index 0000000..7ea7190
--- /dev/null
+++ b/tests/Hakyll/Core/Compiler/Tests.hs
@@ -0,0 +1,35 @@
+{-# LANGUAGE OverloadedStrings #-}
+module Hakyll.Core.Compiler.Tests
+    where
+
+import qualified Data.Map as M
+
+import Test.Framework (Test)
+import Test.Framework.Providers.HUnit (testCase)
+import qualified Test.HUnit as H
+
+import Hakyll.Core.Compiler
+import Hakyll.Core.Resource.Provider.Dummy
+import Hakyll.Core.Util.Arrow
+import TestSuite.Util
+
+tests :: [Test]
+tests =
+    [ testCase "byExtension" byExtensionTest
+    ]
+
+byExtensionTest :: H.Assertion
+byExtensionTest = do
+    provider <- dummyResourceProvider $ M.empty
+    txt      <- runCompilerJobTest compiler "foo.txt" provider uni
+    css      <- runCompilerJobTest compiler "bar.css" provider uni
+    html     <- runCompilerJobTest compiler "qux.html" provider uni
+    H.assertEqual "byExtension" "txt" txt
+    H.assertEqual "byExtension" "css" css
+    H.assertEqual "byExtension" "unknown" html
+  where
+    uni      = ["foo.txt", "bar.css", "qux.html"]
+    compiler = byExtension (constA ("unknown" :: String))
+        [ (".txt", constA "txt")
+        , (".css", constA "css")
+        ]