Elton Minetto
130 lines
3.3 KiB
Go
130 lines
3.3 KiB
Go
package loan
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/eminetto/clean-architecture-go-v2/entity"
|
|
|
|
bmock "github.com/eminetto/clean-architecture-go-v2/usecase/book/mock"
|
|
umock "github.com/eminetto/clean-architecture-go-v2/usecase/user/mock"
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_Borrow(t *testing.T) {
|
|
controller := gomock.NewController(t)
|
|
defer controller.Finish()
|
|
uMock := umock.NewMockUseCase(controller)
|
|
bMock := bmock.NewMockUseCase(controller)
|
|
uc := NewService(uMock, bMock)
|
|
t.Run("user not found", func(t *testing.T) {
|
|
u := &entity.User{
|
|
ID: entity.NewID(),
|
|
}
|
|
b := &entity.Book{
|
|
ID: entity.NewID(),
|
|
}
|
|
uMock.EXPECT().GetUser(u.ID).Return(nil, entity.ErrNotFound)
|
|
err := uc.Borrow(u, b)
|
|
assert.Equal(t, entity.ErrNotFound, err)
|
|
})
|
|
t.Run("book not found", func(t *testing.T) {
|
|
u := &entity.User{
|
|
ID: entity.NewID(),
|
|
}
|
|
b := &entity.Book{
|
|
ID: entity.NewID(),
|
|
}
|
|
uMock.EXPECT().GetUser(u.ID).Return(u, nil)
|
|
bMock.EXPECT().GetBook(b.ID).Return(nil, entity.ErrNotFound)
|
|
err := uc.Borrow(u, b)
|
|
assert.Equal(t, entity.ErrNotFound, err)
|
|
})
|
|
t.Run("not enough books to borrow", func(t *testing.T) {
|
|
u := &entity.User{
|
|
ID: entity.NewID(),
|
|
}
|
|
b := &entity.Book{
|
|
ID: entity.NewID(),
|
|
}
|
|
b.Quantity = 0
|
|
uMock.EXPECT().GetUser(u.ID).Return(u, nil)
|
|
bMock.EXPECT().GetBook(b.ID).Return(b, nil)
|
|
err := uc.Borrow(u, b)
|
|
assert.Equal(t, entity.ErrNotEnoughBooks, err)
|
|
})
|
|
t.Run("book already borrowed", func(t *testing.T) {
|
|
u := &entity.User{
|
|
ID: entity.NewID(),
|
|
}
|
|
b := &entity.Book{
|
|
ID: entity.NewID(),
|
|
}
|
|
u.AddBook(b.ID)
|
|
b.Quantity = 1
|
|
uMock.EXPECT().GetUser(u.ID).Return(u, nil)
|
|
bMock.EXPECT().GetBook(b.ID).Return(b, nil)
|
|
err := uc.Borrow(u, b)
|
|
assert.Equal(t, entity.ErrBookAlreadyBorrowed, err)
|
|
})
|
|
t.Run("sucess", func(t *testing.T) {
|
|
u := &entity.User{
|
|
ID: entity.NewID(),
|
|
}
|
|
b := &entity.Book{
|
|
ID: entity.NewID(),
|
|
Quantity: 10,
|
|
}
|
|
uMock.EXPECT().GetUser(u.ID).Return(u, nil)
|
|
bMock.EXPECT().GetBook(b.ID).Return(b, nil)
|
|
uMock.EXPECT().UpdateUser(u).Return(nil)
|
|
bMock.EXPECT().UpdateBook(b).Return(nil)
|
|
err := uc.Borrow(u, b)
|
|
assert.Nil(t, err)
|
|
})
|
|
}
|
|
|
|
func Test_Return(t *testing.T) {
|
|
controller := gomock.NewController(t)
|
|
defer controller.Finish()
|
|
uMock := umock.NewMockUseCase(controller)
|
|
bMock := bmock.NewMockUseCase(controller)
|
|
uc := NewService(uMock, bMock)
|
|
t.Run("book not found", func(t *testing.T) {
|
|
b := &entity.Book{
|
|
ID: entity.NewID(),
|
|
}
|
|
bMock.EXPECT().GetBook(b.ID).Return(nil, entity.ErrNotFound)
|
|
err := uc.Return(b)
|
|
assert.Equal(t, entity.ErrNotFound, err)
|
|
})
|
|
t.Run("book not borrowed", func(t *testing.T) {
|
|
u := &entity.User{
|
|
ID: entity.NewID(),
|
|
}
|
|
b := &entity.Book{
|
|
ID: entity.NewID(),
|
|
}
|
|
bMock.EXPECT().GetBook(b.ID).Return(b, nil)
|
|
uMock.EXPECT().ListUsers().Return([]*entity.User{u}, nil)
|
|
err := uc.Return(b)
|
|
assert.Equal(t, entity.ErrBookNotBorrowed, err)
|
|
})
|
|
t.Run("success", func(t *testing.T) {
|
|
u := &entity.User{
|
|
ID: entity.NewID(),
|
|
}
|
|
b := &entity.Book{
|
|
ID: entity.NewID(),
|
|
}
|
|
u.AddBook(b.ID)
|
|
bMock.EXPECT().GetBook(b.ID).Return(b, nil)
|
|
uMock.EXPECT().GetUser(u.ID).Return(u, nil)
|
|
uMock.EXPECT().ListUsers().Return([]*entity.User{u}, nil)
|
|
uMock.EXPECT().UpdateUser(u).Return(nil)
|
|
bMock.EXPECT().UpdateBook(b).Return(nil)
|
|
err := uc.Return(b)
|
|
assert.Nil(t, err)
|
|
})
|
|
}
|