~ Move exsamples to tests.

3 files changed, 93 insertions, 0 deletions

diff --git a/tests/Client.hs b/tests/Client.hs
new file mode 100644
index 00000000..ec86639e
--- /dev/null
+++ b/tests/Client.hs
@@ -0,0 +1,27 @@
+{-# LANGUAGE OverloadedStrings #-}
+module Main (main) where
+
+import qualified Data.ByteString as B
+import System.Environment
+import Remote.KRPC
+import Shared
+
+
+addr :: RemoteAddr
+addr = (0, 6000)
+
+main :: IO ()
+main = do
+  print =<< call addr unitM ()
+  print =<< call addr echoM 0
+  call addr reverseM [1..1000]
+  print =<< call addr swapM (0, 1)
+  print =<< call addr shiftR ((), 1, [2..10])
+  let bs = B.replicate (32 * 1024) 0
+  bs' <- call addr echoBytes bs
+  print (bs == bs')
+
+{-
+  forM_ [1..] $ const $ do
+    async addr myconcat (replicate 100 [1..10])
+-}
diff --git a/tests/Server.hs b/tests/Server.hs
new file mode 100644
index 00000000..f636b0be
--- /dev/null
+++ b/tests/Server.hs
@@ -0,0 +1,16 @@
+{-# LANGUAGE IncoherentInstances #-}
+module Main (main) where
+
+import Remote.KRPC
+import Shared
+
+
+main :: IO ()
+main = server 6000
+  [ unitM ==> return
+  , echoM ==> return
+  , echoBytes ==> return
+  , swapM ==> \(a, b) -> return (b, a)
+  , reverseM ==> return . reverse
+  , shiftR ==> \(a, b, c) -> return (c, a, b)
+  ]
diff --git a/tests/Shared.hs b/tests/Shared.hs
new file mode 100644
index 00000000..e0e5268c
--- /dev/null
+++ b/tests/Shared.hs
@@ -0,0 +1,50 @@
+{-# LANGUAGE OverloadedStrings #-}
+module Shared
+       (echoM, echoBytes, unitM, swapM, reverseM, shiftR
+       ) where
+
+import Data.ByteString (ByteString)
+import Remote.KRPC
+
+unitM :: Method () ()
+unitM = method "unit" [] []
+
+echoM :: Method Int Int
+echoM = method "echo" ["x"] ["x"]
+
+echoBytes :: Method ByteString ByteString
+echoBytes = method "echoBytes" ["x"] ["x"]
+
+reverseM :: Method [Int] [Int]
+reverseM = method "reverse" ["xs"] ["ys"]
+
+swapM :: Method (Int, Int) (Int, Int)
+swapM = method "swap" ["x", "y"] ["b", "a"]
+
+shiftR :: Method ((), Int, [Int]) ([Int], (), Int)
+shiftR = method "shiftR" ["x", "y", "z"] ["a", "b", "c"]
+
+
+
+{-
+type NodeId = Int
+type InfoHashe = Int
+type NodeAddr = Int
+type Token = Int
+type
+
+ping :: Method NodeId NodeId
+ping = method "ping" ["id"] ["id"]
+
+find_node :: Method (NodeId, NodeId) (NodeId, NodeAddr)
+find_node = method "find_node" ["id", "target"] ["id", "nodes"]
+
+get_peers :: Method (NodeId :*: InfoHash) (NodeId, Token, NodeAddr :|: NodeAddr)
+get_peers = method "get_peers"
+  ("id", "target")
+  ("id", "token", view ("values" :|: "nodes"))
+view :: BEncodable -> Maybe BEncodable
+view  = undefined
+announce_peer :: Method (NodeId, InfoHash, PortNumber, Token) NodeId
+announce_peer = undefined
+-}
\ No newline at end of file