124 lines
4.1 KiB
Go
124 lines
4.1 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"testing"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
func TestGetCommandPaletteContinueWatchingFiltersAndLimits(t *testing.T) {
|
|
sqlDB := openCommandPaletteTestDB(t)
|
|
|
|
got, err := New(sqlDB).GetCommandPaletteContinueWatching(context.Background(), "user-a", "continue", 1)
|
|
if err != nil {
|
|
t.Fatalf("GetCommandPaletteContinueWatching: %v", err)
|
|
}
|
|
if len(got) != 1 || got[0].AnimeID != 20 {
|
|
t.Fatalf("continue rows = %+v, want latest anime 20 only", got)
|
|
}
|
|
|
|
got, err = New(sqlDB).GetCommandPaletteContinueWatching(context.Background(), "user-a", "nar", 5)
|
|
if err != nil {
|
|
t.Fatalf("GetCommandPaletteContinueWatching filtered: %v", err)
|
|
}
|
|
if len(got) != 1 || got[0].AnimeID != 10 {
|
|
t.Fatalf("filtered continue rows = %+v, want anime 10", got)
|
|
}
|
|
}
|
|
|
|
func TestGetCommandPaletteWatchlistFiltersAndOrders(t *testing.T) {
|
|
sqlDB := openCommandPaletteTestDB(t)
|
|
|
|
got, err := New(sqlDB).GetCommandPaletteWatchlist(context.Background(), "user-a", "", 5)
|
|
if err != nil {
|
|
t.Fatalf("GetCommandPaletteWatchlist: %v", err)
|
|
}
|
|
if len(got) != 2 {
|
|
t.Fatalf("watchlist rows len = %d, want 2", len(got))
|
|
}
|
|
if got[0].AnimeID != 10 || got[1].AnimeID != 20 {
|
|
t.Fatalf("watchlist order = [%d %d], want watching anime 10 before plan anime 20", got[0].AnimeID, got[1].AnimeID)
|
|
}
|
|
|
|
got, err = New(sqlDB).GetCommandPaletteWatchlist(context.Background(), "user-a", "plan", 5)
|
|
if err != nil {
|
|
t.Fatalf("GetCommandPaletteWatchlist filtered: %v", err)
|
|
}
|
|
if len(got) != 1 || got[0].AnimeID != 20 {
|
|
t.Fatalf("filtered watchlist rows = %+v, want anime 20", got)
|
|
}
|
|
}
|
|
|
|
func openCommandPaletteTestDB(t *testing.T) *sql.DB {
|
|
t.Helper()
|
|
|
|
sqlDB, err := sql.Open("sqlite3", ":memory:")
|
|
if err != nil {
|
|
t.Fatalf("open sqlite: %v", err)
|
|
}
|
|
cleanupSQLDB(t, sqlDB)
|
|
|
|
_, err = sqlDB.ExecContext(context.Background(), `
|
|
CREATE TABLE anime (
|
|
id INTEGER PRIMARY KEY,
|
|
title_original TEXT NOT NULL,
|
|
title_english TEXT,
|
|
title_japanese TEXT,
|
|
image_url TEXT NOT NULL,
|
|
created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
airing BOOLEAN DEFAULT 0,
|
|
duration_seconds REAL
|
|
);
|
|
CREATE TABLE watch_list_entry (
|
|
id TEXT PRIMARY KEY,
|
|
user_id TEXT NOT NULL,
|
|
anime_id INTEGER NOT NULL,
|
|
status TEXT NOT NULL,
|
|
created_at DATETIME NOT NULL,
|
|
updated_at DATETIME NOT NULL,
|
|
current_episode INTEGER,
|
|
last_episode_at DATETIME,
|
|
current_time_seconds REAL NOT NULL DEFAULT 0
|
|
);
|
|
CREATE TABLE continue_watching_entry (
|
|
id TEXT PRIMARY KEY,
|
|
user_id TEXT NOT NULL,
|
|
anime_id INTEGER NOT NULL,
|
|
current_episode INTEGER,
|
|
current_time_seconds REAL NOT NULL DEFAULT 0,
|
|
duration_seconds REAL,
|
|
created_at DATETIME NOT NULL,
|
|
updated_at DATETIME NOT NULL
|
|
);
|
|
INSERT INTO anime (id, title_original, title_english, title_japanese, image_url, airing, duration_seconds) VALUES
|
|
(10, 'Naruto', NULL, NULL, 'naruto.jpg', 0, 1440),
|
|
(20, 'Frieren', 'Frieren: Beyond Journey''s End', NULL, 'frieren.jpg', 0, 1440),
|
|
(30, 'Dropped Show', NULL, NULL, 'dropped.jpg', 0, 1440);
|
|
INSERT INTO watch_list_entry (id, user_id, anime_id, status, created_at, updated_at, current_episode, current_time_seconds) VALUES
|
|
('w1', 'user-a', 10, 'watching', '2026-01-01 00:00:00', '2026-01-01 00:00:00', 3, 0),
|
|
('w2', 'user-a', 20, 'plan_to_watch', '2026-01-02 00:00:00', '2026-01-03 00:00:00', 0, 0),
|
|
('w3', 'user-a', 30, 'dropped', '2026-01-04 00:00:00', '2026-01-04 00:00:00', 0, 0),
|
|
('w4', 'user-b', 10, 'watching', '2026-01-05 00:00:00', '2026-01-05 00:00:00', 1, 0);
|
|
INSERT INTO continue_watching_entry (id, user_id, anime_id, current_episode, current_time_seconds, duration_seconds, created_at, updated_at) VALUES
|
|
('c1', 'user-a', 10, 4, 120, 1440, '2026-01-01 00:00:00', '2026-01-01 00:00:00'),
|
|
('c2', 'user-a', 20, 1, 60, 1440, '2026-01-02 00:00:00', '2026-01-03 00:00:00'),
|
|
('c3', 'user-b', 10, 1, 30, 1440, '2026-01-04 00:00:00', '2026-01-04 00:00:00');
|
|
`)
|
|
if err != nil {
|
|
t.Fatalf("seed command palette db: %v", err)
|
|
}
|
|
|
|
return sqlDB
|
|
}
|
|
|
|
func cleanupSQLDB(t *testing.T, sqlDB *sql.DB) {
|
|
t.Helper()
|
|
t.Cleanup(func() {
|
|
if err := sqlDB.Close(); err != nil {
|
|
t.Errorf("close sqlite: %v", err)
|
|
}
|
|
})
|
|
}
|