1 files changed, 44 insertions, 0 deletions

diff --git a/tests/Network/BitTorrent/Tracker/ListSpec.hs b/tests/Network/BitTorrent/Tracker/ListSpec.hs
new file mode 100644
index 00000000..8decd3c9
--- /dev/null
+++ b/tests/Network/BitTorrent/Tracker/ListSpec.hs
@@ -0,0 +1,44 @@
+module Network.BitTorrent.Tracker.ListSpec (spec) where
+import Control.Exception
+import Data.Default
+import Data.Foldable as F
+import Data.List as L
+import Data.Maybe
+import Network.URI
+import Test.Hspec
+
+import Data.Torrent
+import Network.BitTorrent.Tracker.List
+import Network.BitTorrent.Tracker.RPC
+
+
+uris :: [URI]
+uris = fmap (fromJust . parseURI . renderURI) [1..10 :: Int]
+  where
+    renderURI n = "http://" ++ show n ++ ".org"
+
+list :: TrackerList URI
+list = fromJust $ trackerList def { tAnnounceList = Just [uris] }
+
+spec :: Spec
+spec = do
+  describe "TrackerList" $ do
+    it "trackerList is not empty" $ do
+      pending
+
+    it "shuffleTiers (may fail with very small probability)" $ do
+      list' <- shuffleTiers list
+      list' `shouldSatisfy` (/= list)
+
+    it "traverseAll" $ do
+      xs <- traverseAll (\ uri -> if uri == L.last uris
+               then throwIO (GenericException "")
+               else return uri { uriScheme = "udp://" }) list
+      let udps = F.sum $ fmap (fromEnum . ("udp://" ==) . uriScheme) xs
+      udps `shouldBe` pred (L.length uris)
+
+    it "traverseTiers" $ do
+      xs' <- traverseTiers (\ uri -> if uri == L.last uris then return uri
+                              else throwIO (GenericException "")) list
+
+      return ()