Go mock 是一個模擬框架,允許開發人員為我們的介面建立模擬結構並定義其行為。
例如,它可以幫助我們產生 Reponsitory 的模擬實現,並根據某些輸入定義預期輸出。
原始倉庫:https://github.com/golang/mock
現在它不再維護了,所以我們應該使用 https://github.com/uber-go/mock 來代替。
go install go.uber.org/mock/mockgen@latest
依照本文檔了解如何使用mockgen CLI:
https://github.com/uber-go/mock?tab=readme-ov-file#running-mockgen
假設你有一個 IUserRepo 接口,其中包含一些方法:
ports.go
package user type IUserRepo interface { GetUserByID(id int) (*User, error) Insert(user User) error Update(id int, user User) error }
domain.go
package user type User struct { ID int Name string }
service.go
package user import "fmt" type UserService struct { repo IUserRepo } var invalidUserIDError = fmt.Errorf("invalid user id") func (u *UserService) Upsert(user User) error { if user.ID <= 0 { return invalidUserIDError } existingUser, err := u.repo.GetUserByID(user.ID) if err != nil { return err } if existingUser == nil { return u.repo.Insert(user) } return u.repo.Update(user.ID, user) }
1。運行mockgen產生模擬實例
go run go.uber.org/mock/mockgen@latest -source=interface.go -destination=mock.go -package=user
2。指定期望值
service_test.go
package user import ( "go.uber.org/mock/gomock" "github.com/stretchr/testify/assert" "testing" ) func TestUpsertUser(t *testing.T) { mockCtl := gomock.NewController(t) defer mockCtl.Finish() tests := []struct { name string user User specifyFunctionCalls func(mock *MockIUserRepo) expectedError error }{ { user: User{ID: 1, Name: "User 1"}, name: "Should insert", specifyFunctionCalls: func(mockRepo *MockIUserRepo) { mockRepo.EXPECT().GetUserByID(1).Return(nil, nil).Times(1) mockRepo.EXPECT().Insert(User{ID: 1, Name: "User 1"}).Return(nil).Times(1) }, }, { name: "User existed - Should update", user: User{ID: 1, Name: "New User Name"}, specifyFunctionCalls: func(mockRepo *MockIUserRepo) { mockRepo.EXPECT().GetUserByID(1).Return(&User{ID: 1, Name: "User 1"}, nil).Times(1) mockRepo.EXPECT().Update(1, User{ID: 1, Name: "New User Name"}).Return(nil).Times(1) }, }, { expectedError: invalidUserIDError, }, } for _, test := range tests { t.Run(test.name, func(t *testing.T) { mockRepo := NewMockIUserRepo(mockCtl) if test.specifyFunctionCalls != nil { test.specifyFunctionCalls(mockRepo) } userService := UserService{repo: mockRepo} err := userService.Upsert(test.user) assert.Equal(t, test.expectedError, err) }) } }
在上面的測試文件中:
https://github.com/huantt/gomock-demo
以上是去模擬的詳細內容。更多資訊請關注PHP中文網其他相關文章!