From 5e92eec501e0a1ca6d09a01e078cf54ff3277273 Mon Sep 17 00:00:00 2001
From: Sam T <pxqr.sta@gmail.com>
Date: Sat, 25 May 2013 21:41:15 +0400
Subject: ~ Use PortNumber instead of Word16 in Port message.

---
 tests/Encoding.hs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'tests')

diff --git a/tests/Encoding.hs b/tests/Encoding.hs
index fe0e1d7e..d43045bc 100644
--- a/tests/Encoding.hs
+++ b/tests/Encoding.hs
@@ -13,6 +13,8 @@ import Test.Framework.Providers.QuickCheck2 (testProperty)
 import Test.QuickCheck
 
 import Network.URI
+import Network
+
 
 import Data.Bitfield
 import Data.Torrent
@@ -33,6 +35,9 @@ instance Arbitrary Block where
 
 deriving instance Arbitrary Bitfield
 
+instance Arbitrary PortNumber where
+  arbitrary = fromIntegral <$> (arbitrary :: Gen Word16)
+
 instance Arbitrary Message where
   arbitrary = oneof
     [ pure KeepAlive
@@ -45,7 +50,7 @@ instance Arbitrary Message where
     , Request <$> arbitrary
     , Piece <$> arbitrary
     , Cancel <$> arbitrary
-    , Port <$> choose (0, fromIntegral (maxBound :: Word16))
+    , Port <$> arbitrary
     ]
 
 instance Arbitrary PeerID where
-- 
cgit v1.2.3