3 files changed, 9 insertions, 3 deletions

diff --git a/src/Network/BitTorrent/DHT.hs b/src/Network/BitTorrent/DHT.hs
index 44c55128..98ba1a56 100644
--- a/src/Network/BitTorrent/DHT.hs
+++ b/src/Network/BitTorrent/DHT.hs
@@ -156,7 +156,7 @@ isBootstrapped = T.full <$> getTable
 --
 --   This is blocking operation, use
 --   'Control.Concurrent.Async.Lifted.async' if needed.
-restore :: ByteString -> DHT ip ()
+restore :: ByteString -> IO (Node ip)
 restore = error "DHT.restore: not implemented"
 
 -- | Serialize current DHT session to byte string.
@@ -179,6 +179,8 @@ lookup topic = do      -- TODO retry getClosest if bucket is empty
   closest <- lift $ getClosest topic
   sourceList [closest] $= search topic (getPeersQ topic)
 
+-- TODO do not republish if the topic is already in announceSet
+
 -- | Announce that /this/ peer may have some pieces of the specified
 -- torrent. DHT will reannounce this data periodically using
 -- 'optReannounce' interval.
diff --git a/src/Network/BitTorrent/DHT/Session.hs b/src/Network/BitTorrent/DHT/Session.hs
index 43bd65e5..755985fc 100644
--- a/src/Network/BitTorrent/DHT/Session.hs
+++ b/src/Network/BitTorrent/DHT/Session.hs
@@ -212,6 +212,7 @@ nullSessionTokens = SessionTokens
   <*> liftIO getCurrentTime
   <*> pure defaultUpdateInterval
 
+-- TODO invalidate *twice* if needed
 invalidateTokens :: UTCTime -> SessionTokens -> SessionTokens
 invalidateTokens curTime ts @ SessionTokens {..}
   | curTime `diffUTCTime` lastUpdate > maxInterval = SessionTokens
diff --git a/tests/Network/BitTorrent/Tracker/RPC/UDPSpec.hs b/tests/Network/BitTorrent/Tracker/RPC/UDPSpec.hs
index 540a887c..d0f0f26c 100644
--- a/tests/Network/BitTorrent/Tracker/RPC/UDPSpec.hs
+++ b/tests/Network/BitTorrent/Tracker/RPC/UDPSpec.hs
@@ -33,8 +33,11 @@ validateInfo AnnounceQuery {..}  AnnounceInfo {..} = do
     peerList = getPeerList respPeers
 
 spec :: Spec
-spec = parallel $ do
-  forM_ trackerURIs $ \ uri ->
+spec = do
+--  describe "RpcException" $
+
+  parallel $ do
+   forM_ trackerURIs $ \ uri ->
     context (show uri) $ do
       describe "announce" $ do
         it "have valid response" $ do