package repository

import (
	"net/url"
	"strings"
	"testing"

	"github.com/dhao2001/mygo/internal/config"
)

func TestSQLiteImmediateDSNPreservesParameters(t *testing.T) {
	dsn := sqliteImmediateDSN("file:test.db?cache=shared&_busy_timeout=9000&_txlock=deferred")
	base, rawQuery, found := strings.Cut(dsn, "?")
	if !found || base != "file:test.db" {
		t.Fatalf("DSN base = %q, want file:test.db", base)
	}
	query, err := url.ParseQuery(rawQuery)
	if err != nil {
		t.Fatalf("parse DSN query: %v", err)
	}
	if query.Get("cache") != "shared" || query.Get("_busy_timeout") != "9000" {
		t.Fatalf("DSN did not preserve parameters: %q", dsn)
	}
	if query.Get("_txlock") != "immediate" {
		t.Fatalf("_txlock = %q, want immediate", query.Get("_txlock"))
	}
}

func TestOpenSQLite(t *testing.T) {
	cfg := config.DatabaseConfig{
		Driver: "sqlite3",
		SQLite: config.SQLiteConfig{Path: ":memory:"},
	}

	db, err := Open(cfg)
	if err != nil {
		t.Fatalf("Open(sqlite3) = %v", err)
	}

	sqlDB, err := db.DB()
	if err != nil {
		t.Fatalf("db.DB() = %v", err)
	}
	if err := sqlDB.Ping(); err != nil {
		t.Fatalf("ping = %v", err)
	}
}

func TestOpenUnsupportedDriver(t *testing.T) {
	cfg := config.DatabaseConfig{Driver: "mysql"}
	_, err := Open(cfg)
	if err == nil {
		t.Fatal("expected error for unsupported driver, got nil")
	}
}

func TestAutoMigrate(t *testing.T) {
	cfg := config.DatabaseConfig{
		Driver: "sqlite3",
		SQLite: config.SQLiteConfig{Path: ":memory:"},
	}

	db, err := Open(cfg)
	if err != nil {
		t.Fatalf("Open = %v", err)
	}

	if err := AutoMigrate(db); err != nil {
		t.Fatalf("AutoMigrate = %v", err)
	}

	// Verify tables exist
	for _, table := range []string{"users", "sessions", "files"} {
		if !db.Migrator().HasTable(table) {
			t.Errorf("table %q not found after migration", table)
		}
	}
}