统一会员门禁逻辑并补权限分支测试

internal/smoke/service/membership_gate_test.go

@@ -0,0 +1,104 @@
+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)
+	}
+}