mirror of
https://github.com/SamyRai/tercul-backend.git
synced 2025-12-27 05:11:34 +00:00
411 lines
11 KiB
Go
411 lines
11 KiB
Go
package services_test
|
|
|
|
import (
|
|
"context"
|
|
"tercul/internal/models"
|
|
"tercul/internal/repositories"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/mock"
|
|
"github.com/stretchr/testify/suite"
|
|
"gorm.io/gorm"
|
|
"tercul/services"
|
|
)
|
|
|
|
// MockWorkRepository is a mock implementation of the WorkRepository interface
|
|
type MockWorkRepository struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (m *MockWorkRepository) Create(ctx context.Context, entity *models.Work) error {
|
|
args := m.Called(ctx, entity)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func (m *MockWorkRepository) CreateInTx(ctx context.Context, _ *gorm.DB, _ *models.Work) error {
|
|
return nil
|
|
}
|
|
|
|
func (m *MockWorkRepository) GetByID(ctx context.Context, id uint) (*models.Work, error) {
|
|
args := m.Called(ctx, id)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(*models.Work), args.Error(1)
|
|
}
|
|
|
|
func (m *MockWorkRepository) GetByIDWithOptions(ctx context.Context, _ uint, _ *repositories.QueryOptions) (*models.Work, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (m *MockWorkRepository) Update(ctx context.Context, entity *models.Work) error {
|
|
args := m.Called(ctx, entity)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func (m *MockWorkRepository) UpdateInTx(ctx context.Context, _ *gorm.DB, _ *models.Work) error {
|
|
return nil
|
|
}
|
|
|
|
func (m *MockWorkRepository) Delete(ctx context.Context, id uint) error {
|
|
args := m.Called(ctx, id)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func (m *MockWorkRepository) DeleteInTx(ctx context.Context, _ *gorm.DB, _ uint) error { return nil }
|
|
|
|
func (m *MockWorkRepository) List(ctx context.Context, page, pageSize int) (*repositories.PaginatedResult[models.Work], error) {
|
|
args := m.Called(ctx, page, pageSize)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(*repositories.PaginatedResult[models.Work]), args.Error(1)
|
|
}
|
|
|
|
func (m *MockWorkRepository) ListWithOptions(ctx context.Context, _ *repositories.QueryOptions) ([]models.Work, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (m *MockWorkRepository) ListAll(ctx context.Context) ([]models.Work, error) {
|
|
args := m.Called(ctx)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).([]models.Work), args.Error(1)
|
|
}
|
|
|
|
func (m *MockWorkRepository) GetAllForSync(ctx context.Context, batchSize, offset int) ([]models.Work, error) {
|
|
args := m.Called(ctx, batchSize, offset)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).([]models.Work), args.Error(1)
|
|
}
|
|
|
|
func (m *MockWorkRepository) Count(ctx context.Context) (int64, error) {
|
|
args := m.Called(ctx)
|
|
return args.Get(0).(int64), args.Error(1)
|
|
}
|
|
|
|
func (m *MockWorkRepository) CountWithOptions(ctx context.Context, _ *repositories.QueryOptions) (int64, error) {
|
|
return 0, nil
|
|
}
|
|
|
|
func (m *MockWorkRepository) FindWithPreload(ctx context.Context, preloads []string, id uint) (*models.Work, error) {
|
|
args := m.Called(ctx, preloads, id)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(*models.Work), args.Error(1)
|
|
}
|
|
|
|
func (m *MockWorkRepository) Exists(ctx context.Context, _ uint) (bool, error) { return false, nil }
|
|
func (m *MockWorkRepository) BeginTx(ctx context.Context) (*gorm.DB, error) { return nil, nil }
|
|
func (m *MockWorkRepository) WithTx(ctx context.Context, fn func(tx *gorm.DB) error) error {
|
|
return nil
|
|
}
|
|
|
|
func (m *MockWorkRepository) FindByTitle(ctx context.Context, title string) ([]models.Work, error) {
|
|
args := m.Called(ctx, title)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).([]models.Work), args.Error(1)
|
|
}
|
|
|
|
func (m *MockWorkRepository) FindByAuthor(ctx context.Context, authorID uint) ([]models.Work, error) {
|
|
args := m.Called(ctx, authorID)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).([]models.Work), args.Error(1)
|
|
}
|
|
|
|
func (m *MockWorkRepository) FindByCategory(ctx context.Context, categoryID uint) ([]models.Work, error) {
|
|
args := m.Called(ctx, categoryID)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).([]models.Work), args.Error(1)
|
|
}
|
|
|
|
func (m *MockWorkRepository) FindByLanguage(ctx context.Context, language string, page, pageSize int) (*repositories.PaginatedResult[models.Work], error) {
|
|
args := m.Called(ctx, language, page, pageSize)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(*repositories.PaginatedResult[models.Work]), args.Error(1)
|
|
}
|
|
|
|
func (m *MockWorkRepository) GetWithTranslations(ctx context.Context, id uint) (*models.Work, error) {
|
|
args := m.Called(ctx, id)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(*models.Work), args.Error(1)
|
|
}
|
|
|
|
func (m *MockWorkRepository) ListWithTranslations(ctx context.Context, page, pageSize int) (*repositories.PaginatedResult[models.Work], error) {
|
|
args := m.Called(ctx, page, pageSize)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(*repositories.PaginatedResult[models.Work]), args.Error(1)
|
|
}
|
|
|
|
// MockAnalyzer is a mock implementation of the Analyzer interface
|
|
type MockAnalyzer struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (m *MockAnalyzer) AnalyzeWork(ctx context.Context, workID uint) error {
|
|
args := m.Called(ctx, workID)
|
|
return args.Error(0)
|
|
}
|
|
|
|
// WorkServiceSuite is a test suite for the WorkService
|
|
type WorkServiceSuite struct {
|
|
suite.Suite
|
|
mockRepo *MockWorkRepository
|
|
mockAnalyzer *MockAnalyzer
|
|
service services.WorkService
|
|
}
|
|
|
|
// SetupTest sets up each test
|
|
func (s *WorkServiceSuite) SetupTest() {
|
|
s.mockRepo = new(MockWorkRepository)
|
|
s.mockAnalyzer = new(MockAnalyzer)
|
|
s.service = services.NewWorkService(s.mockRepo, s.mockAnalyzer)
|
|
}
|
|
|
|
// TestCreateWork tests the CreateWork method
|
|
func (s *WorkServiceSuite) TestCreateWork() {
|
|
// Setup
|
|
work := &models.Work{
|
|
Title: "Test Work",
|
|
Description: "Test description",
|
|
Status: "published",
|
|
}
|
|
work.Language = "en"
|
|
|
|
// Mock repository
|
|
s.mockRepo.On("Create", mock.Anything, work).Return(nil)
|
|
|
|
// Execute
|
|
err := s.service.CreateWork(context.Background(), work)
|
|
|
|
// Assert
|
|
s.Require().NoError(err)
|
|
|
|
// Verify mocks
|
|
s.mockRepo.AssertCalled(s.T(), "Create", mock.Anything, work)
|
|
}
|
|
|
|
// TestCreateWorkWithEmptyTitle tests the CreateWork method with an empty title
|
|
func (s *WorkServiceSuite) TestCreateWorkWithEmptyTitle() {
|
|
// Setup
|
|
work := &models.Work{
|
|
Description: "Test description",
|
|
Status: "published",
|
|
}
|
|
work.Language = "en"
|
|
|
|
// Execute
|
|
err := s.service.CreateWork(context.Background(), work)
|
|
|
|
// Assert
|
|
s.Require().Error(err)
|
|
s.Contains(err.Error(), "title cannot be empty")
|
|
|
|
// Verify mocks
|
|
s.mockRepo.AssertNotCalled(s.T(), "Create")
|
|
}
|
|
|
|
// TestCreateWorkWithEmptyLanguage tests the CreateWork method with an empty language
|
|
func (s *WorkServiceSuite) TestCreateWorkWithEmptyLanguage() {
|
|
// Setup
|
|
work := &models.Work{
|
|
Title: "Test Work",
|
|
Description: "Test description",
|
|
Status: "published",
|
|
}
|
|
// Language is empty
|
|
|
|
// Execute
|
|
err := s.service.CreateWork(context.Background(), work)
|
|
|
|
// Assert
|
|
s.Require().Error(err)
|
|
s.Contains(err.Error(), "language cannot be empty")
|
|
|
|
// Verify mocks
|
|
s.mockRepo.AssertNotCalled(s.T(), "Create")
|
|
}
|
|
|
|
// TestGetWorkByID tests the GetWorkByID method
|
|
func (s *WorkServiceSuite) TestGetWorkByID() {
|
|
// Setup
|
|
id := uint(1)
|
|
expectedWork := &models.Work{
|
|
Title: "Test Work",
|
|
Description: "Test description",
|
|
Status: "published",
|
|
}
|
|
expectedWork.ID = id
|
|
expectedWork.Language = "en"
|
|
|
|
// Mock repository
|
|
s.mockRepo.On("GetByID", mock.Anything, id).Return(expectedWork, nil)
|
|
|
|
// Execute
|
|
result, err := s.service.GetWorkByID(context.Background(), id)
|
|
|
|
// Assert
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(result)
|
|
s.Equal(expectedWork.ID, result.ID)
|
|
s.Equal(expectedWork.Title, result.Title)
|
|
s.Equal(expectedWork.Language, result.Language)
|
|
|
|
// Verify mocks
|
|
s.mockRepo.AssertCalled(s.T(), "GetByID", mock.Anything, id)
|
|
}
|
|
|
|
// TestGetWorkByIDWithInvalidID tests the GetWorkByID method with an invalid ID
|
|
func (s *WorkServiceSuite) TestGetWorkByIDWithInvalidID() {
|
|
// Setup
|
|
id := uint(0) // Invalid ID
|
|
|
|
// Execute
|
|
result, err := s.service.GetWorkByID(context.Background(), id)
|
|
|
|
// Assert
|
|
s.Require().Error(err)
|
|
s.Nil(result)
|
|
s.Contains(err.Error(), "invalid work ID")
|
|
|
|
// Verify mocks
|
|
s.mockRepo.AssertNotCalled(s.T(), "GetByID")
|
|
}
|
|
|
|
// TestUpdateWork tests the UpdateWork method
|
|
func (s *WorkServiceSuite) TestUpdateWork() {
|
|
// Setup
|
|
work := &models.Work{
|
|
Title: "Test Work",
|
|
Description: "Test description",
|
|
Status: "published",
|
|
}
|
|
work.ID = 1
|
|
work.Language = "en"
|
|
|
|
// Mock repository
|
|
s.mockRepo.On("Update", mock.Anything, work).Return(nil)
|
|
|
|
// Execute
|
|
err := s.service.UpdateWork(context.Background(), work)
|
|
|
|
// Assert
|
|
s.Require().NoError(err)
|
|
|
|
// Verify mocks
|
|
s.mockRepo.AssertCalled(s.T(), "Update", mock.Anything, work)
|
|
}
|
|
|
|
// TestUpdateWorkWithInvalidID tests the UpdateWork method with an invalid ID
|
|
func (s *WorkServiceSuite) TestUpdateWorkWithInvalidID() {
|
|
// Setup
|
|
work := &models.Work{
|
|
Title: "Test Work",
|
|
Description: "Test description",
|
|
Status: "published",
|
|
}
|
|
work.ID = 0 // Invalid ID
|
|
work.Language = "en"
|
|
|
|
// Execute
|
|
err := s.service.UpdateWork(context.Background(), work)
|
|
|
|
// Assert
|
|
s.Require().Error(err)
|
|
s.Contains(err.Error(), "work ID cannot be zero")
|
|
|
|
// Verify mocks
|
|
s.mockRepo.AssertNotCalled(s.T(), "Update")
|
|
}
|
|
|
|
// TestDeleteWork tests the DeleteWork method
|
|
func (s *WorkServiceSuite) TestDeleteWork() {
|
|
// Setup
|
|
id := uint(1)
|
|
|
|
// Mock repository
|
|
s.mockRepo.On("Delete", mock.Anything, id).Return(nil)
|
|
|
|
// Execute
|
|
err := s.service.DeleteWork(context.Background(), id)
|
|
|
|
// Assert
|
|
s.Require().NoError(err)
|
|
|
|
// Verify mocks
|
|
s.mockRepo.AssertCalled(s.T(), "Delete", mock.Anything, id)
|
|
}
|
|
|
|
// TestDeleteWorkWithInvalidID tests the DeleteWork method with an invalid ID
|
|
func (s *WorkServiceSuite) TestDeleteWorkWithInvalidID() {
|
|
// Setup
|
|
id := uint(0) // Invalid ID
|
|
|
|
// Execute
|
|
err := s.service.DeleteWork(context.Background(), id)
|
|
|
|
// Assert
|
|
s.Require().Error(err)
|
|
s.Contains(err.Error(), "invalid work ID")
|
|
|
|
// Verify mocks
|
|
s.mockRepo.AssertNotCalled(s.T(), "Delete")
|
|
}
|
|
|
|
// TestAnalyzeWork tests the AnalyzeWork method
|
|
func (s *WorkServiceSuite) TestAnalyzeWork() {
|
|
// Setup
|
|
id := uint(1)
|
|
ctx := context.Background()
|
|
|
|
// Mock analyzer
|
|
s.mockAnalyzer.On("AnalyzeWork", ctx, id).Return(nil)
|
|
|
|
// Execute
|
|
err := s.service.AnalyzeWork(ctx, id)
|
|
|
|
// Assert
|
|
s.Require().NoError(err)
|
|
|
|
// Verify mocks
|
|
s.mockAnalyzer.AssertCalled(s.T(), "AnalyzeWork", ctx, id)
|
|
}
|
|
|
|
// TestAnalyzeWorkWithInvalidID tests the AnalyzeWork method with an invalid ID
|
|
func (s *WorkServiceSuite) TestAnalyzeWorkWithInvalidID() {
|
|
// Setup
|
|
id := uint(0) // Invalid ID
|
|
ctx := context.Background()
|
|
|
|
// Execute
|
|
err := s.service.AnalyzeWork(ctx, id)
|
|
|
|
// Assert
|
|
s.Require().Error(err)
|
|
s.Contains(err.Error(), "invalid work ID")
|
|
|
|
// Verify mocks
|
|
s.mockAnalyzer.AssertNotCalled(s.T(), "AnalyzeWork")
|
|
}
|
|
|
|
// TestWorkServiceSuite runs the test suite
|
|
func TestWorkServiceSuite(t *testing.T) {
|
|
suite.Run(t, new(WorkServiceSuite))
|
|
}
|