func TestJSONRPC(t *testing.T) {
type Args struct{ A, B int }
type Reply int
lis, err := net.Listen(network, addr)
if err != nil {
t.Fatal(err)
}
srv := rpc2.NewServer()
srv.Handle("add", func(client *rpc2.Client, args *Args, reply *Reply) error {
*reply = Reply(args.A + args.B)
var rep Reply
err := client.Call("mult", Args{2, 3}, &rep)
if err != nil {
t.Fatal(err)
}
if rep != 6 {
t.Fatalf("not expected: %d", rep)
}
return nil
})
srv.Handle("addPos", func(client *rpc2.Client, args []interface{}, result *float64) error {
*result = args[0].(float64) + args[1].(float64)
return nil
})
number := make(chan int, 1)
srv.Handle("set", func(client *rpc2.Client, i int, _ *struct{}) error {
number <- i
return nil
})
go func() {
conn, err := lis.Accept()
if err != nil {
t.Fatal(err)
}
srv.ServeCodec(NewJSONCodec(conn))
}()
conn, err := net.Dial(network, addr)
if err != nil {
t.Fatal(err)
}
clt := rpc2.NewClientWithCodec(NewJSONCodec(conn))
clt.Handle("mult", func(client *rpc2.Client, args *Args, reply *Reply) error {
*reply = Reply(args.A * args.B)
return nil
})
go clt.Run()
// Test Call.
var rep Reply
err = clt.Call("add", Args{1, 2}, &rep)
if err != nil {
t.Fatal(err)
}
if rep != 3 {
t.Fatalf("not expected: %d", rep)
}
// Test notification.
err = clt.Notify("set", 6)
if err != nil {
t.Fatal(err)
}
select {
case i := <-number:
if i != 6 {
t.Fatalf("unexpected number: %d", i)
}
case <-time.After(time.Second):
t.Fatal("did not get notification")
}
// Test undefined method.
err = clt.Call("foo", 1, &rep)
if err.Error() != "rpc2: can't find method foo" {
t.Fatal(err)
}
// Test Positional arguments.
var result float64
err = clt.Call("addPos", []interface{}{1, 2}, &result)
if err != nil {
t.Fatal(err)
}
if result != 3 {
t.Fatalf("not expected: %d", result)
}
}
// Registers a new BiJsonRpc name
func (s *Server) BijsonRegisterName(method string, handlerFunc interface{}) {
if s.bijsonSrv == nil {
s.bijsonSrv = rpc2.NewServer()
}
s.bijsonSrv.Handle(method, handlerFunc)
}