对导入的包进行单元测试

Question

我目前正在为一个项目创建单元测试，我需要测试一个没有实体接口的依赖注入的函数。下面是代码

func (u UserUsecase) CreateUser(ctx context.Context, p request.StoreUser) (entity.User, error) {
user := &entity.User{
Username:  p.Username,
Email:     p.Email,
Phone:     p.Phone,
Password:  p.Password,
CompanyID: p.CompanyID,
}
err := user.Create(ctx, u.userRepo)
if err != nil {
return entity.User{}, err
}
return *user, nil
}

在user.Create(ctx, u.userRepo)中有一个函数调用另一个函数，所以当我创建一个测试函数时，我一直被卡住并获得无效的内存地址或空指针。以下是user.Create(ctx, u.userRepo)

中的代码

func (u *User) Create(ctx context.Context, ur UserStoreRepo) error {
u.UUID = uuid.New()
u.Status = constanta.ACTIVE
pass, err := helpers.Bcrypting(u.Password)
if err != nil {
return err
}
u.Password = pass
err = ur.Store(ctx, u)
return err
}

覆盖user.Create(ctx, u.userRepo)以便测试CreateUser(ctx context.Context, p request.StoreUser)的最佳方法是什么?

Answer 1

使用接口并使用mock结构体实现它来测试

type User interface {
CreateUser(ctx context.Context, p request.StoreUser) (entity.User, error)
}
type user struct {
//... user struct 
}
func (u *user) CreateUser(ctx context.Context, p request.StoreUser) (entity.User, error) {
//... real code goes here
}

然后创建一个mock

type mockUser struct {
result entity.User
err error
}
func (u *mockUser) CreateUser(ctx context.Context, p request.StoreUser) (entity.User, error) {
return u.result, u.err
}

然后创建一个像 这样的测试

func TestError(t *testing.T) {
errMockUser := &mockUser{nil,erorrs.New("error")}
result, err := errMockUser.CreateUser(context.Background(), 
request.StoreUser{})
assert.Nil(t, result)
assert.Equal(t, errors.New("error"), err)
}