package repository

import (
	"testing"

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

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)
		}
	}
}