1 files changed, 42 insertions, 0 deletions

diff --git a/tests/System/Torrent/StorageSpec.hs b/tests/System/Torrent/StorageSpec.hs
new file mode 100644
index 00000000..8d9dfd8f
--- /dev/null
+++ b/tests/System/Torrent/StorageSpec.hs
@@ -0,0 +1,42 @@
+module System.Torrent.StorageSpec (spec) where
+import Control.Exception
+import System.FilePath
+import System.Directory
+import System.IO.Unsafe
+import Test.Hspec
+
+import Data.Torrent.Layout
+import Data.Torrent.Piece
+import System.Torrent.Storage
+
+
+layout :: FileLayout FileSize
+layout =
+  [ (dir </> "_a", 20)
+  , (dir </> "_b", 50)
+  , (dir </> "_c", 100)
+  , (dir </> "_d", 5)
+  ]
+  where
+    dir = unsafePerformIO $ getTemporaryDirectory
+
+createLayout :: IO ()
+createLayout =
+  bracket (open ReadWriteEx 0 layout) close (const (return ()))
+
+spec :: Spec
+spec = before createLayout $ do
+  describe "writePiece" $ do
+    it "should fail gracefully on write operation in RO mode" $ do
+      s <- open ReadOnly 0 layout
+      writePiece (Piece 0 "") s `shouldThrow` (== StorageIsRO)
+      close s
+
+    it "should fail on negative index" $ do
+      withStorage ReadWrite 0 layout $ \ s ->
+        writePiece (Piece (-1) "") s `shouldThrow` (== InvalidIndex (-1))
+
+  describe "readPiece" $ do
+    it "should fail on negative index" $
+      withStorage ReadOnly 0 layout $ \ s ->
+        readPiece (-1) s `shouldThrow` (== InvalidIndex (-1))