Go语言RPC接口测试全攻略

type Calculator struct{}<br><br>func (c *Calculator) Add(args Args, result *int) error {<br>    *result = add(args.A, args.B)<br>    return nil<br>}<br><br>// 独立的业务函数，便于测试<br>func add(a, b int) int {<br>    return a + b<br>}

func TestAdd(t *testing.T) {<br>    if got := add(2, 3); got != 5 {<br>        t.Errorf("add(2,3) = %d, want 5", got)<br>    }<br>}

func TestCalculator_Add_RPC(t *testing.T) {<br>    // 注册服务<br>    calc := new(Calculator)<br>    rpc.Register(calc)<br><br>    // 使用监听器<br>    listener, _ := net.Listen("tcp", "127.0.0.1:0")<br>    defer listener.Close()<br><br>    go rpc.Accept(listener)<br><br>    // 拨号连接<br>    client, err := rpc.Dial("tcp", listener.Addr().String())<br>    if err != nil {<br>        t.Fatal("dialing:", err)<br>    }<br>    defer client.Close()<br><br>    // 调用远程方法<br>    args := Args{A: 2, B: 3}<br>    var reply int<br>    err = client.Call("Calculator.Add", args, &reply)<br>    if err != nil {<br>        t.Error("call failed:", err)<br>    }<br><br>    if reply != 5 {<br>        t.Errorf("reply = %d, want 5", reply)<br>    }<br>}

type MathClient interface {<br>    Add(a, b int) (int, error)<br>}<br><br>type RPCMathClient struct {<br>    client *rpc.Client<br>}<br><br>func (c *RPCMathClient) Add(a, b int) (int, error) {<br>    args := Args{A: a, B: b}<br>    var reply int<br>    err := c.client.Call("Calculator.Add", args, &reply)<br>    return reply, err<br>}