Fix test compilation

itkovian · Oct 9, 2013 · 60b3610 · 60b3610
1 parent 1662a06
commit 60b3610
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/NumberSix/Handlers/Imdb/Tests.hs b/tests/NumberSix/Handlers/Imdb/Tests.hs
@@ -6,7 +6,7 @@ module NumberSix.Handlers.Imdb.Tests
 
 
 --------------------------------------------------------------------------------
-import qualified Data.ByteString         as B
+import qualified Data.Text               as T
 import           Test.Framework          (Test, testGroup)
 import           Test.HUnit              (assert)
 
@@ -22,7 +22,7 @@ tests = testGroup "NumberSix.Handlers.Imdb.Tests"
     [ cases "imdb"
         [ do
             bs <- imdb "once upon a time in america"
-            assert $ "Once Upon a Time in America (1984)" `B.isPrefixOf` bs
-            assert $ "http://www.imdb.com/title/tt0087843/" `B.isSuffixOf` bs
+            assert $ "Once Upon a Time in America (1984)" `T.isPrefixOf` bs
+            assert $ "http://www.imdb.com/title/tt0087843/" `T.isSuffixOf` bs
         ]
     ]
diff --git a/tests/NumberSix/Tests/Util.hs b/tests/NumberSix/Tests/Util.hs
@@ -16,4 +16,4 @@ cases :: String -> [Assertion] -> Test
 cases name assertions = testGroup name
     [ testCase ("case " ++ show i) assertion
     | (i, assertion) <- zip [1 :: Int ..] assertions
-    ] 
+    ]