Добавил бот для Telegram

internal/worker/review_test.go

@@ -8,6 +8,7 @@ import (
 	"os"
 	"path/filepath"
 	"testing"
+	"time"
 
 	"git.vakhrushev.me/av/jellybit/internal/layout"
 	"git.vakhrushev.me/av/jellybit/internal/qbt"
@@ -15,6 +16,61 @@ import (
 	"git.vakhrushev.me/av/jellybit/internal/store"
 )
 
+// recordingNotifier ловит события пинга (Notify асинхронен — через канал).
+type notifyEvent struct {
+	id int64
+	ev NotifyEvent
+}
+type recordingNotifier struct{ ch chan notifyEvent }
+
+func (n *recordingNotifier) Notify(_ context.Context, id int64, ev NotifyEvent) {
+	n.ch <- notifyEvent{id, ev}
+}
+
+func waitNotify(t *testing.T, n *recordingNotifier) notifyEvent {
+	t.Helper()
+	select {
+	case e := <-n.ch:
+		return e
+	case <-time.After(2 * time.Second):
+		t.Fatal("пинг не пришёл")
+		return notifyEvent{}
+	}
+}
+
+func TestNotifier_FiresOnReview(t *testing.T) {
+	st := newMemStore()
+	st.put(completedDownload(1))
+	qb := &fakeQbt{
+		torrents: []qbt.Torrent{{Hash: ihTest, Name: "Show", SavePath: "/d"}},
+		files:    []qbt.File{{Name: "Show/e1.mkv", Size: 1}},
+	}
+	w := testWorkerWith(st, qb, &fakeRecognizer{result: seriesResult()}, nil)
+	n := &recordingNotifier{ch: make(chan notifyEvent, 4)}
+	w.SetNotifier(n)
+
+	w.recognizeOne(context.Background(), 1)
+
+	e := waitNotify(t, n)
+	if e.id != 1 || e.ev != EventReview {
+		t.Errorf("event = %+v, want {1 review}", e)
+	}
+}
+
+func TestNotifier_FiresOnDone(t *testing.T) {
+	f := newApplyFixture(t, seriesResult().Plan)
+	n := &recordingNotifier{ch: make(chan notifyEvent, 4)}
+	f.w.SetNotifier(n)
+
+	if err := f.w.Apply(context.Background(), 1); err != nil {
+		t.Fatalf("Apply: %v", err)
+	}
+	e := waitNotify(t, n)
+	if e.id != 1 || e.ev != EventDone {
+		t.Errorf("event = %+v, want {1 done}", e)
+	}
+}
+
 // memStore — полноценный in-memory store для тестов Ф3.
 type memStore struct {
 	downloads map[int64]*store.Download