tercul-backend/internal/data/sql/city_repository_test.go
google-labs-jules[bot] 89505b407b feat: Add unit tests for models, repositories, and services
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.
2025-09-07 11:42:30 +00:00

67 lines
2.2 KiB
Go

package sql_test
import (
"context"
"database/sql"
"regexp"
"testing"
repo "tercul/internal/data/sql"
"tercul/internal/domain"
"time"
"github.com/DATA-DOG/go-sqlmock"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestNewCityRepository(t *testing.T) {
db, _, err := newMockDb()
require.NoError(t, err)
repo := repo.NewCityRepository(db)
assert.NotNil(t, repo)
}
func TestCityRepository_ListByCountryID(t *testing.T) {
t.Run("should return cities for a given country id", func(t *testing.T) {
db, mock, err := newMockDb()
require.NoError(t, err)
repo := repo.NewCityRepository(db)
countryID := uint(1)
expectedCities := []domain.City{
{TranslatableModel: domain.TranslatableModel{BaseModel: domain.BaseModel{ID: 1, CreatedAt: time.Now(), UpdatedAt: time.Now()}}, Name: "City 1", CountryID: countryID},
{TranslatableModel: domain.TranslatableModel{BaseModel: domain.BaseModel{ID: 2, CreatedAt: time.Now(), UpdatedAt: time.Now()}}, Name: "City 2", CountryID: countryID},
}
rows := sqlmock.NewRows([]string{"id", "created_at", "updated_at", "name", "country_id"}).
AddRow(expectedCities[0].ID, expectedCities[0].CreatedAt, expectedCities[0].UpdatedAt, expectedCities[0].Name, expectedCities[0].CountryID).
AddRow(expectedCities[1].ID, expectedCities[1].CreatedAt, expectedCities[1].UpdatedAt, expectedCities[1].Name, expectedCities[1].CountryID)
mock.ExpectQuery(regexp.QuoteMeta(`SELECT * FROM "cities" WHERE country_id = $1`)).
WithArgs(countryID).
WillReturnRows(rows)
cities, err := repo.ListByCountryID(context.Background(), countryID)
require.NoError(t, err)
assert.Equal(t, expectedCities, cities)
assert.NoError(t, mock.ExpectationsWereMet())
})
t.Run("should return error if query fails", func(t *testing.T) {
db, mock, err := newMockDb()
require.NoError(t, err)
repo := repo.NewCityRepository(db)
countryID := uint(1)
mock.ExpectQuery(regexp.QuoteMeta(`SELECT * FROM "cities" WHERE country_id = $1`)).
WithArgs(countryID).
WillReturnError(sql.ErrNoRows)
cities, err := repo.ListByCountryID(context.Background(), countryID)
require.Error(t, err)
assert.Nil(t, cities)
assert.NoError(t, mock.ExpectationsWereMet())
})
}