696d0227a3
- Modified the SQL query in the MarkResolved function to select torrents with both 'active' and 'manual_needed' statuses, enhancing the functionality of the method.
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package store
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"ncore-hnr/internal/model"
|
|
)
|
|
|
|
func TestMarkResolvedResolvesManualNeededTorrent(t *testing.T) {
|
|
db, err := Open(t.TempDir() + "/state.sqlite")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer db.Close()
|
|
|
|
now := time.Date(2026, 5, 7, 1, 0, 0, 0, time.UTC)
|
|
torrent := model.Torrent{ID: 123, Name: "The Black Dagger Brotherhood S01E01 720p"}
|
|
if _, err := db.UpsertSeen(torrent, now); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.MarkManualNeeded(torrent.ID, now.Add(time.Hour)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
resolvedAt := now.Add(2 * time.Hour)
|
|
if err := db.MarkResolved(map[int64]bool{}, resolvedAt); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
status, err := db.Status()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(status.Torrents) != 1 {
|
|
t.Fatalf("expected 1 tracked torrent, got %d", len(status.Torrents))
|
|
}
|
|
got := status.Torrents[0]
|
|
if got.Status != "resolved" {
|
|
t.Fatalf("expected status resolved, got %q", got.Status)
|
|
}
|
|
if got.LastResolvedAt != resolvedAt.Format(time.RFC3339) {
|
|
t.Fatalf("expected last_resolved_at %q, got %q", resolvedAt.Format(time.RFC3339), got.LastResolvedAt)
|
|
}
|
|
}
|