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 중국어 웹사이트의 기타 관련 기사를 참조하세요!