2 files changed, 28 insertions, 0 deletions
diff --git a/tests/Hakyll/Core/Route/Tests.hs b/tests/Hakyll/Core/Route/Tests.hs
new file mode 100644
index 0000000..17a4123
--- /dev/null
+++ b/tests/Hakyll/Core/Route/Tests.hs
@@ -0,0 +1,25 @@
+{-# LANGUAGE OverloadedStrings #-}
+module Hakyll.Core.Route.Tests
+    ( tests
+    ) where
+
+import Test.Framework
+import Test.Framework.Providers.HUnit
+import Test.HUnit hiding (Test)
+
+import Hakyll.Core.Route
+
+tests :: [Test]
+tests = zipWith testCase names matchCases 
+  where
+    names = map (\n -> "runRoute [" ++ show n ++ "]") [1 :: Int ..]
+
+-- | Collection of simple cases
+--
+matchCases :: [Assertion]
+matchCases =
+    [ Just "foo.html" @=? runRoute (setExtension "html") "foo"
+    , Just "foo.html" @=? runRoute (setExtension ".html") "foo"
+    , Just "foo.html" @=? runRoute (setExtension "html") "foo.markdown"
+    , Just "foo.html" @=? runRoute (setExtension ".html") "foo.markdown"
+    ]
diff --git a/tests/TestSuite.hs b/tests/TestSuite.hs
index f75001f..68c4f28 100644
--- a/tests/TestSuite.hs
+++ b/tests/TestSuite.hs
@@ -4,6 +4,7 @@ import Test.Framework (defaultMain, testGroup)
 
 import qualified Hakyll.Core.DirectedGraph.Tests
 import qualified Hakyll.Core.Identifier.Tests
+import qualified Hakyll.Core.Route.Tests
 
 main :: IO ()
 main = defaultMain
@@ -11,4 +12,6 @@ main = defaultMain
         Hakyll.Core.DirectedGraph.Tests.tests
     , testGroup "Hakyll.Core.Identifier.Tests"
         Hakyll.Core.Identifier.Tests.tests
+    , testGroup "Hakyll.Core.Route.Tests"
+        Hakyll.Core.Route.Tests.tests
     ]