MeBox/internal/service/subscription_identity_test.go

package service

import (
	"errors"
	"path/filepath"
	"sync"
	"testing"
	"time"

	"go.uber.org/zap"

	"github.com/ShukeBta/MediaStationGo/internal/config"
	"github.com/ShukeBta/MediaStationGo/internal/database"
	"github.com/ShukeBta/MediaStationGo/internal/model"
	"github.com/ShukeBta/MediaStationGo/internal/repository"
)

func TestSubscriptionCreateRejectsConcurrentDuplicateRules(t *testing.T) {
	cfg := &config.Config{}
	cfg.Database.Type = "sqlite"
	cfg.Database.DBPath = filepath.Join(t.TempDir(), "subscriptions.db")
	cfg.Database.WALMode = true
	cfg.Database.BusyTimeout = 5000
	cfg.Database.MaxOpenConns = 4
	db, err := database.Open(cfg, zap.NewNop())
	if err != nil {
		t.Fatal(err)
	}
	if sqlDB, dbErr := db.DB(); dbErr == nil {
		t.Cleanup(func() { _ = sqlDB.Close() })
	}
	if err := database.AutoMigrate(db); err != nil {
		t.Fatal(err)
	}
	repos := repository.New(db)
	svc := NewSubscriptionService(cfg, zap.NewNop(), repos, nil, nil, NewHub(zap.NewNop()))

	start := make(chan struct{})
	results := make(chan error, 2)
	var wg sync.WaitGroup
	for i := 0; i < 2; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()
			<-start
			results <- svc.Create(t.Context(), &model.Subscription{
				UserID:     "user-1",
				Name:       "Example Show",
				FeedURL:    "site-search://search?keyword=Example+Show",
				Filter:     "Example Show",
				MediaType:  "tv",
				Resolution: "1080p",
				Enabled:    true,
			})
		}()
	}
	close(start)
	wg.Wait()
	close(results)

	var created, conflicts int
	for createErr := range results {
		switch {
		case createErr == nil:
			created++
		case errors.Is(createErr, ErrSubscriptionAlreadyExists):
			conflicts++
		default:
			t.Fatalf("unexpected create error: %v", createErr)
		}
	}
	if created != 1 || conflicts != 1 {
		t.Fatalf("created=%d conflicts=%d, want 1/1", created, conflicts)
	}
	var count int64
	if err := db.Model(&model.Subscription{}).Where("archived_at IS NULL").Count(&count).Error; err != nil {
		t.Fatal(err)
	}
	if count != 1 {
		t.Fatalf("active subscription count = %d, want 1", count)
	}
}

func TestSubscriptionUpdateAndRestoreReturnExistingConflict(t *testing.T) {
	db := newServiceTestDB(t, &model.Subscription{})
	if err := database.AutoMigrate(db); err != nil {
		t.Fatal(err)
	}
	repos := repository.New(db)
	svc := NewSubscriptionService(&config.Config{}, zap.NewNop(), repos, nil, nil, NewHub(zap.NewNop()))
	first := &model.Subscription{UserID: "user-1", Name: "Example", FeedURL: "rss://example", Filter: "Example", Resolution: "1080p", Enabled: true}
	second := &model.Subscription{UserID: "user-1", Name: "Example", FeedURL: "rss://example", Filter: "Example", Resolution: "2160p", Enabled: true}
	if err := svc.Create(t.Context(), first); err != nil {
		t.Fatal(err)
	}
	if err := svc.Create(t.Context(), second); err != nil {
		t.Fatal(err)
	}
	if err := svc.Update(t.Context(), second.ID, map[string]any{"resolution": "1080p"}); !errors.Is(err, ErrSubscriptionAlreadyExists) || SubscriptionAlreadyExistsID(err) != first.ID {
		t.Fatalf("update error = %v, want conflict with %s", err, first.ID)
	}

	archivedAt := time.Now()
	archived := *first
	archived.ID = ""
	archived.ArchivedAt = &archivedAt
	archived.Enabled = false
	if err := repos.Subscription.Create(t.Context(), &archived); err != nil {
		t.Fatal(err)
	}
	if _, err := svc.Restore(t.Context(), archived.ID); !errors.Is(err, ErrSubscriptionAlreadyExists) || SubscriptionAlreadyExistsID(err) != first.ID {
		t.Fatalf("restore error = %v, want conflict with %s", err, first.ID)
	}
}