3 files changed, 17 insertions, 1 deletions

diff --git a/tests/Client.hs b/tests/Client.hs
index d762976d..1b9ef8d2 100644
--- a/tests/Client.hs
+++ b/tests/Client.hs
@@ -4,6 +4,7 @@ module Main (main) where
 import Control.Concurrent
 import Control.Exception
 import qualified Data.ByteString as B
+import Data.BEncode
 import System.Environment
 import System.Process
 import System.FilePath
@@ -65,4 +66,7 @@ tests =
   , testCase "echo bytestring" $
       let bs = B.replicate 400 0 in
       bs ==? call addr echoBytes bs
+
+  , testCase "raw method" $
+      BInteger 10 ==? call addr rawM (BInteger 10)
   ]
diff --git a/tests/Server.hs b/tests/Server.hs
index f636b0be..7cd6a5d6 100644
--- a/tests/Server.hs
+++ b/tests/Server.hs
@@ -1,6 +1,7 @@
 {-# LANGUAGE IncoherentInstances #-}
 module Main (main) where
 
+import Data.BEncode
 import Remote.KRPC
 import Shared
 
@@ -13,4 +14,5 @@ main = server 6000
   , swapM ==> \(a, b) -> return (b, a)
   , reverseM ==> return . reverse
   , shiftR ==> \(a, b, c) -> return (c, a, b)
+  , rawM      ==> return
   ]
diff --git a/tests/Shared.hs b/tests/Shared.hs
index bf29365b..a04b6093 100644
--- a/tests/Shared.hs
+++ b/tests/Shared.hs
@@ -1,9 +1,16 @@
 {-# LANGUAGE OverloadedStrings #-}
 module Shared
-       (echoM, echoBytes, unitM, swapM, reverseM, shiftR
+       ( echoM
+       , echoBytes
+       , unitM
+       , swapM
+       , reverseM
+       , shiftR
+       , rawM
        ) where
 
 import Data.ByteString (ByteString)
+import Data.BEncode
 import Remote.KRPC
 
 unitM :: Method () ()
@@ -23,3 +30,6 @@ swapM = method "swap" ["x", "y"] ["b", "a"]
 
 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