package stardb

import (
	"testing"
	"time"
)

type User struct {
	ID        int64     `db:"id"`
	Name      string    `db:"name"`
	Email     string    `db:"email"`
	Age       int       `db:"age"`
	Balance   float64   `db:"balance"`
	Active    bool      `db:"active"`
	CreatedAt time.Time `db:"created_at"`
}

type Profile struct {
	UserID int    `db:"user_id"`
	Bio    string `db:"bio"`
	Avatar string `db:"avatar"`
}

type NestedUser struct {
	ID      int64  `db:"id"`
	Name    string `db:"name"`
	Profile `db:"---"`
}

func TestBuildInsertSQL(t *testing.T) {
	user := User{
		ID:        1,
		Name:      "Test",
		Email:     "test@example.com",
		Age:       30,
		Balance:   100.0,
		Active:    true,
		CreatedAt: time.Now(),
	}

	query, params, err := buildInsertSQL(&user, "users", "id")
	if err != nil {
		t.Fatalf("buildInsertSQL failed: %v", err)
	}

	expectedQuery := "INSERT INTO users (name, email, age, balance, active, created_at) VALUES (?, ?, ?, ?, ?, ?)"
	if query != expectedQuery {
		t.Errorf("Expected query:\n%s\nGot:\n%s", expectedQuery, query)
	}

	expectedParams := []string{":name", ":email", ":age", ":balance", ":active", ":created_at"}
	if len(params) != len(expectedParams) {
		t.Errorf("Expected %d params, got %d", len(expectedParams), len(params))
	}

	for i, param := range params {
		if param != expectedParams[i] {
			t.Errorf("Expected param %s, got %s", expectedParams[i], param)
		}
	}
}

func TestBuildUpdateSQL(t *testing.T) {
	user := User{
		ID:        1,
		Name:      "Test",
		Email:     "test@example.com",
		Age:       30,
		Balance:   100.0,
		Active:    true,
		CreatedAt: time.Now(),
	}

	query, params, err := buildUpdateSQL(&user, "users", "id")
	if err != nil {
		t.Fatalf("buildUpdateSQL failed: %v", err)
	}

	expectedQuery := "UPDATE users SET id = ?, name = ?, email = ?, age = ?, balance = ?, active = ?, created_at = ? WHERE id = ?"
	if query != expectedQuery {
		t.Errorf("Expected query:\n%s\nGot:\n%s", expectedQuery, query)
	}

	expectedParamCount := 8 // 7 fields + 1 primary key
	if len(params) != expectedParamCount {
		t.Errorf("Expected %d params, got %d", expectedParamCount, len(params))
	}
}