package store

import (
	"context"
	"errors"
	"testing"
)

// Запись байтов .torrent на ветке создания и чтение их воркером.
func TestTorrentBlobWriteRead(t *testing.T) {
	st := newTestStore(t)
	ctx := context.Background()
	blob := []byte("d8:announce…fake torrent bytes")

	d := &Download{SourceType: SourceTorrent, SourceRef: "Some.Release", State: StateCatched}
	existing, err := st.CreateDownloadIfNoActive(ctx, d, []string{hashN(1)}, blob)
	if err != nil || existing != nil {
		t.Fatalf("create: err=%v existing=%v", err, existing)
	}
	got, err := st.GetTorrentData(ctx, d.ID)
	if err != nil {
		t.Fatalf("get: %v", err)
	}
	if string(got) != string(blob) {
		t.Errorf("blob = %q, want %q", got, blob)
	}
}

// При дедупе на активную задачу байты не пишутся (второй приём того же хеша).
func TestTorrentBlobNotWrittenOnDedup(t *testing.T) {
	st := newTestStore(t)
	ctx := context.Background()

	first := &Download{SourceType: SourceTorrent, SourceRef: "A", State: StateCatched}
	if _, err := st.CreateDownloadIfNoActive(ctx, first, []string{hashN(2)}, []byte("first")); err != nil {
		t.Fatalf("create first: %v", err)
	}
	// Второй приём того же инфохэша — дедуп; его байты писаться не должны.
	second := &Download{SourceType: SourceTorrent, SourceRef: "B", State: StateCatched}
	existing, err := st.CreateDownloadIfNoActive(ctx, second, []string{hashN(2)}, []byte("second"))
	if err != nil {
		t.Fatalf("create second: %v", err)
	}
	if existing == nil || existing.ID != first.ID {
		t.Fatalf("ожидался дедуп на первую задачу, got %v", existing)
	}
	got, err := st.GetTorrentData(ctx, first.ID)
	if err != nil {
		t.Fatalf("get first: %v", err)
	}
	if string(got) != "first" {
		t.Errorf("байты первой задачи перезаписаны: %q", got)
	}
}

// Нет байтов (magnet-источник) → ErrNotFound.
func TestTorrentBlobMissing(t *testing.T) {
	st := newTestStore(t)
	ctx := context.Background()
	d := &Download{SourceType: SourceMagnet, SourceRef: "magnet:?xt=urn:btih:" + hashN(3), State: StateCatched}
	if _, err := st.CreateDownloadIfNoActive(ctx, d, []string{hashN(3)}, nil); err != nil {
		t.Fatalf("create: %v", err)
	}
	if _, err := st.GetTorrentData(ctx, d.ID); !errors.Is(err, ErrNotFound) {
		t.Errorf("ожидался ErrNotFound, got %v", err)
	}
}