105 lines
4.0 KiB
Go
105 lines
4.0 KiB
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)
|
||
|
|
}
|
||
|
|
}
|