glchat/internal/database/push_migration_test.go

package database

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

// TestPushSubscriptionsMigration проверяет миграцию 00021 на чистой базе:
// таблица подписок Web Push, уникальность эндпоинта, внешний ключ на users и
// индекс по пользователю (AGENT.md 7.16, Фаза 7).
func TestPushSubscriptionsMigration(t *testing.T) {
	ctx := context.Background()
	db, err := Open(ctx, Options{
		Path:     filepath.Join(t.TempDir(), "glchat.db"),
		ReadPool: 2,
		Migrate:  true,
	})
	if err != nil {
		t.Fatalf("Open() вернул ошибку: %v", err)
	}
	defer func() {
		if err := db.Close(); err != nil {
			t.Errorf("Close(): %v", err)
		}
	}()

	version, err := db.SchemaVersion(ctx)
	if err != nil {
		t.Fatalf("SchemaVersion(): %v", err)
	}
	if version < 21 {
		t.Fatalf("schema_version = %d, ожидалось >= 21", version)
	}

	columns := map[string]bool{}
	rows, err := db.Reader.QueryContext(ctx, `SELECT name FROM pragma_table_info('push_subscriptions')`)
	if err != nil {
		t.Fatalf("pragma_table_info: %v", err)
	}
	defer rows.Close()
	for rows.Next() {
		var name string
		if err := rows.Scan(&name); err != nil {
			t.Fatalf("scan: %v", err)
		}
		columns[name] = true
	}
	if err := rows.Err(); err != nil {
		t.Fatalf("rows: %v", err)
	}
	for _, expected := range []string{
		"id", "user_id", "endpoint", "p256dh", "auth", "user_agent",
		"created_at", "last_success_at", "last_failure_at", "failure_count",
	} {
		if !columns[expected] {
			t.Errorf("в push_subscriptions нет колонки %q", expected)
		}
	}

	var indexes int
	if err := db.Reader.QueryRowContext(ctx,
		`SELECT COUNT(*) FROM sqlite_master WHERE type = 'index' AND name = 'push_subscriptions_user_idx'`,
	).Scan(&indexes); err != nil {
		t.Fatalf("чтение индекса: %v", err)
	}
	if indexes != 1 {
		t.Errorf("индекс push_subscriptions_user_idx не создан")
	}

	userID := int64(1000)
	if _, err := db.Writer.ExecContext(ctx,
		`INSERT INTO users (id, username, username_lower, display_name, email_enc, email_index,
			password_hash, locale, created_at, updated_at)
		 VALUES (?, 'push', 'push', 'Push', 'enc', 'idx', 'hash', 'ru', '2026-01-01T00:00:00.000Z', '2026-01-01T00:00:00.000Z')`,
		userID); err != nil {
		t.Fatalf("вставка пользователя: %v", err)
	}
	insert := `INSERT INTO push_subscriptions (id, user_id, endpoint, p256dh, auth, created_at)
		VALUES (?, ?, ?, 'key', 'auth', '2026-01-01T00:00:00.000Z')`
	if _, err := db.Writer.ExecContext(ctx, insert, 1, userID, "https://push.example.com/one"); err != nil {
		t.Fatalf("вставка подписки: %v", err)
	}
	// Эндпоинт уникален: повторная подписка с того же устройства обновляет
	// запись, а не создаёт вторую.
	if _, err := db.Writer.ExecContext(ctx, insert, 2, userID, "https://push.example.com/one"); err == nil {
		t.Fatal("повторный эндпоинт принят, ожидалась ошибка уникальности")
	}
	// Внешний ключ: подписка без пользователя невозможна.
	if _, err := db.Writer.ExecContext(ctx, insert, 3, 999999, "https://push.example.com/two"); err == nil {
		t.Fatal("подписка без пользователя принята, ожидалась ошибка внешнего ключа")
	}
	// Удаление пользователя убирает его подписки.
	if _, err := db.Writer.ExecContext(ctx, `DELETE FROM users WHERE id = ?`, userID); err != nil {
		t.Fatalf("удаление пользователя: %v", err)
	}
	var remaining int
	if err := db.Reader.QueryRowContext(ctx, `SELECT COUNT(*) FROM push_subscriptions`).Scan(&remaining); err != nil {
		t.Fatalf("подсчёт подписок: %v", err)
	}
	if remaining != 0 {
		t.Errorf("после удаления пользователя осталось подписок: %d", remaining)
	}
}