Uncomment session spec

1 files changed, 16 insertions, 0 deletions

diff --git a/tests/Config.hs b/tests/Config.hs
index 3dea6603..b37b727b 100644
--- a/tests/Config.hs
+++ b/tests/Config.hs
@@ -11,8 +11,12 @@ module Config
          -- * For item specs
        , getEnvOpts
        , getThisOpts
+       , getMyAddr
+
        , getRemoteOpts
        , withRemote
+       , withRemoteAddr
+
        , getTestTorrent
        ) where
 
@@ -29,6 +33,7 @@ import System.IO.Unsafe
 import Test.Hspec
 
 import Data.Torrent
+import Network.BitTorrent.Core (IP, PeerAddr (PeerAddr), genPeerId)
 
 
 type ClientName = String
@@ -138,6 +143,17 @@ withRemote action = do
     Nothing   -> pendingWith "Remote client isn't running"
     Just opts -> action opts
 
+withRemoteAddr :: (PeerAddr IP -> Expectation) -> Expectation
+withRemoteAddr action = do
+  withRemote $ \ ClientOpts {..} ->
+    action (PeerAddr Nothing "0.0.0.0" peerPort)
+
+getMyAddr :: IO (PeerAddr (Maybe IP))
+getMyAddr = do
+  ClientOpts {..} <- getThisOpts
+  pid <- genPeerId
+  return $ PeerAddr (Just pid) Nothing peerPort
+
 getTestTorrent :: IO Torrent
 getTestTorrent = do
   EnvOpts {..} <- getEnvOpts