mirror of
https://github.com/SamyRai/tercul-backend.git
synced 2025-12-27 05:11: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.
105 lines
3.2 KiB
Go
105 lines
3.2 KiB
Go
package sql_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"tercul/internal/data/sql"
|
|
"tercul/internal/domain"
|
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
"github.com/stretchr/testify/suite"
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type CollectionRepositoryTestSuite struct {
|
|
suite.Suite
|
|
db *gorm.DB
|
|
mock sqlmock.Sqlmock
|
|
repo domain.CollectionRepository
|
|
}
|
|
|
|
func (s *CollectionRepositoryTestSuite) SetupTest() {
|
|
db, mock, err := sqlmock.New()
|
|
s.Require().NoError(err)
|
|
|
|
gormDB, err := gorm.Open(postgres.New(postgres.Config{Conn: db}), &gorm.Config{})
|
|
s.Require().NoError(err)
|
|
|
|
s.db = gormDB
|
|
s.mock = mock
|
|
s.repo = sql.NewCollectionRepository(s.db)
|
|
}
|
|
|
|
func (s *CollectionRepositoryTestSuite) TearDownTest() {
|
|
s.Require().NoError(s.mock.ExpectationsWereMet())
|
|
}
|
|
|
|
func TestCollectionRepositoryTestSuite(t *testing.T) {
|
|
suite.Run(t, new(CollectionRepositoryTestSuite))
|
|
}
|
|
|
|
func (s *CollectionRepositoryTestSuite) TestListByUserID() {
|
|
userID := uint(1)
|
|
rows := sqlmock.NewRows([]string{"id", "user_id"}).
|
|
AddRow(1, userID).
|
|
AddRow(2, userID)
|
|
|
|
s.mock.ExpectQuery(`SELECT \* FROM "collections" WHERE user_id = \$1`).
|
|
WithArgs(userID).
|
|
WillReturnRows(rows)
|
|
|
|
collections, err := s.repo.ListByUserID(context.Background(), userID)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(collections, 2)
|
|
}
|
|
|
|
func (s *CollectionRepositoryTestSuite) TestAddWorkToCollection() {
|
|
collectionID, workID := uint(1), uint(1)
|
|
s.mock.ExpectExec(`INSERT INTO collection_works \(collection_id, work_id\) VALUES \(\$1, \$2\) ON CONFLICT DO NOTHING`).
|
|
WithArgs(collectionID, workID).
|
|
WillReturnResult(sqlmock.NewResult(1, 1))
|
|
|
|
err := s.repo.AddWorkToCollection(context.Background(), collectionID, workID)
|
|
s.Require().NoError(err)
|
|
}
|
|
|
|
func (s *CollectionRepositoryTestSuite) TestRemoveWorkFromCollection() {
|
|
collectionID, workID := uint(1), uint(1)
|
|
s.mock.ExpectExec(`DELETE FROM collection_works WHERE collection_id = \$1 AND work_id = \$2`).
|
|
WithArgs(collectionID, workID).
|
|
WillReturnResult(sqlmock.NewResult(1, 1))
|
|
|
|
err := s.repo.RemoveWorkFromCollection(context.Background(), collectionID, workID)
|
|
s.Require().NoError(err)
|
|
}
|
|
|
|
func (s *CollectionRepositoryTestSuite) TestListPublic() {
|
|
rows := sqlmock.NewRows([]string{"id", "is_public"}).
|
|
AddRow(1, true).
|
|
AddRow(2, true)
|
|
|
|
s.mock.ExpectQuery(`SELECT \* FROM "collections" WHERE is_public = \$1`).
|
|
WithArgs(true).
|
|
WillReturnRows(rows)
|
|
|
|
collections, err := s.repo.ListPublic(context.Background())
|
|
s.Require().NoError(err)
|
|
s.Require().Len(collections, 2)
|
|
}
|
|
|
|
func (s *CollectionRepositoryTestSuite) TestListByWorkID() {
|
|
workID := uint(1)
|
|
rows := sqlmock.NewRows([]string{"id"}).
|
|
AddRow(1).
|
|
AddRow(2)
|
|
|
|
s.mock.ExpectQuery(`SELECT "collections"\."id","collections"\."created_at","collections"\."updated_at","collections"\."language","collections"\."slug","collections"\."name","collections"\."description","collections"\."user_id","collections"\."is_public","collections"\."cover_image_url" FROM "collections" JOIN collection_works ON collection_works\.collection_id = collections\.id WHERE collection_works\.work_id = \$1`).
|
|
WithArgs(workID).
|
|
WillReturnRows(rows)
|
|
|
|
collections, err := s.repo.ListByWorkID(context.Background(), workID)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(collections, 2)
|
|
}
|