54 lines
3.7 KiB
TypeScript
54 lines
3.7 KiB
TypeScript
import assert from "node:assert/strict"
|
|
import { pool, closePool } from "../db/index.js"
|
|
import { initDatabase } from "../db/bootstrap.js"
|
|
import { readGreBgpRunSnapshots } from "./gre-bgp-snapshot-history.js"
|
|
import { buildSignalSnapshotFromCollectors } from "./alert-engine/signals-from-collectors.js"
|
|
import { SCHEDULER_RUN_SNAPSHOT_VERSION } from "../types/scheduler-run-snapshot.js"
|
|
|
|
const name = process.env.MMAPP_OWN_SQL_DB
|
|
if (!name) console.log("GRE/BGP SQL: skipped; explicit owned database required")
|
|
else {
|
|
assert(/^mm_accept_166_\d+$/.test(name))
|
|
assert.equal(decodeURIComponent(new URL(process.env.DATABASE_URL!).pathname.slice(1)), name)
|
|
assert.equal((await pool.query("SELECT current_database() AS name")).rows[0].name, name)
|
|
const prefix = "own-history166-" + Date.now() + "-"
|
|
let sequence = 0
|
|
const start = Date.now() - 60_000
|
|
const old = { v: SCHEDULER_RUN_SNAPSHOT_VERSION, job: "gre_bgp", sampledAt: new Date(start).toISOString(), greWritten: 1, bgpWritten: 1, greTunnels: [{ targetLabel: "old-gre", status: "down" }], bgpPeers: [{ key: "old-peer", state: "Idle" }] }
|
|
async function insert(status: "ok" | "error", payload: unknown) {
|
|
const id = prefix + sequence++, time = new Date(start + sequence * 1000).toISOString()
|
|
await pool.query("INSERT INTO scheduler_runs(id,job_key,started_at,finished_at,status,error,result_json) VALUES($1,'gre_bgp',$2,$2,$3,$4,$5::jsonb)", [id, time, status, status === "error" ? "owned failure" : null, payload === null ? null : JSON.stringify(payload)])
|
|
}
|
|
try {
|
|
await initDatabase()
|
|
await insert("ok", old)
|
|
await insert("error", { ...old, greWritten: 0, greTunnels: [], bgpPeers: [{ key: "current-peer", state: "Established" }], errors: ["GRE: owned refusal"] })
|
|
assert.deepEqual((await readGreBgpRunSnapshots())[0].bgpPeers, [{ key: "current-peer", state: "Established" }])
|
|
let signals = await buildSignalSnapshotFromCollectors()
|
|
assert.deepEqual(signals.bgpSessions, [{ key: "current-peer", state: "Established", prevState: null }]); assert.deepEqual(signals.greTunnels, [])
|
|
for (const payload of [null, "{bad"]) {
|
|
await insert("error", payload)
|
|
assert((await readGreBgpRunSnapshots())[0].fatalError)
|
|
signals = await buildSignalSnapshotFromCollectors()
|
|
assert.deepEqual(signals.bgpSessions, []); assert.deepEqual(signals.greTunnels, [])
|
|
}
|
|
await insert("ok", { ...old, greTunnels: [], bgpPeers: [] })
|
|
signals = await buildSignalSnapshotFromCollectors()
|
|
assert.deepEqual(signals.bgpSessions, []); assert.deepEqual(signals.greTunnels, [])
|
|
await insert("ok", { ...old, greTunnels: [{ targetLabel: "duplicate", status: "up" }, { targetLabel: "duplicate", status: "down" }], bgpPeers: [{ key: "duplicate", state: "Established" }, { key: "duplicate", state: "Idle" }] })
|
|
signals = await buildSignalSnapshotFromCollectors()
|
|
assert.deepEqual(signals.bgpSessions, [{ key: "duplicate", state: "Unknown", prevState: null }]); assert.deepEqual(signals.greTunnels, [])
|
|
for (const sampledAt of [new Date(Date.now() - 120_000).toISOString(), new Date(Date.now() + 120_000).toISOString()]) {
|
|
await insert("ok", { ...old, sampledAt })
|
|
assert((await readGreBgpRunSnapshots())[0].fatalError)
|
|
signals = await buildSignalSnapshotFromCollectors()
|
|
assert.deepEqual(signals.bgpSessions, []); assert.deepEqual(signals.greTunnels, [])
|
|
}
|
|
console.log("GRE/BGP owned SQL: 8 real rows; newest error/partial, NULL/malformed, empty/duplicate and stale/future guards passed")
|
|
} finally {
|
|
await pool.query("DELETE FROM scheduler_runs WHERE id LIKE $1", [prefix + "%"])
|
|
assert.equal((await pool.query("SELECT count(*)::int AS n FROM scheduler_runs WHERE id LIKE $1", [prefix + "%"])).rows[0].n, 0)
|
|
await closePool()
|
|
}
|
|
}
|