导入包的单元测试

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

使用接口并使用模拟结构实现它来测试

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
}

然后创建一个模拟

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)
}

导入包的单元测试

Unit testing on imported package

unit-testing

dependency-injection

mocking

stub

go