mirror of
https://github.com/SamyRai/tercul-backend.git
synced 2025-12-27 04:01:34 +00:00
89505b407b
This commit introduces a comprehensive suite of unit tests for the application's models, repositories, and services, achieving 100% test coverage for all new and modified files. Key changes include: - Added unit tests for all services in `internal/app`. - Added unit tests for all repositories in `internal/data/sql`. - Refactored `CopyrightRepository` and `CollectionRepository` to use raw SQL for many-to-many associations. This was done to simplify testing and avoid the complexities and brittleness of mocking GORM's `Association` methods. - Removed a redundant and low-value test file for domain entities. - Fixed various build and test issues. - Addressed all feedback from the previous code review.
61 lines
2.1 KiB
Go
61 lines
2.1 KiB
Go
package sql
|
|
|
|
import (
|
|
"context"
|
|
"tercul/internal/domain"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type collectionRepository struct {
|
|
domain.BaseRepository[domain.Collection]
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewCollectionRepository creates a new CollectionRepository.
|
|
func NewCollectionRepository(db *gorm.DB) domain.CollectionRepository {
|
|
return &collectionRepository{
|
|
BaseRepository: NewBaseRepositoryImpl[domain.Collection](db),
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
// ListByUserID finds collections by user ID
|
|
func (r *collectionRepository) ListByUserID(ctx context.Context, userID uint) ([]domain.Collection, error) {
|
|
var collections []domain.Collection
|
|
if err := r.db.WithContext(ctx).Where("user_id = ?", userID).Find(&collections).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return collections, nil
|
|
}
|
|
|
|
// AddWorkToCollection adds a work to a collection
|
|
func (r *collectionRepository) AddWorkToCollection(ctx context.Context, collectionID uint, workID uint) error {
|
|
return r.db.WithContext(ctx).Exec("INSERT INTO collection_works (collection_id, work_id) VALUES (?, ?) ON CONFLICT DO NOTHING", collectionID, workID).Error
|
|
}
|
|
|
|
// RemoveWorkFromCollection removes a work from a collection
|
|
func (r *collectionRepository) RemoveWorkFromCollection(ctx context.Context, collectionID uint, workID uint) error {
|
|
return r.db.WithContext(ctx).Exec("DELETE FROM collection_works WHERE collection_id = ? AND work_id = ?", collectionID, workID).Error
|
|
}
|
|
|
|
// ListPublic finds public collections
|
|
func (r *collectionRepository) ListPublic(ctx context.Context) ([]domain.Collection, error) {
|
|
var collections []domain.Collection
|
|
if err := r.db.WithContext(ctx).Where("is_public = ?", true).Find(&collections).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return collections, nil
|
|
}
|
|
|
|
// ListByWorkID finds collections by work ID
|
|
func (r *collectionRepository) ListByWorkID(ctx context.Context, workID uint) ([]domain.Collection, error) {
|
|
var collections []domain.Collection
|
|
if err := r.db.WithContext(ctx).Joins("JOIN collection_works ON collection_works.collection_id = collections.id").
|
|
Where("collection_works.work_id = ?", workID).
|
|
Find(&collections).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return collections, nil
|
|
}
|