test(server): clean up valkey stubbing in server/main.test.ts

This commit is contained in:
Tyler Gillispie 2026-08-23 21:24:21 -07:00
parent 8390dcac3d
commit 5db9108ecd

View File

@ -76,40 +76,31 @@ Deno.test("Tier 1 & 2: GET /api/forward-auth - Missing session", async () => {
headers: { "X-Forwarded-Host": "test.app.local" }, headers: { "X-Forwarded-Host": "test.app.local" },
}); });
let valkeyStub: any; const valkeyStub = stub(valkey, "get", (key: any) => {
if ((valkey.get as any).restore) { const k = String(key);
valkeyStub = valkey.get as any; if (k.startsWith("auth:app_by_host:")) {
} else { return Promise.resolve(
valkeyStub = stub(valkey, "get", (key: any) => { JSON.stringify({ id: "app-id-1", name: "test-app" }),
const k = String(key); );
if (k.startsWith("auth:app_by_host:")) { }
return Promise.resolve( return Promise.resolve(null);
JSON.stringify({ id: "app-id-1", name: "test-app" }), });
);
}
return Promise.resolve(null);
});
}
const res = await app.request(req); const res = await app.request(req);
assertEquals(res.status, 401); assertEquals(res.status, 401);
valkeyStub.restore(); valkeyStub.restore();
}); });
Deno.test("Tier 1 & 2: GET /api/forward-auth - Expired session", async () => { Deno.test("Tier 1 & 2: GET /api/forward-auth - Expired session", async () => {
let valkeyStub: any; const valkeyStub = stub(valkey, "get", (key: any) => {
if ((valkey.get as any).restore) { const k = String(key);
valkeyStub = valkey.get as any; if (k.startsWith("auth:app_by_host:")) {
} else { return Promise.resolve(
valkeyStub = stub(valkey, "get", (key: any) => { JSON.stringify({ id: "app-id-1", name: "test-app" }),
const k = String(key); );
if (k.startsWith("auth:app_by_host:")) { }
return Promise.resolve( return Promise.resolve(null);
JSON.stringify({ id: "app-id-1", name: "test-app" }), });
);
}
return Promise.resolve(null);
});
}
const req = new Request("http://localhost/api/forward-auth", { const req = new Request("http://localhost/api/forward-auth", {
headers: { headers: {
@ -131,22 +122,17 @@ Deno.test("Tier 1 & 2: GET /api/forward-auth - Suspended account", async () => {
setMockSql(() => Promise.resolve([mockUser])); setMockSql(() => Promise.resolve([mockUser]));
let valkeyStub: any; const valkeyStub = stub(valkey, "get", (key: any) => {
if ((valkey.get as any).restore) { const k = String(key);
valkeyStub = valkey.get as any; if (k.startsWith("auth:app_by_host:")) {
} else {
valkeyStub = stub(valkey, "get", (key: any) => {
const k = String(key);
if (k.startsWith("auth:app_by_host:")) {
return Promise.resolve(
JSON.stringify({ id: "app-id-1", name: "test-app" }),
);
}
return Promise.resolve( return Promise.resolve(
JSON.stringify({ uuid: "user-id-1", username: "alice" }), JSON.stringify({ id: "app-id-1", name: "test-app" }),
); );
}); }
} return Promise.resolve(
JSON.stringify({ uuid: "user-id-1", username: "alice" }),
);
});
const req = new Request("http://localhost/api/forward-auth", { const req = new Request("http://localhost/api/forward-auth", {
headers: { headers: {
@ -177,17 +163,11 @@ Deno.test("Tier 3: ValidateSession ConnectRPC - Default-Deny", async () => {
}); });
// Re-define valkey stub // Re-define valkey stub
const valkeyStub = stub(valkey, "get", () => {
let valkeyStub: any; return Promise.resolve(
if ((valkey.get as any).restore) { JSON.stringify({ uuid: "user-1", username: "alice" }),
valkeyStub = valkey.get as any; );
} else { });
valkeyStub = stub(valkey, "get", () => {
return Promise.resolve(
JSON.stringify({ uuid: "user-1", username: "alice" }),
);
});
}
let auditCalled = false; let auditCalled = false;
const originalAudit = auditWrapper.auditLog; const originalAudit = auditWrapper.auditLog;
@ -246,16 +226,11 @@ Deno.test("Tier 3: ValidateSession ConnectRPC - Valid RBAC Grant", async () => {
} }
}); });
let valkeyStub: any; const valkeyStub = stub(valkey, "get", () => {
if ((valkey.get as any).restore) { return Promise.resolve(
valkeyStub = valkey.get as any; JSON.stringify({ uuid: "user-1", username: "alice" }),
} else { );
valkeyStub = stub(valkey, "get", () => { });
return Promise.resolve(
JSON.stringify({ uuid: "user-1", username: "alice" }),
);
});
}
const req = new Request( const req = new Request(
"http://localhost/auth.v1.AuthService/ValidateSession", "http://localhost/auth.v1.AuthService/ValidateSession",