Go mockgen，模拟函数未调用

Question

我使用Go 1.19在windows机器上有8个核心，操作系统是Windows 10 Pro。我使用mockgen工具生成模拟。调试测试时，我看到在执行EXPECT()函数时记录了模拟的方法。将调用模拟函数，但模拟函数上的“缺失调用”将导致测试失败。我看不出我做错了什么，谁能指出出来吗？

Directory Structure :
cmd
 configure.go
 configure_test.go
mocks
  mock_validator.go
validator
  validator.go
user
  user.go 
go.mod
main.go 

* Contents of main.go
package main
import (
                "localdev/mockexample/cmd"
)
func main() {
                cmd.Configure()
}
* Contents of configure.go
package cmd
import (
                "fmt"
                "localdev/mockexample/user"
                "os"
                "localdev/mockexample/validator"
)
var (
                name, password string
)
func Configure() {
                name := os.Args[1]
                password := os.Args[2]
                user, err := validate(validator.NewValidator(name, password))
                if err != nil {
                                fmt.Printf("%v\n", err)
                                return
                }
                fmt.Printf("Credentials are valid. Welcome: %s %s\n", user.FirstName, user.LastName)
}
func validate(validator validator.Validator) (*user.Data, error) {
                user, err := validator.ValidateUser()
                if err != nil {
                                return nil, fmt.Errorf("some thing went wrong. %v", err)
                }
                return user, nil
}
* Contents of validator.go
package validator
import (
                "fmt"
                "localdev/mockexample/user"
)
//go:generate mockgen -destination=../mocks/mock_validator.go -package=mocks localdev/mockexample/validator Validator
type Validator interface {
                ValidateUser() (*user.Data, error)
}
type ValidationRequest struct {
                Command  string
                Name     string
                Password string
}
func (vr ValidationRequest) ValidateUser() (*user.Data, error) {
                if vr.Name == "bob" && vr.Password == "1234" {
                                return &user.Data{UserID: "123", UserName: "bsmith", FirstName: "Bob", LastName: "Smith"}, nil
                }
                return nil, fmt.Errorf("invalid credentials")
}
func NewValidator(name string, password string) Validator {
                return &ValidationRequest{Name: name, Password: password}
}
* Contents of user.go
package user
type Data struct {
                UserID    string `json:"user_id"`
                UserName  string `json:"user_name"`
                FirstName string `json:"first_name"`
                LastName  string `json:"last_name"`
}
* Contents of configure_test.go
package cmd
import (
                "localdev/mockexample/mocks"
                "localdev/mockexample/user"
                "os"
                "testing"
 
                "github.com/golang/mock/gomock"
)
func TestConfigure(t *testing.T) {
                t.Run("ConfigureWithMock", func(t *testing.T) {
                                os.Args[1] = "bob"
                                os.Args[2] = "1234"
 
                                ctrl := gomock.NewController(t)
                                mockValidator := mocks.NewMockValidator(ctrl)
                                //mockValidator.EXPECT().ValidateUser().AnyTimes() // zero more calls, so this will also pass.
                                userData := user.Data{UserID: "testId"}
                                mockValidator.EXPECT().ValidateUser().Return(&userData, nil).Times(1) //(gomock.Any(), gomock.Any()) //(&userData, nil)
                                Configure()
                })
}
Contents of generated mock
// Code generated by MockGen. DO NOT EDIT.
// Source: localdev/mockexample/validator (interfaces: Validator)
// Package mocks is a generated GoMock package.
package mocks
import (
                user "localdev/mockexample/user"
                reflect "reflect"
                gomock "github.com/golang/mock/gomock"
)
// MockValidator is a mock of Validator interface.
type MockValidator struct {
                ctrl     *gomock.Controller
                recorder *MockValidatorMockRecorder
}
// MockValidatorMockRecorder is the mock recorder for MockValidator.
type MockValidatorMockRecorder struct {
                mock *MockValidator
}
// NewMockValidator creates a new mock instance.
func NewMockValidator(ctrl *gomock.Controller) *MockValidator {
                mock := &MockValidator{ctrl: ctrl}
                mock.recorder = &MockValidatorMockRecorder{mock}
                return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockValidator) EXPECT() *MockValidatorMockRecorder {
                return m.recorder
}
// ValidateUser mocks base method.
func (m *MockValidator) ValidateUser() (*user.Data, error) {
                m.ctrl.T.Helper()
                ret := m.ctrl.Call(m, "ValidateUser")
                ret0, _ := ret[0].(*user.Data)
                ret1, _ := ret[1].(error)
                return ret0, ret1
}
// ValidateUser indicates an expected call of ValidateUser.
func (mr *MockValidatorMockRecorder) ValidateUser() *gomock.Call {
                mr.mock.ctrl.T.Helper()
                return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateUser", reflect.TypeOf((*MockValidator)(nil).ValidateUser))
}

Answer 1

根本问题是，函数Configure从不使用模拟结构，因此您将得到一个missing call(s) to *mocks.MockValidator.ValidateUser()错误。

在文件configure_test.go中，根本不使用mockValidator。为了被Configure函数调用，必须有某种类型的模拟注入。

您可以进行以下更改来修复测试，作为我所指的注入的一个例子。这并不是说这是最好的方法，但我试图对您的代码进行尽可能少的更改。

configure_test.go

func TestConfigure(t *testing.T) {
    t.Run("ConfigureWithMock", func(t *testing.T) {
        os.Args[1] = "bob"
        os.Args[2] = "1234"

        ctrl := gomock.NewController(t)
        mockValidator := mocks.NewMockValidator(ctrl)
        //mockValidator.EXPECT().ValidateUser().AnyTimes() // zero more calls, so this will also pass.
        userData := user.Data{UserID: "testId"}
        mockValidator.
            EXPECT().
            ValidateUser("bob", "1234").
            Return(&userData, nil).
            Times(1) //(gomock.Any(), gomock.Any()) //(&userData, nil)
        Configure(mockValidator)
    })
}

configure.go

func Configure(v validator.Validator) {
    name := os.Args[1]
    password := os.Args[2]
    user, err := v.ValidateUser(name, password)
    if err != nil {
        fmt.Printf("some thing went wrong. %v\n", err)
        return
    }
    fmt.Printf("Credentials are valid. Welcome: %s %s\n", user.FirstName, user.LastName)
}

validator.go

type Validator interface {
    ValidateUser(name, password string) (*user.Data, error)
}
type ValidationRequest struct {
    Command string
    // Name     string
    // Password string
}

func (vr ValidationRequest) ValidateUser(name, password string) (*user.Data, error) {
    if name == "bob" && password == "1234" {
        return &user.Data{UserID: "123", UserName: "bsmith", FirstName: "Bob", LastName: "Smith"}, nil
    }
    return nil, fmt.Errorf("invalid credentials")
}

func NewValidator() Validator {
    return &ValidationRequest{}
}

考虑到您需要再次生成模拟。希望这有助于您理解模拟测试。