Add naive session tokens implementation

2 files changed, 17 insertions, 3 deletions

diff --git a/tests/Network/BitTorrent/DHT/MessageSpec.hs b/tests/Network/BitTorrent/DHT/MessageSpec.hs
index 11a8fb15..ce2ac0de 100644
--- a/tests/Network/BitTorrent/DHT/MessageSpec.hs
+++ b/tests/Network/BitTorrent/DHT/MessageSpec.hs
@@ -12,9 +12,9 @@ import Network.Socket (PortNumber)
 import Test.Hspec
 import Test.QuickCheck
 
-import Network.BitTorrent.CoreSpec ()
-import Data.Torrent.InfoHashSpec   ()
-
+import Network.BitTorrent.CoreSpec      ()
+import Network.BitTorrent.DHT.TokenSpec ()
+import Data.Torrent.InfoHashSpec        ()
 
 remoteAddr :: SockAddr
 remoteAddr = SockAddrInet 6881 (256 * 256 * 256 + 127)
diff --git a/tests/Network/BitTorrent/DHT/TokenSpec.hs b/tests/Network/BitTorrent/DHT/TokenSpec.hs
new file mode 100644
index 00000000..90ff0f03
--- /dev/null
+++ b/tests/Network/BitTorrent/DHT/TokenSpec.hs
@@ -0,0 +1,14 @@
+{-# OPTIONS_GHC -fno-warn-orphans #-}
+module Network.BitTorrent.DHT.TokenSpec (spec) where
+import Control.Applicative
+import Data.String
+import Test.Hspec
+import Test.QuickCheck
+import Network.BitTorrent.DHT.Token
+
+
+instance Arbitrary Token where
+  arbitrary = fromString <$> arbitrary
+
+spec :: Spec
+spec = return ()
\ No newline at end of file