63 lines
3.6 KiB
TypeScript
Vendored
63 lines
3.6 KiB
TypeScript
Vendored
import { describe, expect, it } from "vitest";
|
|
import { CONCURRENT_INDEX_QUERY_TIMEOUT_SECS, frontendQueryTimeoutDelayMs, frontendQueryTimeoutSecsForSql, queryTimeoutSecsForConcurrentIndex, queryTimeoutSecsForConnection } from "@/lib/sql/queryTimeout";
|
|
|
|
describe("queryTimeout", () => {
|
|
it("gives CREATE INDEX CONCURRENTLY a dedicated long budget instead of the 60s default", () => {
|
|
expect(CONCURRENT_INDEX_QUERY_TIMEOUT_SECS).toBe(1800);
|
|
expect(CONCURRENT_INDEX_QUERY_TIMEOUT_SECS).toBeGreaterThan(30);
|
|
expect(frontendQueryTimeoutDelayMs(CONCURRENT_INDEX_QUERY_TIMEOUT_SECS)).toBe(1_800_000);
|
|
});
|
|
|
|
it.each([
|
|
["non-concurrent", 30, false, 30],
|
|
["non-concurrent", 3600, false, 3600],
|
|
["concurrent + 0 (unlimited)", 0, true, 0],
|
|
["concurrent + 30 (below floor)", 30, true, 1800],
|
|
["concurrent + 600 (below floor)", 600, true, 1800],
|
|
["concurrent + 1800 (floor)", 1800, true, 1800],
|
|
["concurrent + 3600 (above floor)", 3600, true, 3600],
|
|
] as const)("queryTimeoutSecsForConcurrentIndex preserves %s", (_name, configured, concurrent, expected) => {
|
|
expect(queryTimeoutSecsForConcurrentIndex(configured, concurrent)).toBe(expected);
|
|
});
|
|
it("lets PostgreSQL row queries use the backend inactivity timeout", () => {
|
|
expect(frontendQueryTimeoutSecsForSql("SELECT * FROM sample_records LIMIT 2000", "postgres", 30)).toBe(0);
|
|
expect(frontendQueryTimeoutSecsForSql("/* page */\nWITH rows AS (SELECT 1) SELECT * FROM rows", "postgres", 30)).toBe(0);
|
|
expect(frontendQueryTimeoutSecsForSql("UPDATE sample_records SET state = 'ready' RETURNING id", "postgres", 30)).toBe(0);
|
|
});
|
|
|
|
it("uses conservative package splitting for an unresolved openGauss mode", () => {
|
|
const packageSpec = `CREATE OR REPLACE PACKAGE pkg_utils AS
|
|
FUNCTION get_version RETURN VARCHAR2;
|
|
END pkg_utils;`;
|
|
expect(frontendQueryTimeoutSecsForSql(`${packageSpec}\n/\nSELECT 1;`, "opengauss", 30)).toBe(60);
|
|
});
|
|
|
|
it("keeps the frontend guard for non-row PostgreSQL statements", () => {
|
|
expect(frontendQueryTimeoutSecsForSql("UPDATE sample_records SET state = 'ready'", "postgres", 30)).toBe(30);
|
|
expect(frontendQueryTimeoutSecsForSql("INSERT INTO sample_records(note) VALUES ('RETURNING is text')", "postgres", 30)).toBe(30);
|
|
expect(frontendQueryTimeoutSecsForSql("UPDATE sample_records SET note = 'ready' /* RETURNING */", "postgres", 30)).toBe(30);
|
|
});
|
|
|
|
it("keeps the existing frontend guard for other database types", () => {
|
|
expect(frontendQueryTimeoutSecsForSql("SELECT * FROM sample_records LIMIT 2000", "mysql", 30)).toBe(30);
|
|
expect(frontendQueryTimeoutSecsForSql("SELECT SLEEP(30)", "mysql", 10)).toBe(10);
|
|
expect(queryTimeoutSecsForConnection({ query_timeout_secs: undefined })).toBe(60);
|
|
});
|
|
|
|
it("does not schedule frontend timeouts beyond the browser timer limit", () => {
|
|
expect(frontendQueryTimeoutDelayMs(60)).toBe(60_000);
|
|
expect(frontendQueryTimeoutDelayMs(11_401_200)).toBeUndefined();
|
|
expect(frontendQueryTimeoutDelayMs(0)).toBeUndefined();
|
|
});
|
|
|
|
it("uses the global timeout only for inheriting connections", () => {
|
|
expect(queryTimeoutSecsForConnection({ query_timeout_secs: 30, query_timeout_inherit: true }, 12)).toBe(12);
|
|
expect(queryTimeoutSecsForConnection({ query_timeout_secs: 30, query_timeout_inherit: false }, 12)).toBe(30);
|
|
expect(queryTimeoutSecsForConnection({ query_timeout_secs: 0, query_timeout_inherit: false }, 12)).toBe(0);
|
|
});
|
|
|
|
it("falls back safely when an inherited global timeout is invalid", () => {
|
|
expect(queryTimeoutSecsForConnection({ query_timeout_inherit: true }, Number.NaN)).toBe(60);
|
|
});
|
|
});
|