mirror of
https://github.com/MeshCore-Beacon/beacon-server.git
synced 2026-11-11 18:57:33 +00:00
MrAlders0n
41baee9197
* feat(db): packet type domains, row-cursor indexes and the inverted-heard index * feat(api): exact page cursors for packets, nodes and observers * feat(db): normalise packet list filters before choosing a statement * feat(db): one packets statement per filter shape with an exact row cursor * fix(db): let the window statements decide since after until * feat(db): exact keyset cursors for nodes and observers; observers page by last_seen * feat(api): pageCursor on packets, nodes and observers; it wins over cursor * test(planguard): packets statements and node/observer keysets leave pending; diff and storediff cases * chore: bump version to 2.1.0 * test(db): turn off JIT in the keyset cursor tests
81 lines
3.0 KiB
Go
81 lines
3.0 KiB
Go
// Copyright 2026 Beacon Contributors
|
|
// SPDX-License-Identifier: AGPL-3.0-or-later
|
|
|
|
package db
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
sqlc "github.com/MeshCore-Beacon/beacon-server/db/sqlc"
|
|
"github.com/jackc/pgx/v5"
|
|
)
|
|
|
|
// The packet and its observation are separate ingest writes. A list request
|
|
// between them must return the packet without inventing an observer or failing.
|
|
func TestListPacketsBeforeObservationPostgres(t *testing.T) {
|
|
dsn := os.Getenv("BEACON_TEST_POSTGRES_DSN")
|
|
if dsn == "" {
|
|
t.Skip("set BEACON_TEST_POSTGRES_DSN for the PostgreSQL regression test")
|
|
}
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
|
|
defer cancel()
|
|
conn, err := pgx.Connect(ctx, dsn)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer conn.Close(context.Background())
|
|
tx, err := conn.Begin(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer tx.Rollback(context.Background())
|
|
for _, table := range []string{"packets", "packet_observations", "observers", "transport_scopes"} {
|
|
if _, err := tx.Exec(ctx, "CREATE TEMP TABLE "+table+" (LIKE public."+table+" INCLUDING ALL) ON COMMIT DROP"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
_, err = tx.Exec(ctx, `INSERT INTO packets (packet_hash, payload_type, payload_version, route_type, raw_payload, raw_header, first_heard_at, last_heard_at)
|
|
VALUES ('\x01', 4, 0, 1, '\x00', '\x00', NOW(), NOW())`)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
store := &Store{q: sqlc.New(tx)}
|
|
page, err := store.ListPackets(ctx, nil, nil, nil, nil, time.Time{}, time.Time{}, 0, nil, 50)
|
|
if err != nil {
|
|
t.Fatalf("packet without observation must remain readable: %v", err)
|
|
}
|
|
if len(page.Items) != 1 || page.Items[0].PacketHash != "01" {
|
|
t.Fatalf("packet missing: %+v", page)
|
|
}
|
|
if page.Items[0].LatestObserver != nil || page.Items[0].ObservationCount != 0 {
|
|
t.Fatalf("invented observation: %+v", page.Items[0])
|
|
}
|
|
_, err = tx.Exec(ctx, `
|
|
INSERT INTO observers (id, public_key) VALUES ('00000000-0000-0000-0000-000000000001', '\x02');
|
|
INSERT INTO packet_observations (id, packet_hash, observer_id, iata, heard_at, path_length_byte, hash_size, hop_count, path_bytes)
|
|
VALUES (1, '\x01', '00000000-0000-0000-0000-000000000001', 'YVR', NOW(), 2, 1, 2, '\x1122');
|
|
UPDATE packets SET observation_count = 1 WHERE packet_hash = '\x01';`)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, iatas := range [][]string{nil, {"YVR"}} {
|
|
page, err := store.ListPackets(ctx, nil, nil, iatas, nil, time.Time{}, time.Time{}, 0, nil, 50)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(page.Items) != 1 || page.Items[0].ObservationCount != 1 {
|
|
t.Fatalf("observed packet missing for %v: %+v", iatas, page)
|
|
}
|
|
observer := page.Items[0].LatestObserver
|
|
if observer == nil || observer.IATA != "YVR" || observer.PathLength == nil || observer.PathBytes == nil {
|
|
t.Fatalf("observation metadata missing: %+v", observer)
|
|
}
|
|
if observer.PathLength.Raw != "02" || observer.PathLength.HashSize != 1 || observer.PathLength.HopCount != 2 || *observer.PathBytes != "1122" {
|
|
t.Fatalf("observation path changed: %+v", observer)
|
|
}
|
|
}
|
|
}
|