mirror of
https://github.com/Nezumi-2711/uptime-monitoring.git
synced 2026-09-22 13:48:31 +00:00
100 lines
4.0 KiB
TypeScript
100 lines
4.0 KiB
TypeScript
import { applyD1Migrations, type D1Migration } from 'cloudflare:test';
|
|
import { env } from 'cloudflare:workers';
|
|
import { beforeAll, beforeEach, describe, expect, it } from 'vitest';
|
|
import { runDailyRollup } from '../src/worker/scheduled/rollup';
|
|
|
|
describe('daily monitor rollups', () => {
|
|
beforeAll(async () => {
|
|
const testEnv = env as Env & { TEST_MIGRATIONS: D1Migration[] };
|
|
await applyD1Migrations(testEnv.DB, testEnv.TEST_MIGRATIONS);
|
|
});
|
|
beforeEach(async () => {
|
|
await env.DB.batch([
|
|
env.DB.prepare('DELETE FROM ai_events'),
|
|
env.DB.prepare('DELETE FROM maintenance_window_monitors'),
|
|
env.DB.prepare('DELETE FROM maintenance_windows'),
|
|
env.DB.prepare('DELETE FROM monitor_daily_stats'),
|
|
env.DB.prepare('DELETE FROM checks'),
|
|
env.DB.prepare('DELETE FROM incidents'),
|
|
env.DB.prepare('DELETE FROM monitors'),
|
|
]);
|
|
});
|
|
|
|
it('aggregates the previous UTC day and safely upserts on rerun', async () => {
|
|
const now = new Date('2026-08-28T00:05:00.000Z');
|
|
const createdAt = now.getTime();
|
|
const insert = await env.DB.prepare(
|
|
"INSERT INTO monitors (name, url, method, expected_status, interval_seconds, timeout_ms, enabled, alerts_enabled, created_at, updated_at) VALUES ('API', 'https://example.com', 'GET', 200, 300, 10000, 1, 1, ?, ?)",
|
|
)
|
|
.bind(createdAt, createdAt)
|
|
.run();
|
|
const id = Number(insert.meta.last_row_id);
|
|
await env.DB.batch([
|
|
env.DB.prepare('INSERT INTO checks (monitor_id, ok, status_code, latency_ms, checked_at) VALUES (?, 1, 200, 100, ?)').bind(
|
|
id,
|
|
Date.parse('2026-08-27T02:00:00Z'),
|
|
),
|
|
env.DB.prepare('INSERT INTO checks (monitor_id, ok, status_code, latency_ms, checked_at) VALUES (?, 1, 200, 200, ?)').bind(
|
|
id,
|
|
Date.parse('2026-08-27T12:00:00Z'),
|
|
),
|
|
env.DB.prepare('INSERT INTO checks (monitor_id, ok, status_code, latency_ms, checked_at) VALUES (?, 0, 500, 300, ?)').bind(
|
|
id,
|
|
Date.parse('2026-08-27T22:00:00Z'),
|
|
),
|
|
env.DB.prepare(
|
|
'INSERT INTO checks (monitor_id, ok, status_code, latency_ms, checked_at, maintenance) VALUES (?, 0, 503, 900, ?, 1)',
|
|
).bind(id, Date.parse('2026-08-27T23:00:00Z')),
|
|
]);
|
|
|
|
expect(await runDailyRollup(env, now)).toEqual({ day: '2026-08-27', monitors: 1 });
|
|
await runDailyRollup(env, now);
|
|
const rows = await env.DB.prepare(
|
|
'SELECT day, total_checks, up_checks, avg_latency_ms, min_latency_ms, max_latency_ms FROM monitor_daily_stats WHERE monitor_id = ?',
|
|
)
|
|
.bind(id)
|
|
.all();
|
|
expect(rows.results).toHaveLength(1);
|
|
expect(rows.results[0]).toEqual({
|
|
day: Date.parse('2026-08-27T00:00:00Z'),
|
|
total_checks: 3,
|
|
up_checks: 2,
|
|
avg_latency_ms: 200,
|
|
min_latency_ms: 100,
|
|
max_latency_ms: 300,
|
|
});
|
|
});
|
|
|
|
it('rolls up checks correctly when ids and timestamps are out of order', async () => {
|
|
const createdAt = Date.parse('2026-08-28T00:05:00Z');
|
|
const inserted = await env.DB.prepare(
|
|
"INSERT INTO monitors (name, url, method, expected_status, interval_seconds, timeout_ms, enabled, alerts_enabled, created_at, updated_at) VALUES ('API', 'https://example.com', 'GET', 200, 300, 10000, 1, 1, ?, ?)",
|
|
)
|
|
.bind(createdAt, createdAt)
|
|
.run();
|
|
const monitorId = Number(inserted.meta.last_row_id);
|
|
// The newer timestamp deliberately receives the lower primary key.
|
|
await env.DB.batch([
|
|
env.DB.prepare('INSERT INTO checks (monitor_id, ok, latency_ms, checked_at) VALUES (?, 1, 100, ?)').bind(
|
|
monitorId,
|
|
Date.parse('2026-08-28T12:00:00Z'),
|
|
),
|
|
env.DB.prepare('INSERT INTO checks (monitor_id, ok, latency_ms, checked_at) VALUES (?, 0, 300, ?)').bind(
|
|
monitorId,
|
|
Date.parse('2026-08-27T12:00:00Z'),
|
|
),
|
|
]);
|
|
|
|
await runDailyRollup(env, new Date('2026-08-28T00:05:00Z'));
|
|
await runDailyRollup(env, new Date('2026-08-29T00:05:00Z'));
|
|
|
|
const rows = await env.DB.prepare('SELECT day, total_checks, up_checks FROM monitor_daily_stats WHERE monitor_id = ? ORDER BY day')
|
|
.bind(monitorId)
|
|
.all();
|
|
expect(rows.results).toEqual([
|
|
{ day: Date.parse('2026-08-27T00:00:00Z'), total_checks: 1, up_checks: 0 },
|
|
{ day: Date.parse('2026-08-28T00:00:00Z'), total_checks: 1, up_checks: 1 },
|
|
]);
|
|
});
|
|
});
|