2 files changed, 75 insertions, 0 deletions

diff --git a/bittorrent.cabal b/bittorrent.cabal
index a398dd6d..8f7b86a8 100644
--- a/bittorrent.cabal
+++ b/bittorrent.cabal
@@ -196,6 +196,7 @@ test-suite spec
                        Network.BitTorrent.DHTSpec
                        Network.BitTorrent.DHT.MessageSpec
                        Network.BitTorrent.DHT.RoutingSpec
+                       Network.BitTorrent.DHT.SessionSpec
                        Network.BitTorrent.DHT.TokenSpec
                        Network.BitTorrent.Tracker.CacheSpec
                        Network.BitTorrent.Tracker.ListSpec
diff --git a/tests/Network/BitTorrent/DHT/SessionSpec.hs b/tests/Network/BitTorrent/DHT/SessionSpec.hs
new file mode 100644
index 00000000..75c95462
--- /dev/null
+++ b/tests/Network/BitTorrent/DHT/SessionSpec.hs
@@ -0,0 +1,74 @@
+{-# LANGUAGE ScopedTypeVariables #-}
+module Network.BitTorrent.DHT.SessionSpec (spec) where
+import Control.Monad.Reader
+import Data.Default
+import Test.Hspec
+import Test.QuickCheck
+
+import Network.BitTorrent.Core
+import Network.BitTorrent.DHT
+import Network.BitTorrent.DHT.Message
+import Network.BitTorrent.DHT.Session
+
+import Data.Torrent.InfoHashSpec ()
+import Network.BitTorrent.CoreSpec ()
+import Network.BitTorrent.DHT.TokenSpec ()
+
+
+myAddr :: NodeAddr IPv4
+myAddr = "127.0.0.1:60000"
+
+simpleDHT :: DHT IPv4 a -> IO a
+simpleDHT = dht def myAddr
+
+isRight :: Either a b -> Bool
+isRight (Left  _) = False
+isRight (Right _) = True
+
+isLeft :: Either a b -> Bool
+isLeft = not . isRight
+
+spec :: Spec
+spec = do
+  describe "tokens" $ do
+    it "should not complain about valid token" $
+      property $ \ (addrs :: [NodeAddr IPv4]) -> do
+        simpleDHT $ do
+          forM_ addrs $ \ addr -> do
+            token <- grantToken addr
+            checkToken addr token
+{-
+    it "" $ property $ \ (addr :: NodeAddr IPv4) token -> do
+      simpleDHT (checkToken addr token) `shouldThrow` (== undefined)
+-}
+  describe "routing table" $ do
+    return ()
+
+  describe "peer storage" $ do
+    it "should return nodes, if there are no peers" $ property $ \ ih -> do
+      nodes <- simpleDHT $ do getPeerList ih
+      nodes `shouldSatisfy` isLeft
+
+    it "should return peers, if any" $ property $ \ ih addr -> do
+      peers <- simpleDHT $ do
+                 insertPeer ih addr
+                 getPeerList ih
+      peers `shouldSatisfy` isRight
+
+  describe "topic storage" $ do
+    return ()
+
+  describe "messaging" $ do
+    describe "queryNode" $ do
+      it "should always ping this node" $ do
+        (rid, tid) <- simpleDHT $ do
+          (remoteId, Ping) <- queryNode myAddr Ping
+          thisId    <-  asks thisNodeId
+          return (remoteId, thisId)
+        rid `shouldBe` tid
+
+    describe "queryParallel" $ do
+      return ()
+
+    describe "(<@>) operator" $ do
+      return ()
\ No newline at end of file