如何在用例的循环中测试 Go Mock Repository
我在使用 testify 测试存储库模拟数据库时遇到问题。
我想测试为每次迭代创建记录数据库的服务/用例。这是代码：
此代码包含数据库的模拟
模拟仓库.go
package service
import(
"errors"
"github.com/stretchr/testify/mock"
)
type TestRepositoryMock struct {
Mock mock.Mock
}
func (repository *TestRepositoryMock) CreateTodo(todo *Todo) error {
arguments := repository.Mock.Called(todo)
if arguments.Get(0) == nil {
return errors.New("Error")
} else {
return nil
}
}
func (repository *TestRepositoryMock) CreateTodoDetail(todo *TodoDetail) error {
arguments := repository.Mock.Called(todo)
if arguments.Get(0) == nil {
return errors.New("Error")
} else if arguments.Get(1) == nil {
return errors.New("Error")
} else {
return nil
}
}
此用例中的逻辑是为每个项目数保存 TodoDetail(NumberOfItems variable)
用例.go
package service
import(
"strconv"
)
type UseCase interface {
SaveTodo(numberOfItems int) (string, error)
}
func NewUseCase(repo Repository) UseCase {
return &useCase{repo: repo}
}
type useCase struct {
repo Repository
}
func (uc *useCase) SaveTodo(numberOfItems int) (string, error){
todo := Todo{
Title: "this is title",
}
uc.repo.CreateTodo(&todo)
for i := 0; i < numberOfItems; i++ {
todoDetail := TodoDetail{
ID: todo.ID,
Item: "item "+ strconv.Itoa(i),
}
uc.repo.CreateTodoDetail(&todoDetail)
}
return "success", nil
}
package service
import(
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/mock"
)
var repository = &TestRepositoryMock{Mock: mock.Mock{}}
var testService = useCase{repo: repository}
func TestService_SaveTodoSuccess(t *testing.T) {
todo := Todo{
Title: "this is title",
}
todoDetail := TodoDetail{
ID: todo.ID,
Item: "item 1",
}
repository.Mock.On("CreateTodo", &todo).Return(nil)
repository.Mock.On("CreateTodoDetail", &todoDetail).Return(nil).Once()
result, err := testService.SaveTodo(3)
assert.Nil(t, err)
assert.NotNil(t, result)
}
如何测试CreateTodoDetailfor循环内部的多次调用？
当年话下1回答
-
一只名叫tom的猫
为了测试循环用例，我们应该提供大量模拟数据。与循环迭代相同。例子。如果用例有 3 个迭代循环调用存储库三次，那么我们必须提供 3 个数据模拟。代码应该是这样的：func TestService_SaveTodoSuccess(t *testing.T) { // Mock Entity Todo todo := Todo{ Title: "this is title", } // Mock Entity TodoDetail todoDetail1 := TodoDetail{ ID: todo.ID, Item: "item 0", } todoDetail2 := TodoDetail{ ID: todo.ID, Item: "item 1", } todoDetail3 := TodoDetail{ ID: todo.ID, Item: "item 2", } repository.Mock.On("CreateTodo", &todo).Return(todo)// calls 3 times "CreateTodoDetail" Repository repository.Mock.On("CreateTodoDetail", &todoDetail1).Return(todoDetail1) repository.Mock.On("CreateTodoDetail", &todoDetail2).Return(todoDetail2) repository.Mock.On("CreateTodoDetail", &todoDetail3).Return(todoDetail3) result, err := testService.SaveTodo(3) assert.Nil(t, err) assert.NotNil(t, result)}该场景在 todoDetail 上构建 3 个不同的数据以模拟CreateTodoDetail存储库的 3 次迭代
随时随地看视频慕课网APP
Go