1 files changed, 19 insertions, 19 deletions

diff --git a/tests/Network/BitTorrent/DHT/MessageSpec.hs b/tests/Network/BitTorrent/DHT/MessageSpec.hs
index ae9ab487..3b1f5acd 100644
--- a/tests/Network/BitTorrent/DHT/MessageSpec.hs
+++ b/tests/Network/BitTorrent/DHT/MessageSpec.hs
@@ -61,16 +61,16 @@ spec = do
          running <- retry 5 $ timeout (100 * 1000) $ do
                       nid <- genNodeId
                       Response _remoteAddr Ping <-
-                          rpc (query remoteAddr (Query nid Ping))
+                          rpc (query remoteAddr (Query nid False Ping))
                       return ()
          running `shouldSatisfy` isJust
 
   describe "ping" $ do
     it "properly bencoded" $ do
       BE.decode "d2:id20:abcdefghij0123456789e"
-        `shouldBe` Right (Query "abcdefghij0123456789" Ping)
+        `shouldBe` Right (Query "abcdefghij0123456789" False Ping)
 
-      BE.encode (Query "abcdefghij0123456789" Ping)
+      BE.encode (Query "abcdefghij0123456789" False Ping)
         `shouldBe` "d2:id20:abcdefghij0123456789e"
 
       BE.decode "d2:id20:mnopqrstuvwxyz123456e"
@@ -80,22 +80,22 @@ spec = do
         `shouldBe` "d2:id20:mnopqrstuvwxyz123456e"
 
     it "properly bencoded (iso)" $ property $ \ nid -> do
-      prop_bencode (Query    nid Ping)
+      prop_bencode (Query    nid False Ping)
       prop_bencode (Response nid Ping)
 
     it "does compatible with existing DHT" $ do
       nid <- genNodeId
-      Response _remoteAddr Ping <- rpc (query remoteAddr (Query nid Ping))
+      Response _remoteAddr Ping <- rpc (query remoteAddr (Query nid False Ping))
       return ()
 
   describe "find_node" $ do
     it "properly bencoded" $ do
       BE.decode "d2:id20:abcdefghij0123456789\
                  \6:target20:mnopqrstuvwxyz123456e"
-        `shouldBe` Right (Query "abcdefghij0123456789"
+        `shouldBe` Right (Query "abcdefghij0123456789" False
                       (FindNode "mnopqrstuvwxyz123456"))
 
-      BE.encode (Query "abcdefghij0123456789"
+      BE.encode (Query "abcdefghij0123456789" False
                  (FindNode "mnopqrstuvwxyz123456"))
         `shouldBe`
          "d2:id20:abcdefghij01234567896:target20:mnopqrstuvwxyz123456e"
@@ -109,13 +109,13 @@ spec = do
        `shouldBe` Right (Response nid (NodeFound [NodeInfo nid' naddr]))
 
     it "properly bencoded (iso)" $ property $ \ nid x xs -> do
-      prop_bencode (Query    nid (FindNode  x))
+      prop_bencode (Query    nid False (FindNode  x))
       prop_bencode (Response nid (NodeFound (xs :: [NodeInfo IPv4] )))
 
     it "does compatible with existing DHT" $ do
       nid <- genNodeId
       Response _remoteAddr (NodeFound xs) <- rpc $ do
-        query remoteAddr (Query nid (FindNode nid))
+        query remoteAddr (Query nid False (FindNode nid))
       L.length (xs :: [NodeInfo IPv4]) `shouldSatisfy` (> 0)
 
   describe "get_peers" $ do
@@ -123,7 +123,7 @@ spec = do
        BE.decode "d2:id20:abcdefghij0123456789\
                   \9:info_hash20:mnopqrstuvwxyz123456\
                   \e"
-         `shouldBe` Right (Query "abcdefghij0123456789"
+         `shouldBe` Right (Query "abcdefghij0123456789" False
                            (GetPeers "mnopqrstuvwxyz123456"))
 
        BE.decode "d2:id20:abcdefghij0123456789\
@@ -147,7 +147,7 @@ spec = do
              }))
 
     it "properly bencoded (iso)" $ property $ \ nid topic exs token -> do
-      prop_bencode (Query    nid (GetPeers topic))
+      prop_bencode (Query    nid False (GetPeers topic))
       let _ = exs :: Either [NodeInfo IPv4] [PeerAddr IPv4]
       let nullPeerId paddr = paddr {peerId = Nothing}
       let nullPeerIds      = either Left (Right . L.map nullPeerId)
@@ -156,7 +156,7 @@ spec = do
     it "does compatible with existing DHT" $ do
       nid <- genNodeId
       Response _remoteId (GotPeers {..})
-        <- rpc $ query remoteAddr (Query nid (GetPeers def))
+        <- rpc $ query remoteAddr (Query nid False (GetPeers def))
       let _ = peers :: Either [NodeInfo IPv4] [PeerAddr IPv4]
       either L.length L.length peers `shouldSatisfy` (> 0)
 
@@ -167,7 +167,7 @@ spec = do
                  \4:porti6881e\
                  \5:token8:aoeusnth\
                 \e" `shouldBe` Right
-        (Query "abcdefghij0123456789"
+        (Query "abcdefghij0123456789" False
                    (Announce False "mnopqrstuvwxyz123456" 6881 "aoeusnth"))
 
       BE.decode "d2:id20:abcdefghij0123456789\
@@ -176,7 +176,7 @@ spec = do
                  \4:porti6881e\
                  \5:token8:aoeusnth\
                 \e" `shouldBe` Right
-        (Query "abcdefghij0123456789"
+        (Query "abcdefghij0123456789" False
                    (Announce True "mnopqrstuvwxyz123456" 6881 "aoeusnth"))
 
 
@@ -185,27 +185,27 @@ spec = do
         (Response "mnopqrstuvwxyz123456" Announced)
 
     it "properly bencoded (iso)" $ property $ \ nid flag topic port token -> do
-      prop_bencode (Query    nid (Announce flag topic port token))
+      prop_bencode (Query    nid False (Announce flag topic port token))
       prop_bencode (Response nid (Announced))
 
 
     it "does compatible with existing DHT" $ do
       nid <- genNodeId
       Response _remoteId Announced <- rpc $ do
-        Response _ GotPeers {..} <- query remoteAddr (Query nid (GetPeers def))
+        Response _ GotPeers {..} <- query remoteAddr (Query nid False (GetPeers def))
         let _ = peers :: Either [NodeInfo IPv4] [PeerAddr IPv4]
-        query remoteAddr (Query nid (Announce False def thisPort grantedToken))
+        query remoteAddr (Query nid False (Announce False def thisPort grantedToken))
       return ()
 
     it "does fail on invalid token" $ do
       nid <- genNodeId
       (rpc $ do
-        Response _ GotPeers {..} <- query remoteAddr (Query nid (GetPeers def))
+        Response _ GotPeers {..} <- query remoteAddr (Query nid False (GetPeers def))
         let _ = peers :: Either [NodeInfo IPv4] [PeerAddr IPv4]
         let invalidToken = ""
         let q :: MonadKRPC h m => SockAddr -> Query Announce
               -> m (Response Announced)
             q = query
-        q remoteAddr (Query nid (Announce False def thisPort invalidToken)))
+        q remoteAddr (Query nid False (Announce False def thisPort invalidToken)))
           `shouldThrow` isQueryError
       return ()