~ Allow passing raw dictionaries.

We need this in Kademlia DHT -- there are method which return
dictionaries with different keys depending on DHT server state.

3 files changed, 14 insertions, 1 deletions

diff --git a/tests/Client.hs b/tests/Client.hs
index 1b9ef8d2..313cd56e 100644
--- a/tests/Client.hs
+++ b/tests/Client.hs
@@ -5,6 +5,7 @@ import Control.Concurrent
 import Control.Exception
 import qualified Data.ByteString as B
 import Data.BEncode
+import Data.Map
 import System.Environment
 import System.Process
 import System.FilePath
@@ -69,4 +70,11 @@ tests =
 
   , testCase "raw method" $
       BInteger 10 ==? call addr rawM (BInteger 10)
+
+  , testCase "raw dict" $
+      let dict = BDict $ fromList
+                 [ ("some_int", BInteger 100)
+                 , ("some_list", BList [BInteger 10])
+                 ]
+      in dict ==? call addr rawDictM dict
   ]
diff --git a/tests/Server.hs b/tests/Server.hs
index 7cd6a5d6..aaf6d9f2 100644
--- a/tests/Server.hs
+++ b/tests/Server.hs
@@ -15,4 +15,5 @@ main = server 6000
   , reverseM ==> return . reverse
   , shiftR ==> \(a, b, c) -> return (c, a, b)
   , rawM      ==> return
+  , rawDictM  ==> return
   ]
diff --git a/tests/Shared.hs b/tests/Shared.hs
index a04b6093..f64112da 100644
--- a/tests/Shared.hs
+++ b/tests/Shared.hs
@@ -7,6 +7,7 @@ module Shared
        , reverseM
        , shiftR
        , rawM
+       , rawDictM
        ) where
 
 import Data.ByteString (ByteString)
@@ -32,4 +33,7 @@ shiftR :: Method ((), Int, [Int]) ([Int], (), Int)
 shiftR = method "shiftR" ["x", "y", "z"] ["a", "b", "c"]
 
 rawM :: Method BEncode BEncode
-rawM = method "rawM" [""] [""]
\ No newline at end of file
+rawM = method "rawM" [""] [""]
+
+rawDictM :: Method BEncode BEncode
+rawDictM = method "m" [] []
\ No newline at end of file