Go语言RPC接口测试全攻略

type Calculator struct{}

func (c *Calculator) Add(args Args, result *int) error {
    *result = add(args.A, args.B)
    return nil
}

// 独立的业务函数，便于测试
func add(a, b int) int {
    return a + b
}

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

func TestCalculator_Add_RPC(t *testing.T) {
    // 注册服务
    calc := new(Calculator)
    rpc.Register(calc)

    // 使用监听器
    listener, _ := net.Listen("tcp", "127.0.0.1:0")
    defer listener.Close()

    go rpc.Accept(listener)

    // 拨号连接
    client, err := rpc.Dial("tcp", listener.Addr().String())
    if err != nil {
        t.Fatal("dialing:", err)
    }
    defer client.Close()

    // 调用远程方法
    args := Args{A: 2, B: 3}
    var reply int
    err = client.Call("Calculator.Add", args, &reply)
    if err != nil {
        t.Error("call failed:", err)
    }

    if reply != 5 {
        t.Errorf("reply = %d, want 5", reply)
    }
}

type MathClient interface {
    Add(a, b int) (int, error)
}

type RPCMathClient struct {
    client *rpc.Client
}

func (c *RPCMathClient) Add(a, b int) (int, error) {
    args := Args{A: a, B: b}
    var reply int
    err := c.client.Call("Calculator.Add", args, &reply)
    return reply, err
}