Gomock 教程：为模拟函数设置返回值

package mypackage

// 定义一个业务结构体
type Question struct {
    ID   int
    Text string
}

// 定义一个接口
type QuestionGateway interface {
    GetQuestionById(id int) Question
}

// 假设这是在你的测试文件中，例如 mypackage_test.go
package mypackage_test

import (
    "testing"

    "mypackage" // 导入你的业务包
    "mypackage/mock_mypackage" // 导入 gomock 生成的 mock 包

    "github.com/golang/mock/gomock"
)

func TestGetQuestionByIdReturnsSpecificValue(t *testing.T) {
    // 1. 创建一个gomock控制器
    // 控制器管理所有mock对象的生命周期和期望。
    ctrl := gomock.NewController(t)
    // 确保在测试结束时调用ctrl.Finish()来验证所有期望是否被满足。
    defer ctrl.Finish()

    // 2. 创建MockQuestionGateway实例
    mockGateway := mock_mypackage.NewMockQuestionGateway(ctrl)

    // 3. 设置期望和返回值
    // 当mockGateway的GetQuestionById方法以参数1被调用时，
    // 它应该返回Question{ID: 1, Text: "Foo"}。
    expectedQuestion := mypackage.Question{ID: 1, Text: "Foo"}
    mockGateway.EXPECT().GetQuestionById(1).Return(expectedQuestion)

    // 4. 调用被测试代码（这里直接调用mock方法来演示）
    // 在实际测试中，这里会是你业务逻辑中调用mockGateway的方法。
    returnedQuestion := mockGateway.GetQuestionById(1)

    // 5. 验证返回值
    if returnedQuestion != expectedQuestion {
        t.Errorf("Expected question %v, but got %v", expectedQuestion, returnedQuestion)
    }

    // 如果我们尝试以不同的参数调用，gomock会报错，因为没有设置相应的期望
    // returnedQuestion2 := mockGateway.GetQuestionById(2) // 这会导致测试失败，因为没有期望GetQuestionById(2)
}

package mypackage

import "errors"

type QuestionGateway interface {
    GetQuestionById(id int) (Question, error) // 返回Question和error
}

func TestGetQuestionByIdReturnsMultipleValues(t *testing.T) {
    ctrl := gomock.NewController(t)
    defer ctrl.Finish()

    mockGateway := mock_mypackage.NewMockQuestionGateway(ctrl)

    // 期望成功返回一个Question和nil错误
    expectedQuestion := mypackage.Question{ID: 1, Text: "Another Foo"}
    mockGateway.EXPECT().GetQuestionById(1).Return(expectedQuestion, nil)

    // 期望返回一个空Question和一个特定的错误
    expectedError := errors.New("question not found")
    mockGateway.EXPECT().GetQuestionById(2).Return(mypackage.Question{}, expectedError)

    // 测试成功情况
    q1, err1 := mockGateway.GetQuestionById(1)
    if err1 != nil || q1 != expectedQuestion {
        t.Errorf("Expected (%v, nil), got (%v, %v)", expectedQuestion, q1, err1)
    }

    // 测试错误情况
    q2, err2 := mockGateway.GetQuestionById(2)
    if err2 == nil || err2.Error() != expectedError.Error() || q2 != (mypackage.Question{}) {
        t.Errorf("Expected ({}, %v), got (%v, %v)", expectedError, q2, err2)
    }
}