MeBox/internal/service/play_profile_test.go

package service

import (
	"errors"
	"testing"

	"github.com/ShukeBta/MediaStationGo/internal/model"
	"github.com/ShukeBta/MediaStationGo/internal/repository"
	"github.com/glebarez/sqlite"
	"go.uber.org/zap"
	"gorm.io/gorm"
)

func newPlayProfileTestService(t *testing.T) *PlayProfileService {
	t.Helper()
	db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
	if err != nil {
		t.Fatal(err)
	}
	if err := db.AutoMigrate(&model.PlayProfile{}); err != nil {
		t.Fatal(err)
	}
	return NewPlayProfileService(zap.NewNop(), repository.New(db))
}

func TestPlayProfileVerifyPIN(t *testing.T) {
	service := newPlayProfileTestService(t)
	profile, err := service.Create(t.Context(), PlayProfileInput{
		UserID:     "user-1",
		Name:       "成人模式",
		AllowAdult: true,
		RequirePIN: true,
		PIN:        "1234",
	})
	if err != nil {
		t.Fatal(err)
	}

	if _, err := service.VerifyPIN(t.Context(), profile.ID, "user-1", "0000"); !errors.Is(err, ErrPlayProfilePINInvalid) {
		t.Fatalf("wrong PIN error = %v", err)
	}
	if _, err := service.VerifyPIN(t.Context(), profile.ID, "user-2", "1234"); !errors.Is(err, ErrPlayProfileForbidden) {
		t.Fatalf("wrong owner error = %v", err)
	}
	if verified, err := service.VerifyPIN(t.Context(), profile.ID, "user-1", "1234"); err != nil || verified.ID != profile.ID {
		t.Fatalf("verify PIN got profile=%v err=%v", verified, err)
	}
}

func TestPlayProfileCreateRequiresPINWhenEnabled(t *testing.T) {
	service := newPlayProfileTestService(t)
	if _, err := service.Create(t.Context(), PlayProfileInput{
		UserID:     "user-1",
		Name:       "锁定模式",
		RequirePIN: true,
	}); err == nil {
		t.Fatal("expected PIN-required profile create to fail without PIN")
	}
}

func TestPlayProfileCreateLimitIsPerUser(t *testing.T) {
	service := newPlayProfileTestService(t)

	for i := 1; i <= MaxPlayProfilesPerUser; i++ {
		if _, err := service.Create(t.Context(), PlayProfileInput{
			UserID: "user-1",
			Name:   "模式 " + string(rune('0'+i)),
		}); err != nil {
			t.Fatalf("create profile %d: %v", i, err)
		}
	}

	if _, err := service.Create(t.Context(), PlayProfileInput{
		UserID: "user-1",
		Name:   "超限模式",
	}); !errors.Is(err, ErrPlayProfileLimit) {
		t.Fatalf("expected limit error, got %v", err)
	}

	if _, err := service.Create(t.Context(), PlayProfileInput{
		UserID: "user-2",
		Name:   "另一个用户的模式",
	}); err != nil {
		t.Fatalf("different user should have independent limit: %v", err)
	}
}

func TestPlayProfileUpdateDeleteRequireOwner(t *testing.T) {
	service := newPlayProfileTestService(t)
	profile, err := service.Create(t.Context(), PlayProfileInput{
		UserID: "user-1",
		Name:   "私人模式",
	})
	if err != nil {
		t.Fatal(err)
	}

	if _, err := service.UpdateForUser(t.Context(), profile.ID, "user-2", PlayProfileInput{
		Name: "越权修改",
	}); !errors.Is(err, ErrPlayProfileForbidden) {
		t.Fatalf("expected forbidden update, got %v", err)
	}

	if err := service.DeleteForUser(t.Context(), profile.ID, "user-2"); !errors.Is(err, ErrPlayProfileForbidden) {
		t.Fatalf("expected forbidden delete, got %v", err)
	}

	updated, err := service.UpdateForUser(t.Context(), profile.ID, "user-1", PlayProfileInput{
		Name: "已修改",
	})
	if err != nil {
		t.Fatalf("owner update failed: %v", err)
	}
	if updated.Name != "已修改" || updated.UserID != "user-1" {
		t.Fatalf("unexpected updated profile: %+v", updated)
	}

	if err := service.DeleteForUser(t.Context(), profile.ID, "user-1"); err != nil {
		t.Fatalf("owner delete failed: %v", err)
	}
}