Function TestRawCaller

rpc/mux/rawcaller_test.go:44–78

(t *testing.T)

}

func TestRawCaller(t *testing.T) {
	Convey("test raw caller methods", t, func() {
		s := NewServer().WithAcceptConnFunc(rpc.AcceptRawConn)
		err := s.RegisterService("Test", &testService{})
		So(err, ShouldBeNil)
		l, err := net.Listen("tcp", ":0")
		So(err, ShouldBeNil)
		s.SetListener(l)
		go s.Serve()
		defer s.Stop()
		c := NewRawCaller(l.Addr().String())
		defer c.Close()
		var resp int
		err = c.Call("Test.Test", 1, &resp)
		So(err, ShouldBeNil)
		So(resp, ShouldEqual, 2)
		err = c.Call("Test.TestFailed", 1, nil)
		So(err, ShouldNotBeNil)
		So(errors.Cause(err).Error(), ShouldEqual, "failed")
		err = c.Call("Test.TestReconnect", 1, nil)
		So(err, ShouldNotBeNil)
		So(err.Error(), ShouldContainSubstring, "shut down")
		So(c.Target(), ShouldEqual, l.Addr().String())
		err = c.Call("Test.Test", 2, &resp)
		So(err, ShouldBeNil)
		So(resp, ShouldEqual, 3)

		// test new client
		c2 := c.New()
		defer c2.Close()
		err = c2.Call("Test.Test", 4, &resp)
		So(err, ShouldBeNil)
		So(resp, ShouldEqual, 5)
	})
}