5. Поднятие PostgreSQL в тестах

go get github.com/testcontainers/testcontainers-go/modules/postgres

package integration_test

import (
    "context"
    "testing"
    "time"

    "github.com/jackc/pgx/v5/pgxpool"
    "github.com/stretchr/testify/require"
    "github.com/testcontainers/testcontainers-go"
    "github.com/testcontainers/testcontainers-go/modules/postgres"
    "github.com/testcontainers/testcontainers-go/wait"
)

func setupPostgres(t *testing.T) *pgxpool.Pool {
    t.Helper()
    ctx := context.Background()

    // Конфигурируем контейнер с экстремальными оптимизациями скорости
    pgContainer, err := postgres.RunContainer(ctx,
        testcontainers.WithImage("postgres:16-alpine"),
        postgres.WithDatabase("test_db"),
        postgres.WithUsername("postgres"),
        postgres.WithPassword("postgres"),
        // Ждем именно второе вхождение готовности!
        wait.ForLog("database system is ready to accept connections").
            WithOccurrence(2).
            WithStartupTimeout(15*time.Second),

        // 1. Полностью отключаем сброс страниц на диск (fsync)
        testcontainers.WithCommand("-c", "fsync=off", "-c", "full_page_writes=off"),

        // 2. Размещаем директорию хранения данных в оперативной памяти (RAM-диск tmpfs)
        testcontainers.WithTmpfs(map[string]string{"/var/lib/postgresql/data": "rw"}),
    )
    require.NoError(t, err, "не удалось поднять контейнер PostgreSQL")

    // Регистрируем обязательную зачистку контейнера
    t.Cleanup(func() {
        if err := pgContainer.Terminate(context.Background()); err != nil {
            t.Logf("Ошибка при остановке контейнера PostgreSQL: %v", err)
        }
    })

    // Извлекаем динамический DSN для подключения
    connStr, err := pgContainer.ConnectionString(ctx, "sslmode=disable")
    require.NoError(t, err)

    // Инициализируем высокопроизводительный пул pgxpool
    pool, err := pgxpool.New(ctx, connStr)
    require.NoError(t, err)

    return pool
}

package repository_test

import (
    "context"
    "log"
    "os"
    "testing"

    "github.com/jackc/pgx/v5/pgxpool"
    // импорты драйвера, testcontainers и библиотеки миграций
)

// Глобальный пул соединений для тестов текущего пакета
var testDB *pgxpool.Pool

func TestMain(m *testing.M) {
    ctx := context.Background()

    // 1. Поднимаем инстанс PostgreSQL (один на весь пакет)
    pgContainer, err := startOptimizedPostgres(ctx)
    if err != nil {
        log.Fatalf("Не удалось запустить PostgreSQL: %v", err)
    }

    dsn, _ := pgContainer.ConnectionString(ctx, "sslmode=disable")
    testDB, err = pgxpool.New(ctx, dsn)
    if err != nil {
        log.Fatalf("Не удалось инициализировать пул pgx: %v", err)
    }

    // 2. Однократно применяем файлы миграций схемы
    if err := applyMigrations(dsn, "../../migrations"); err != nil {
        log.Fatalf("Ошибка наката миграций: %v", err)
    }

    // 3. Запускаем параллельные тесты пакета
    code := m.Run()

    // 4. Финальная уборка инфраструктуры
    testDB.Close()
    _ = pgContainer.Terminate(ctx)

    os.Exit(code)
}

func TestUserRepo_FindByID(t *testing.T) {
    // Безопасный запуск на всех доступных ядрах процессора!
    t.Parallel()

    // 1. Открываем изолированную транзакцию из глобального пула
    tx, err := testDB.Begin(context.Background())
    require.NoError(t, err)

    // 2. Гарантированный откат изменений в финале
    t.Cleanup(func() {
        _ = tx.Rollback(context.Background())
    })

    // 3. Внедряем транзакцию в репозиторий
    repo := repository.NewUserRepo(tx)

    // 4. Наполняем базу фикстурами прямо внутри транзакции
    _, err = tx.Exec(context.Background(), "INSERT INTO users (id, name) VALUES (1, 'Gopher')")
    require.NoError(t, err)

    // 5. Исполняем и верифицируем доменный метод
    user, err := repo.FindByID(context.Background(), 1)
    require.NoError(t, err)
    require.Equal(t, "Gopher", user.Name)
}