tercul-backend/internal/app/work/main_test.go

package work

import (
	"context"
	"tercul/internal/domain"
)

type mockWorkRepository struct {
	domain.WorkRepository
	createFunc              func(ctx context.Context, work *domain.Work) error
	updateFunc              func(ctx context.Context, work *domain.Work) error
	deleteFunc              func(ctx context.Context, id uint) error
	getByIDFunc             func(ctx context.Context, id uint) (*domain.Work, error)
	listFunc                func(ctx context.Context, page, pageSize int) (*domain.PaginatedResult[domain.Work], error)
	getWithTranslationsFunc func(ctx context.Context, id uint) (*domain.Work, error)
	findByTitleFunc         func(ctx context.Context, title string) ([]domain.Work, error)
	findByAuthorFunc        func(ctx context.Context, authorID uint) ([]domain.Work, error)
	findByCategoryFunc      func(ctx context.Context, categoryID uint) ([]domain.Work, error)
	findByLanguageFunc      func(ctx context.Context, language string, page, pageSize int) (*domain.PaginatedResult[domain.Work], error)
	isAuthorFunc            func(ctx context.Context, workID uint, authorID uint) (bool, error)
	listByCollectionIDFunc  func(ctx context.Context, collectionID uint) ([]domain.Work, error)
}

func (m *mockWorkRepository) IsAuthor(ctx context.Context, workID uint, authorID uint) (bool, error) {
	if m.isAuthorFunc != nil {
		return m.isAuthorFunc(ctx, workID, authorID)
	}
	return false, nil
}

func (m *mockWorkRepository) Create(ctx context.Context, work *domain.Work) error {
	if m.createFunc != nil {
		return m.createFunc(ctx, work)
	}
	return nil
}
func (m *mockWorkRepository) Update(ctx context.Context, work *domain.Work) error {
	if m.updateFunc != nil {
		return m.updateFunc(ctx, work)
	}
	return nil
}
func (m *mockWorkRepository) Delete(ctx context.Context, id uint) error {
	if m.deleteFunc != nil {
		return m.deleteFunc(ctx, id)
	}
	return nil
}
func (m *mockWorkRepository) GetByID(ctx context.Context, id uint) (*domain.Work, error) {
	if m.getByIDFunc != nil {
		return m.getByIDFunc(ctx, id)
	}
	return &domain.Work{TranslatableModel: domain.TranslatableModel{BaseModel: domain.BaseModel{ID: id}}}, nil
}
func (m *mockWorkRepository) List(ctx context.Context, page, pageSize int) (*domain.PaginatedResult[domain.Work], error) {
	if m.listFunc != nil {
		return m.listFunc(ctx, page, pageSize)
	}
	return nil, nil
}

func (m *mockWorkRepository) ListByCollectionID(ctx context.Context, collectionID uint) ([]domain.Work, error) {
	if m.listByCollectionIDFunc != nil {
		return m.listByCollectionIDFunc(ctx, collectionID)
	}
	return nil, nil
}
func (m *mockWorkRepository) GetWithTranslations(ctx context.Context, id uint) (*domain.Work, error) {
	if m.getWithTranslationsFunc != nil {
		return m.getWithTranslationsFunc(ctx, id)
	}
	return nil, nil
}
func (m *mockWorkRepository) FindByTitle(ctx context.Context, title string) ([]domain.Work, error) {
	if m.findByTitleFunc != nil {
		return m.findByTitleFunc(ctx, title)
	}
	return nil, nil
}
func (m *mockWorkRepository) FindByAuthor(ctx context.Context, authorID uint) ([]domain.Work, error) {
	if m.findByAuthorFunc != nil {
		return m.findByAuthorFunc(ctx, authorID)
	}
	return nil, nil
}
func (m *mockWorkRepository) FindByCategory(ctx context.Context, categoryID uint) ([]domain.Work, error) {
	if m.findByCategoryFunc != nil {
		return m.findByCategoryFunc(ctx, categoryID)
	}
	return nil, nil
}
func (m *mockWorkRepository) FindByLanguage(ctx context.Context, language string, page, pageSize int) (*domain.PaginatedResult[domain.Work], error) {
	if m.findByLanguageFunc != nil {
		return m.findByLanguageFunc(ctx, language, page, pageSize)
	}
	return nil, nil
}

type mockSearchClient struct {
	indexWorkFunc func(ctx context.Context, work *domain.Work, pipeline string) error
}

func (m *mockSearchClient) IndexWork(ctx context.Context, work *domain.Work, pipeline string) error {
	if m.indexWorkFunc != nil {
		return m.indexWorkFunc(ctx, work, pipeline)
	}
	return nil
}