From 1c38e5857764fe2ab8cb5ec55b31f13239278599 Mon Sep 17 00:00:00 2001
From: Sam Truzjan <pxqr.sta@gmail.com>
Date: Tue, 25 Feb 2014 19:23:42 +0400
Subject: Add metadata transfer scheduler spec

---
 .../BitTorrent/Exchange/Session/MetadataSpec.hs    | 71 ++++++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 tests/Network/BitTorrent/Exchange/Session/MetadataSpec.hs

(limited to 'tests/Network/BitTorrent/Exchange/Session')

diff --git a/tests/Network/BitTorrent/Exchange/Session/MetadataSpec.hs b/tests/Network/BitTorrent/Exchange/Session/MetadataSpec.hs
new file mode 100644
index 00000000..975ceb5b
--- /dev/null
+++ b/tests/Network/BitTorrent/Exchange/Session/MetadataSpec.hs
@@ -0,0 +1,71 @@
+{-# LANGUAGE RecordWildCards #-}
+module Network.BitTorrent.Exchange.Session.MetadataSpec (spec) where
+import Control.Concurrent
+import Data.ByteString as BS
+import Data.ByteString.Lazy as BL
+import Test.Hspec
+import Test.QuickCheck
+
+import Data.BEncode as BE
+import Data.Torrent
+import Data.Torrent.Piece as P
+import Network.BitTorrent.Core
+import Network.BitTorrent.Exchange.Message
+import Network.BitTorrent.Exchange.Session.Metadata
+
+import Config
+import Network.BitTorrent.CoreSpec ()
+
+
+placeholderAddr :: PeerAddr IP
+placeholderAddr = "0.0.0.0:0"
+
+chunkBy :: Int -> BS.ByteString -> [BS.ByteString]
+chunkBy s bs
+  | BS.null bs = []
+  | otherwise  = BS.take s bs : chunkBy s (BS.drop s bs)
+
+withUpdates :: Updates a -> IO a
+withUpdates m = do
+  Torrent {..} <- getTestTorrent
+  let infoDictLen = fromIntegral $ BL.length $ BE.encode tInfoDict
+  mvar <- newMVar (nullStatus infoDictLen)
+  runUpdates mvar placeholderAddr m
+
+simulateFetch :: InfoDict -> Updates (Maybe InfoDict)
+simulateFetch dict = go
+  where
+    blocks = chunkBy metadataPieceSize (BL.toStrict (BE.encode dict))
+    packPiece ix = P.Piece ix (blocks !! ix)
+    ih     = idInfoHash dict
+
+    go = do
+      mix <- scheduleBlock
+      case mix of
+        Nothing -> return Nothing
+        Just ix -> do
+          mdict <- pushBlock (packPiece ix) ih
+          maybe go (return . Just) mdict
+
+spec :: Spec
+spec = do
+  describe "scheduleBlock" $ do
+    it "never schedule the same index twice" $ do
+      pending
+
+  describe "resetPending" $ do
+    it "" $ do
+      pending
+
+  describe "cancelPending" $ do
+    it "must not throw an exception if cancel the same piece twice" $ do
+      pending
+
+  describe "pushBlock" $ do
+    it "assemble infodict from chunks" $ do
+      Torrent {..} <- getTestTorrent
+      mdict <- withUpdates $ simulateFetch tInfoDict
+      mdict `shouldBe` Just tInfoDict
+
+    it "must throw an exception if block if not requested" $ do
+      pending
\ No newline at end of file
-- 
cgit v1.2.3