feat(*): 功能开放与页面布局 - 新增运维开关并优化后台及创作界面空间利用

backend/internal/service/feature_availability_test.go

@@ -0,0 +1,84 @@
+package service
+
+import (
+	"errors"
+	"testing"
+
+	"infinite-canvas/backend/internal/model"
+	"infinite-canvas/backend/internal/repository"
+
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+)
+
+func TestFeatureAvailabilityDefaultsToEnabled(t *testing.T) {
+	svc, _ := newFeatureAvailabilityTestService(t)
+
+	setting, err := svc.FeatureAvailability()
+	if err != nil {
+		t.Fatal(err)
+	}
+	if setting.Configured || !setting.ShortDramaEnabled || !setting.TaskCenterEnabled || !setting.CreditsEnabled {
+		t.Fatalf("FeatureAvailability() = %#v", setting)
+	}
+}
+
+func TestUpdateFeatureAvailabilityPersistsAndAudits(t *testing.T) {
+	svc, db := newFeatureAvailabilityTestService(t)
+	actor := &model.User{ID: "admin-1", Role: model.UserRoleAdmin}
+	want := FeatureAvailability{ShortDramaEnabled: false, TaskCenterEnabled: true, CreditsEnabled: false}
+
+	setting, err := svc.UpdateFeatureAvailability(actor, want)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !setting.Configured || setting.ShortDramaEnabled || !setting.TaskCenterEnabled || setting.CreditsEnabled {
+		t.Fatalf("UpdateFeatureAvailability() = %#v", setting)
+	}
+	if err := svc.RequireFeature(FeatureShortDrama); err == nil {
+		t.Fatal("RequireFeature(shortDrama) error = nil")
+	} else {
+		var authErr *AuthError
+		if !errors.As(err, &authErr) || authErr.Status != 403 {
+			t.Fatalf("RequireFeature(shortDrama) error = %#v", err)
+		}
+	}
+	var auditCount int64
+	if err := db.Model(&model.AdminAuditEvent{}).Where("action = ?", "feature_availability.update").Count(&auditCount).Error; err != nil {
+		t.Fatal(err)
+	}
+	if auditCount != 1 {
+		t.Fatalf("audit count = %d, want 1", auditCount)
+	}
+}
+
+func TestTaskBillingOrderSkipsPricingWhenCreditsDisabled(t *testing.T) {
+	svc, _ := newFeatureAvailabilityTestService(t)
+	actor := &model.User{ID: "admin-1", Role: model.UserRoleAdmin}
+	if _, err := svc.UpdateFeatureAvailability(actor, FeatureAvailability{ShortDramaEnabled: true, TaskCenterEnabled: true, CreditsEnabled: false}); err != nil {
+		t.Fatal(err)
+	}
+
+	order, err := svc.taskBillingOrder("user-1", &model.Task{ID: "task-1"}, map[string]any{"config": map[string]any{"channelId": "missing", "model": "missing"}})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if order != nil {
+		t.Fatalf("taskBillingOrder() = %#v, want nil", order)
+	}
+}
+
+func newFeatureAvailabilityTestService(t *testing.T) (*Service, *gorm.DB) {
+	t.Helper()
+	db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if sqlDB, err := db.DB(); err == nil {
+		sqlDB.SetMaxOpenConns(1)
+	}
+	if err := db.AutoMigrate(&model.SystemSetting{}, &model.AdminAuditEvent{}); err != nil {
+		t.Fatal(err)
+	}
+	return New(repository.New(db), t.TempDir()), db
+}