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) } }