package mytest

import (
    "context"
    "testing"

    "github.com/atop0914/containerdb/pkg/mysql"
)

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

    // Spin up a MySQL container
    db, cleanup, err := mysql.New(ctx)
    if err != nil {
        t.Fatal(err)
    }
    defer cleanup()

    // Use db as a normal *sql.DB
    _, err = db.ExecContext(ctx, "CREATE TABLE users (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255))")
    if err != nil {
        t.Fatal(err)
    }

    _, err = db.ExecContext(ctx, "INSERT INTO users (name) VALUES (?)", "Alice")
    if err != nil {
        t.Fatal(err)
    }

    var name string
    err = db.QueryRowContext(ctx, "SELECT name FROM users WHERE id = 1").Scan(&name)
    if err != nil {
        t.Fatal(err)
    }

    if name != "Alice" {
        t.Errorf("expected Alice, got %s", name)
    }
}

import "github.com/atop0914/containerdb/pkg/mysql"

// Default settings
db, cleanup, err := mysql.New(ctx)
defer cleanup()

// Custom settings
db, cleanup, err := mysql.NewWithOptions(ctx,
    mysql.WithImage("mysql:8.4"),
    mysql.WithUsername("testuser"),
    mysql.WithPassword("testpass"),
    mysql.WithDatabase("myapp_test"),
    mysql.WithPoolSettings(20, 10, 30*time.Minute),
)
defer cleanup()

import "github.com/atop0914/containerdb/pkg/postgres"

// Default settings
db, cleanup, err := postgres.New(ctx)
defer cleanup()

// Custom settings
db, cleanup, err := postgres.NewWithOptions(ctx,
    postgres.WithImage("postgres:17"),
    postgres.WithDatabase("myapp_test"),
)
defer cleanup()

import "github.com/atop0914/containerdb/pkg/sqlite"

// In-memory (fastest)
db, cleanup, err := sqlite.InMemory()
defer cleanup()

// Temporary file (auto-cleaned)
db, cleanup, err := sqlite.TempDB()
defer cleanup()

import "github.com/atop0914/containerdb/pkg/migrate"

runner := migrate.NewRunner(db,
    migrate.WithDir("migrations"),
    migrate.WithTableName("schema_migrations"),
)

// Run all pending migrations
err := runner.Up(ctx)

// Rollback last 2 migrations
err := runner.Down(ctx, 2)

// Check migration status
statuses, err := runner.Status(ctx)

// Create new migration files
path, err := migrate.CreateMigration("migrations", "add_users_table")

import "github.com/atop0914/containerdb/pkg/compose"

runner := compose.NewRunner("myproject")

// Start services
err := runner.Up(ctx, true) // true = wait for health checks

// Stop services
err := runner.Down(ctx, false) // false = keep volumes

// Generate compose file
services := []compose.Service{...}
cf := compose.BuildComposeFile(services...)