Handle reflected IP addresses (see bep 42).

1 files changed, 4 insertions, 3 deletions

diff --git a/tests/Network/KRPC/MessageSpec.hs b/tests/Network/KRPC/MessageSpec.hs
index 7aca4489..498ef679 100644
--- a/tests/Network/KRPC/MessageSpec.hs
+++ b/tests/Network/KRPC/MessageSpec.hs
@@ -20,7 +20,8 @@ instance Arbitrary KQuery where
   arbitrary = KQuery <$> pure (BInteger 0) <*> arbitrary <*> arbitrary
 
 instance Arbitrary KResponse where
-  arbitrary = KResponse <$> pure (BList []) <*> arbitrary
+  -- TODO: Abitrary instance for ReflectedIP
+  arbitrary = KResponse <$> pure (BList []) <*> arbitrary <*> pure Nothing
 
 instance Arbitrary KMessage where
   arbitrary = frequency
@@ -64,8 +65,8 @@ spec = do
 
     it "properly bencoded" $ do
       BE.decode "d1:rle1:t2:aa1:y1:re" `shouldBe`
-        Right (KResponse (BList []) "aa")
+        Right (KResponse (BList []) "aa" Nothing)
 
   describe "generic message" $ do
     it "properly bencoded (iso)" $ property $ \ km ->
-      BE.decode (BL.toStrict (BE.encode km)) `shouldBe` Right (km :: KMessage)
\ No newline at end of file
+      BE.decode (BL.toStrict (BE.encode km)) `shouldBe` Right (km :: KMessage)