Golang中使用Mock时遇到常量相关错误的解决方案

这个问题通常是由于GoMock在匹配结构体参数时，对结构体内部字段的比较方式导致的。特别是当结构体包含int64类型字段时，直接使用gomock.Eq()可能无法正确匹配。

问题根源在于GoMock的Eq()匹配器使用reflect.DeepEqual()进行深度比较，而int64类型的常量（如2）在传递给函数时，可能会被推断为不同的整数类型（如int），导致类型不匹配。

以下是解决方案和示例代码：

解决方案1：使用gomock.Eq()时确保类型一致

// 错误的写法
mockClient.EXPECT().GetSellersByStatus(
    gomock.Any(),
    &generated.SellersByStatusRequest{
        SamId:   "2",
        Status:  "PFA",
        PerPage: 2,  // 这里2是int类型，但结构体字段可能是int64
        PageNo:  1,
        Count:   true,
    },
).Return(&response, nil)

// 正确的写法 - 显式指定int64类型
mockClient.EXPECT().GetSellersByStatus(
    gomock.Any(),
    &generated.SellersByStatusRequest{
        SamId:   "2",
        Status:  "PFA",
        PerPage: int64(2),  // 显式转换为int64
        PageNo:  int64(1),  // 显式转换为int64
        Count:   true,
    },
).Return(&response, nil)

解决方案2：使用自定义匹配器

// 创建自定义匹配器
func matchSellersRequest(expected *generated.SellersByStatusRequest) gomock.Matcher {
    return gomock.GotFormatterAdapter(
        gomock.GotFormatterFunc(func(i interface{}) string {
            return fmt.Sprintf("%v", i)
        }),
        gomock.WantFormatter(
            gomock.StringerFunc(func() string {
                return fmt.Sprintf("%v", expected)
            }),
            gomock.Eq(expected),
        ),
    )
}

// 在测试中使用
mockClient.EXPECT().GetSellersByStatus(
    gomock.Any(),
    matchSellersRequest(&generated.SellersByStatusRequest{
        SamId:   "2",
        Status:  "PFA",
        PerPage: 2,
        PageNo:  1,
        Count:   true,
    }),
).Return(&response, nil)

解决方案3：使用gomock.Any()忽略特定字段

mockClient.EXPECT().GetSellersByStatus(
    gomock.Any(),
    gomock.AssignableToTypeOf(&generated.SellersByStatusRequest{}),
).DoAndReturn(func(ctx context.Context, req *generated.SellersByStatusRequest) (*generated.SellersByStatusResponse, error) {
    // 在Do函数中验证具体字段
    if req.SamId != "2" || req.Status != "PFA" || req.PerPage != 2 || req.PageNo != 1 || !req.Count {
        return nil, fmt.Errorf("unexpected request: %v", req)
    }
    return &response, nil
})

关于fmt.Println的作用

fmt.Println(arg1)能解决问题是因为：

1. 强制类型推断：当调用fmt.Println时，Go编译器需要确定参数的具体类型，这可能会改变常量类型的推断
2. 内存布局变化：打印操作可能影响了结构体在内存中的对齐方式
3. 编译器优化：添加打印语句可能改变了编译器的优化行为

但这只是掩盖了问题，不是真正的解决方案。正确的做法是确保测试代码中使用的类型与实际代码中的类型完全一致。

完整示例

func Test_queryResolver_SellersByStatus(t *testing.T) {
    ctrl := gomock.NewController(t)
    defer ctrl.Finish()
    
    mockClient := mocks.NewMockBoQueryClient(ctrl)
    
    // 准备测试数据
    expectedRequest1 := &generated.SellersByStatusRequest{
        SamId:   "2",
        Status:  "PFA",
        PerPage: int64(2),  // 关键：显式指定int64
        PageNo:  int64(1),  // 关键：显式指定int64
        Count:   true,
    }
    
    expectedRequest2 := &generated.SellersByStatusRequest{
        SamId:   "2",
        Status:  "PFA",
        PerPage: int64(2),
        PageNo:  int64(2),
        Count:   true,
    }
    
    // 设置mock期望
    mockClient.EXPECT().GetSellersByStatus(
        gomock.Any(),
        expectedRequest1,
    ).Return(&generated.SellersByStatusResponse{
        // 返回数据
    }, nil)
    
    mockClient.EXPECT().GetSellersByStatus(
        gomock.Any(),
        expectedRequest2,
    ).Return(&generated.SellersByStatusResponse{
        // 返回数据
    }, nil)
    
    // 执行测试
    // ...
}

关键点：确保测试代码中结构体字段的类型与生产代码中结构体定义的类型完全一致，特别是对于数值类型，要显式指定类型（如int64(2)而不是2）。