hello-dd-code
105 lines
2.5 KiB
Go
105 lines
2.5 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
)
|
|
|
|
func newMockGormDB(t *testing.T) (*gorm.DB, sqlmock.Sqlmock, func()) {
|
|
t.Helper()
|
|
|
|
sqlDB, mock, err := sqlmock.New()
|
|
if err != nil {
|
|
t.Fatalf("sqlmock.New: %v", err)
|
|
}
|
|
|
|
gdb, err := gorm.Open(mysql.New(mysql.Config{
|
|
Conn: sqlDB,
|
|
SkipInitializeWithVersion: true,
|
|
}), &gorm.Config{
|
|
Logger: logger.Default.LogMode(logger.Silent),
|
|
})
|
|
if err != nil {
|
|
_ = sqlDB.Close()
|
|
t.Fatalf("gorm.Open: %v", err)
|
|
}
|
|
|
|
cleanup := func() {
|
|
_ = sqlDB.Close()
|
|
}
|
|
return gdb, mock, cleanup
|
|
}
|
|
|
|
func TestHasActiveMembership(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
now := time.Date(2026, 2, 28, 16, 0, 0, 0, time.Local)
|
|
db, mock, cleanup := newMockGormDB(t)
|
|
defer cleanup()
|
|
|
|
mock.ExpectQuery("SELECT count\\(\\*\\) FROM `user_memberships`").
|
|
WithArgs(uint(100), uint(200), "active", now).
|
|
WillReturnRows(sqlmock.NewRows([]string{"count"}).AddRow(1))
|
|
|
|
ok, err := hasActiveMembership(context.Background(), db, 100, 200, now)
|
|
if err != nil {
|
|
t.Fatalf("hasActiveMembership: %v", err)
|
|
}
|
|
if !ok {
|
|
t.Fatalf("hasActiveMembership got=false, want=true")
|
|
}
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Fatalf("unmet expectations: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestHasActiveMembershipNotFound(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
now := time.Date(2026, 2, 28, 16, 0, 0, 0, time.Local)
|
|
db, mock, cleanup := newMockGormDB(t)
|
|
defer cleanup()
|
|
|
|
mock.ExpectQuery("SELECT count\\(\\*\\) FROM `user_memberships`").
|
|
WithArgs(uint(101), uint(201), "active", now).
|
|
WillReturnRows(sqlmock.NewRows([]string{"count"}).AddRow(0))
|
|
|
|
ok, err := hasActiveMembership(context.Background(), db, 101, 201, now)
|
|
if err != nil {
|
|
t.Fatalf("hasActiveMembership: %v", err)
|
|
}
|
|
if ok {
|
|
t.Fatalf("hasActiveMembership got=true, want=false")
|
|
}
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Fatalf("unmet expectations: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestHasActiveMembershipDBError(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
now := time.Date(2026, 2, 28, 16, 0, 0, 0, time.Local)
|
|
db, mock, cleanup := newMockGormDB(t)
|
|
defer cleanup()
|
|
|
|
mock.ExpectQuery("SELECT count\\(\\*\\) FROM `user_memberships`").
|
|
WithArgs(uint(102), uint(202), "active", now).
|
|
WillReturnError(errors.New("db unavailable"))
|
|
|
|
_, err := hasActiveMembership(context.Background(), db, 102, 202, now)
|
|
if err == nil {
|
|
t.Fatalf("expected error when query fails")
|
|
}
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Fatalf("unmet expectations: %v", err)
|
|
}
|
|
}
|