85 lines
3.3 KiB
Go
85 lines
3.3 KiB
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"telesrv/internal/domain"
|
|
)
|
|
|
|
func TestPhoneChangeStoreUpdatesUserWithoutPTSPostgres(t *testing.T) {
|
|
pool := testPool(t)
|
|
ctx := context.Background()
|
|
users := NewUserStore(pool)
|
|
changes := NewPhoneChangeStore(pool)
|
|
events := NewUpdateEventStore(pool)
|
|
suffix := time.Now().UnixNano() % 1_000_000_000
|
|
oldPhone := fmt.Sprintf("1661%d01", suffix)
|
|
occupiedPhone := fmt.Sprintf("1661%d02", suffix)
|
|
newPhone := fmt.Sprintf("1661%d03", suffix)
|
|
u1, err := users.Create(ctx, domain.User{AccessHash: 301, Phone: oldPhone, FirstName: "PhoneOne"})
|
|
if err != nil {
|
|
t.Fatalf("create user1: %v", err)
|
|
}
|
|
u2, err := users.Create(ctx, domain.User{AccessHash: 302, Phone: occupiedPhone, FirstName: "PhoneTwo"})
|
|
if err != nil {
|
|
t.Fatalf("create user2: %v", err)
|
|
}
|
|
t.Cleanup(func() {
|
|
_, _ = pool.Exec(context.Background(), "DELETE FROM dispatch_outbox WHERE target_user_id = ANY($1::bigint[])", []int64{u1.ID, u2.ID})
|
|
_, _ = pool.Exec(context.Background(), "DELETE FROM user_update_events WHERE user_id = ANY($1::bigint[])", []int64{u1.ID, u2.ID})
|
|
_, _ = pool.Exec(context.Background(), "DELETE FROM user_update_watermarks WHERE user_id = ANY($1::bigint[])", []int64{u1.ID, u2.ID})
|
|
_, _ = pool.Exec(context.Background(), "DELETE FROM users WHERE id = ANY($1::bigint[])", []int64{u1.ID, u2.ID})
|
|
})
|
|
|
|
authKeyID := [8]byte{7, 6, 5, 4}
|
|
result, err := changes.ChangePhone(ctx, domain.PhoneChangeRequest{
|
|
UserID: u1.ID, Phone: newPhone, Date: 1700000001,
|
|
ExcludeAuthKeyID: authKeyID, ExcludeSessionID: 77,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("change phone: %v", err)
|
|
}
|
|
if !result.Changed || result.User.Phone != newPhone {
|
|
t.Fatalf("result = %+v", result)
|
|
}
|
|
loaded, found, err := users.ByID(ctx, u1.ID)
|
|
if err != nil || !found || loaded.Phone != newPhone {
|
|
t.Fatalf("loaded user = %+v found=%v err=%v", loaded, found, err)
|
|
}
|
|
storedEvents, err := events.ListAfter(ctx, u1.ID, 0, 10)
|
|
if err != nil || len(storedEvents) != 0 {
|
|
t.Fatalf("stored events = %+v err=%v", storedEvents, err)
|
|
}
|
|
var outboxCount int
|
|
if err := pool.QueryRow(ctx, `SELECT count(*) FROM dispatch_outbox WHERE target_user_id = $1`, u1.ID).Scan(&outboxCount); err != nil {
|
|
t.Fatalf("query outbox: %v", err)
|
|
}
|
|
if outboxCount != 0 {
|
|
t.Fatalf("outbox count = %d, want 0", outboxCount)
|
|
}
|
|
|
|
// 同号重试是幂等读，不得产生 PTS/event/outbox。
|
|
retry, err := changes.ChangePhone(ctx, domain.PhoneChangeRequest{UserID: u1.ID, Phone: newPhone, Date: 1700000002})
|
|
if err != nil || retry.Changed || retry.User.Phone != newPhone {
|
|
t.Fatalf("idempotent retry = %+v err=%v", retry, err)
|
|
}
|
|
if pts, err := events.MaxContiguousPts(ctx, u1.ID); err != nil || pts != 1 {
|
|
t.Fatalf("pts after retry = %d err=%v", pts, err)
|
|
}
|
|
|
|
// 冲突更新整体回滚：号码不变，也不产生 PTS/event。
|
|
if _, err := changes.ChangePhone(ctx, domain.PhoneChangeRequest{UserID: u2.ID, Phone: newPhone}); !errors.Is(err, domain.ErrPhoneNumberOccupied) {
|
|
t.Fatalf("occupied change err = %v", err)
|
|
}
|
|
loaded2, found, err := users.ByID(ctx, u2.ID)
|
|
if err != nil || !found || loaded2.Phone != occupiedPhone {
|
|
t.Fatalf("occupied rollback user = %+v found=%v err=%v", loaded2, found, err)
|
|
}
|
|
if pts, err := events.MaxContiguousPts(ctx, u2.ID); err != nil || pts != 0 {
|
|
t.Fatalf("occupied rollback pts = %d err=%v", pts, err)
|
|
}
|
|
}
|