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