test for Cradle.

2013-03-05 10:22:33 +09:00
parent b7cacee767
commit 89dc02f185
5 changed files with 37 additions and 5 deletions
--- a/test/Expectation.hs
+++ b/test/Expectation.hs
@@ -9,7 +9,12 @@ shouldContain containers element = do
    let res = element `elem` containers
    res `shouldBe` True

-withDirectory :: FilePath -> IO a -> IO a
-withDirectory dir action = bracket getCurrentDirectory
+withDirectory_ :: FilePath -> IO a -> IO a
+withDirectory_ dir action = bracket getCurrentDirectory
                                   setCurrentDirectory
                                   (\_ -> setCurrentDirectory dir >> action)
+
+withDirectory :: FilePath -> (FilePath -> IO a) -> IO a
+withDirectory dir action = bracket getCurrentDirectory
+                                   setCurrentDirectory
+                                   (\d -> setCurrentDirectory dir >> action d)