package database

import (
	"context"
	"fmt"
	"path/filepath"
	"testing"
)

func TestDatabase(t *testing.T) {
	ctx := context.Background()
	dbPath := filepath.Join(t.TempDir(), "test.db")

	db, err := New(ctx, dbPath)
	if err != nil {
		t.Fatalf("failed to create database: %v", err)
	}
	defer func() {
		if err := db.Close(); err != nil {
			t.Errorf("failed to close database: %v", err)
		}
	}()

	// Test connection
	if db.Conn() == nil {
		t.Fatal("database connection is nil")
	}

	// Test schema creation (already done in New)
	// Verify tables exist
	tables := []string{
		"files", "file_chunks", "chunks", "blobs",
		"blob_chunks", "chunk_files", "snapshots",
	}

	for _, table := range tables {
		var name string
		err := db.conn.QueryRow("SELECT name FROM sqlite_master WHERE type='table' AND name=?", table).Scan(&name)
		if err != nil {
			t.Errorf("table %s does not exist: %v", table, err)
		}
	}
}

func TestDatabaseInvalidPath(t *testing.T) {
	ctx := context.Background()

	// Test with invalid path
	_, err := New(ctx, "/invalid/path/that/does/not/exist/test.db")
	if err == nil {
		t.Fatal("expected error for invalid path")
	}
}

func TestDatabaseConcurrentAccess(t *testing.T) {
	ctx := context.Background()
	dbPath := filepath.Join(t.TempDir(), "test.db")

	db, err := New(ctx, dbPath)
	if err != nil {
		t.Fatalf("failed to create database: %v", err)
	}
	defer func() {
		if err := db.Close(); err != nil {
			t.Errorf("failed to close database: %v", err)
		}
	}()

	// Test concurrent writes
	type result struct {
		index int
		err   error
	}
	results := make(chan result, 10)

	for i := 0; i < 10; i++ {
		go func(i int) {
			_, err := db.ExecWithLog(ctx, "INSERT INTO chunks (chunk_hash, size) VALUES (?, ?)",
				fmt.Sprintf("hash%d", i), i*1024)
			results <- result{index: i, err: err}
		}(i)
	}

	// Wait for all goroutines and check results
	for i := 0; i < 10; i++ {
		r := <-results
		if r.err != nil {
			t.Fatalf("concurrent insert %d failed: %v", r.index, r.err)
		}
	}

	// Verify all inserts succeeded
	var count int
	err = db.conn.QueryRowContext(ctx, "SELECT COUNT(*) FROM chunks").Scan(&count)
	if err != nil {
		t.Fatalf("failed to count chunks: %v", err)
	}
	if count != 10 {
		t.Errorf("expected 10 chunks, got %d", count)
	}
}