109 lines
4.7 KiB
TypeScript
109 lines
4.7 KiB
TypeScript
import { describe, expect, it, jest } from "@jest/globals";
|
|
import { GeoChecksPipeline } from "../../../src/worker/worker.check-pipeline.ts";
|
|
import type { Monitor } from "../../../src/domain/monitors/monitor.types.ts";
|
|
import { createMockLogger } from "../../helpers/createMockLogger.ts";
|
|
|
|
const makeMonitor = (overrides?: Partial<Monitor>): Monitor =>
|
|
({
|
|
id: "m1",
|
|
teamId: "team",
|
|
type: "http",
|
|
interval: 60000,
|
|
status: "up",
|
|
geoCheckEnabled: false,
|
|
geoCheckLocations: [],
|
|
...overrides,
|
|
}) as Monitor;
|
|
|
|
const activeWindow = () => {
|
|
const now = Date.now();
|
|
return { active: true, start: new Date(now - 1000).toISOString(), end: new Date(now + 1000).toISOString(), repeat: 0 };
|
|
};
|
|
|
|
// ── GeoChecksPipeline ──────────────────────────────────────────────────────────
|
|
|
|
const createGeoPipeline = (overrides?: Record<string, any>) => {
|
|
const defaults = {
|
|
logger: createMockLogger(),
|
|
geoChecksService: { buildGeoCheck: jest.fn().mockResolvedValue(null) },
|
|
buffer: { addGeoCheckToBuffer: jest.fn() },
|
|
maintenanceWindowsRepository: { findByMonitorId: jest.fn().mockResolvedValue([]) },
|
|
...overrides,
|
|
};
|
|
const pipeline = new GeoChecksPipeline(
|
|
defaults.maintenanceWindowsRepository as any,
|
|
defaults.geoChecksService as any,
|
|
defaults.buffer as any,
|
|
defaults.logger as any
|
|
);
|
|
return { pipeline, defaults };
|
|
};
|
|
|
|
const geoMonitor = (overrides?: Partial<Monitor>) =>
|
|
makeMonitor({ geoCheckEnabled: true, type: "http", geoCheckLocations: ["us-east"], ...overrides });
|
|
|
|
describe("GeoChecksPipeline", () => {
|
|
it("throws when monitor id is missing", async () => {
|
|
const { pipeline } = createGeoPipeline();
|
|
await expect(pipeline.run({} as Monitor)).rejects.toThrow("No monitor id");
|
|
});
|
|
|
|
it("returns null when geoCheckEnabled is false", async () => {
|
|
const { pipeline, defaults } = createGeoPipeline();
|
|
const result = await pipeline.run(makeMonitor({ geoCheckEnabled: false }));
|
|
expect(result).toBeNull();
|
|
expect(defaults.geoChecksService.buildGeoCheck).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("returns null when monitor type does not support geo checks", async () => {
|
|
const { pipeline, defaults } = createGeoPipeline();
|
|
await pipeline.run(geoMonitor({ type: "hardware" }));
|
|
expect(defaults.logger.debug).toHaveBeenCalledWith(expect.objectContaining({ message: expect.stringContaining("does not support geo checks") }));
|
|
expect(defaults.geoChecksService.buildGeoCheck).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("warns when geoCheckLocations is empty", async () => {
|
|
const { pipeline, defaults } = createGeoPipeline();
|
|
await pipeline.run(geoMonitor({ geoCheckLocations: [] }));
|
|
expect(defaults.logger.warn).toHaveBeenCalledWith(expect.objectContaining({ message: expect.stringContaining("No geo check locations") }));
|
|
});
|
|
|
|
it("warns when geoCheckLocations is undefined", async () => {
|
|
const { pipeline, defaults } = createGeoPipeline();
|
|
await pipeline.run(geoMonitor({ geoCheckLocations: undefined as any }));
|
|
expect(defaults.logger.warn).toHaveBeenCalledWith(expect.objectContaining({ message: expect.stringContaining("No geo check locations") }));
|
|
});
|
|
|
|
it("skips when in maintenance window", async () => {
|
|
const { pipeline, defaults } = createGeoPipeline({
|
|
maintenanceWindowsRepository: { findByMonitorId: jest.fn().mockResolvedValue([activeWindow()]) },
|
|
});
|
|
await pipeline.run(geoMonitor());
|
|
expect(defaults.logger.debug).toHaveBeenCalledWith(expect.objectContaining({ message: expect.stringContaining("maintenance window") }));
|
|
expect(defaults.geoChecksService.buildGeoCheck).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("warns when buildGeoCheck returns null", async () => {
|
|
const { pipeline, defaults } = createGeoPipeline();
|
|
await pipeline.run(geoMonitor());
|
|
expect(defaults.logger.warn).toHaveBeenCalledWith(expect.objectContaining({ message: expect.stringContaining("No geo check could be built") }));
|
|
});
|
|
|
|
it("adds geo check to buffer on success", async () => {
|
|
const geoCheck = { id: "gc-1", monitorId: "m1" };
|
|
const { pipeline, defaults } = createGeoPipeline({
|
|
geoChecksService: { buildGeoCheck: jest.fn().mockResolvedValue(geoCheck) },
|
|
});
|
|
await pipeline.run(geoMonitor());
|
|
expect(defaults.buffer.addGeoCheckToBuffer).toHaveBeenCalledWith(geoCheck);
|
|
expect(defaults.logger.debug).toHaveBeenCalledWith(expect.objectContaining({ message: expect.stringContaining("Geo check job executed") }));
|
|
});
|
|
|
|
it("propagates errors (the job glue handles isolation)", async () => {
|
|
const { pipeline } = createGeoPipeline({
|
|
geoChecksService: { buildGeoCheck: jest.fn().mockRejectedValue(new Error("api timeout")) },
|
|
});
|
|
await expect(pipeline.run(geoMonitor())).rejects.toThrow("api timeout");
|
|
});
|
|
});
|