astra/owpengram-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

perf: sync protocol and core hardening updates

Browse source
This commit is contained in:
A 2026-07-11 19:48:26 +08:00
parent 152fed3b87
commit 4390ebf5a9
283 changed files with 29231 additions and 2295 deletions
Download patch file Download diff file Expand all files Collapse all files

53
internal/store/postgres/bootstrap_update_job_integration_test.go Normal file
View file

@ -0,0 +1,53 @@
package postgres
import (
"context"
"testing"
"time"
"telesrv/internal/domain"
)
func TestBootstrapUpdateJobPostgresSameAuthKeyReconnectTakesOverPendingSession(t *testing.T) {
pool := testPool(t)
ctx := context.Background()
user := createLoginCodeDeliveryTestUser(t, ctx, pool, "bootstrap-reconnect")
msg, err := NewMessageStore(pool).Create(ctx, domain.Message{
OwnerUserID: user.ID,
Peer: domain.Peer{Type: domain.PeerTypeUser, ID: domain.OfficialSystemUserID},
From: domain.Peer{Type: domain.PeerTypeUser, ID: domain.OfficialSystemUserID},
Date: int(time.Now().Unix()),
Body: "Login code: 12345",
})
if err != nil {
t.Fatalf("create bootstrap message: %v", err)
}
bootstrap := NewBootstrapUpdateJobStore(pool)
authKeyID := [8]byte{1, 3, 5, 7}
const (
oldSessionID = int64(11001)
newSessionID = int64(22002)
)
job, err := bootstrap.EnqueueLoginMessage(ctx, domain.BootstrapUpdateJob{
Kind: domain.BootstrapUpdateJobLoginMessage, UserID: user.ID,
AuthKeyID: authKeyID, SessionID: oldSessionID, MessageBoxID: msg.ID,
})
if err != nil {
t.Fatalf("enqueue bootstrap: %v", err)
}
if ready, err := bootstrap.MarkReadyForSession(ctx, user.ID, [8]byte{9}, newSessionID); err != nil || ready != 0 {
t.Fatalf("different-auth ready=%d err=%v, want 0/nil", ready, err)
}
ready, err := bootstrap.MarkReadyForSession(ctx, user.ID, authKeyID, newSessionID)
if err != nil || ready != 1 {
t.Fatalf("same-auth reconnect ready=%d err=%v, want 1/nil", ready, err)
}
var status string
var sessionID int64
if err := pool.QueryRow(ctx, `SELECT status, session_id FROM bootstrap_update_jobs WHERE id = $1`, job.ID).Scan(&status, &sessionID); err != nil {
t.Fatalf("load bootstrap job: %v", err)
}
if status != string(domain.BootstrapUpdateJobReady) || sessionID != newSessionID {
t.Fatalf("bootstrap status/session = %s/%d, want ready/%d", status, sessionID, newSessionID)
}
}