MeBox/internal/service/scanner_local_probe_queue_test.go

package service

import (
	"testing"
	"time"

	"go.uber.org/zap"
)

func newLocalProbeQueueTestScanner(capacity int) *ScannerService {
	return &ScannerService{
		log:                  zap.NewNop(),
		probe:                &FFprobeService{},
		localMediaProbeQueue: make(chan localMediaProbeTask, capacity),
		localMediaProbing:    make(map[string]struct{}),
	}
}

func TestNewLocalMediaProbeTaskTrimsAndRejectsInvalidInput(t *testing.T) {
	scanner := newLocalProbeQueueTestScanner(1)
	task, ok := scanner.newLocalMediaProbeTask(" C:/media/movie.mkv ")
	if !ok || task.path != "C:/media/movie.mkv" {
		t.Fatalf("task = %#v ok=%v, want trimmed valid task", task, ok)
	}
	if _, ok := scanner.newLocalMediaProbeTask(" \t "); ok {
		t.Fatal("blank path should be rejected")
	}
	scanner.probe = nil
	if _, ok := scanner.newLocalMediaProbeTask("C:/media/movie.mkv"); ok {
		t.Fatal("scanner without probe should reject local probe task")
	}
}

func TestReserveLocalMediaProbeRejectsDuplicateAndReleaseAllowsRetry(t *testing.T) {
	scanner := newLocalProbeQueueTestScanner(1)
	if !scanner.reserveLocalMediaProbe("C:/media/movie.mkv") {
		t.Fatal("first reserve should succeed")
	}
	if scanner.reserveLocalMediaProbe("C:/media/movie.mkv") {
		t.Fatal("duplicate reserve should fail")
	}
	scanner.releaseLocalMediaProbe("C:/media/movie.mkv")
	if !scanner.reserveLocalMediaProbe("C:/media/movie.mkv") {
		t.Fatal("reserve should succeed after release")
	}
}

func TestEnqueueLocalMediaProbeReportsFullQueue(t *testing.T) {
	scanner := newLocalProbeQueueTestScanner(1)
	if !scanner.enqueueLocalMediaProbe(localMediaProbeTask{path: "C:/media/a.mkv"}) {
		t.Fatal("first enqueue should fit buffer")
	}
	if scanner.enqueueLocalMediaProbe(localMediaProbeTask{path: "C:/media/b.mkv"}) {
		t.Fatal("second enqueue should report full queue")
	}
	select {
	case task := <-scanner.localMediaProbeQueue:
		if task.path != "C:/media/a.mkv" {
			t.Fatalf("queued task = %#v, want first path", task)
		}
	case <-time.After(time.Second):
		t.Fatal("expected queued task")
	}
}