Function TestIncCounter

rpc/mux/server_test.go:109–153

(t *testing.T)

}

func TestIncCounter(t *testing.T) {
	log.SetLevel(log.FatalLevel)
	addr := "127.0.0.1:0"
	l, err := net.Listen("tcp", addr)
	if err != nil {
		log.Fatal(err)
	}

	server, err := NewServerWithService(ServiceMap{"Test": NewTestService()})
	server.SetListener(l)
	go server.WithAcceptConnFunc(rpc.AcceptRawConn).Serve()

	conn, err := net.Dial("tcp", l.Addr().String())
	if err != nil {
		log.Fatal(err)
	}
	stream, err := NewOneOffMuxConn(conn)
	if err != nil {
		log.Fatal(err)
	}
	client := rpc.NewClient(stream)

	rep := new(TestRep)
	err = client.Call("Test.IncCounter", &TestReq{Step: 10}, rep)
	if err != nil {
		log.Fatal(err)
	}
	CheckNum(rep.Ret, 10, t)

	err = client.Call("Test.IncCounter", &TestReq{Step: 10}, rep)
	if err != nil {
		log.Fatal(err)
	}
	CheckNum(rep.Ret, 20, t)

	repSimple := new(int32)
	err = client.Call("Test.IncCounterSimpleArgs", 10, repSimple)
	if err != nil {
		log.Fatal(err)
	}
	CheckNum(*repSimple, 30, t)

	_ = client.Close()
	server.Stop()
}

func TestIncCounterSimpleArgs(t *testing.T) {
	log.SetLevel(log.FatalLevel)