Add Arbitrary instance for AnnounceQuery

1 files changed, 15 insertions, 0 deletions

diff --git a/tests/Network/BitTorrent/Core/PeerIdSpec.hs b/tests/Network/BitTorrent/Core/PeerIdSpec.hs
new file mode 100644
index 00000000..7ed8a976
--- /dev/null
+++ b/tests/Network/BitTorrent/Core/PeerIdSpec.hs
@@ -0,0 +1,15 @@
+module Network.BitTorrent.Core.PeerIdSpec (spec) where
+import Control.Applicative
+import Test.Hspec
+import Test.QuickCheck
+import Network.BitTorrent.Core.PeerId
+
+
+instance Arbitrary PeerId where
+  arbitrary = oneof
+    [ azureusStyle defaultClientId defaultVersionNumber <$> arbitrary
+    , shadowStyle  'X'             defaultVersionNumber <$> arbitrary
+    ]
+
+spec :: Spec
+spec = return ()
\ No newline at end of file