1 files changed, 16 insertions, 0 deletions

diff --git a/tests/System/Torrent/StorageSpec.hs b/tests/System/Torrent/StorageSpec.hs
index 8d9dfd8f..8267b7a5 100644
--- a/tests/System/Torrent/StorageSpec.hs
+++ b/tests/System/Torrent/StorageSpec.hs
@@ -1,5 +1,6 @@
 module System.Torrent.StorageSpec (spec) where
 import Control.Exception
+import Data.ByteString.Lazy as BL
 import System.FilePath
 import System.Directory
 import System.IO.Unsafe
@@ -32,11 +33,26 @@ spec = before createLayout $ do
       writePiece (Piece 0 "") s `shouldThrow` (== StorageIsRO)
       close s
 
+    it "should fail if piece size do not match" $ do
+      withStorage ReadWrite 1 layout $ \ s ->
+        writePiece (Piece 0 "") s `shouldThrow` (== InvalidSize 0)
+
     it "should fail on negative index" $ do
       withStorage ReadWrite 0 layout $ \ s ->
         writePiece (Piece (-1) "") s `shouldThrow` (== InvalidIndex (-1))
 
+    it "should fail on out of upper bound index" $ do
+      withStorage ReadWrite 100 layout $ \ s -> do
+        let bs = BL.replicate 100 0
+        writePiece (Piece 1 bs) s
+        writePiece (Piece 2 bs) s `shouldThrow` (== InvalidIndex 2)
+
   describe "readPiece" $ do
     it "should fail on negative index" $
       withStorage ReadOnly 0 layout $ \ s ->
         readPiece (-1) s `shouldThrow` (== InvalidIndex (-1))
+
+    it "should fail on out of upper bound index" $ do
+      withStorage ReadOnly 100 layout $ \ s -> do
+        _ <- readPiece 1 s
+        readPiece 2 s `shouldThrow` (== InvalidIndex 2)