admin-valentin/agent-ecosystem
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test: isolate runtime usage stubs from process table

Browse source
This commit is contained in:
777genius 2026-05-30 15:46:27 +03:00
parent 61e2678a5d
commit 28a55416ca
1 changed files with 25 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
test/main/services/team/TeamAgentLaunchMatrix.safe-e2e.test.ts
View file

@ -78,6 +78,12 @@ const WORKSPACE_TRUST_TEST_ENV_NAMES = [
type WorkspaceTrustTestEnvName = (typeof WORKSPACE_TRUST_TEST_ENV_NAMES)[number];
type RuntimeUsageStatsForTest = { rssBytes: number; cpuPercent?: number };
type RuntimeUsageStatsStubTarget = {
readRuntimeProcessRowsForUsageSnapshot: () => Promise<null>;
readProcessUsageStatsByPid: (
pids: readonly number[]
) => Promise<Map<number, RuntimeUsageStatsForTest>>;
};
function createRuntimeUsageStatsMap(
entries: readonly (readonly [number, number])[]
@ -85,6 +91,19 @@ function createRuntimeUsageStatsMap(
return new Map(entries.map(([pid, rssBytes]) => [pid, { rssBytes }]));
}
function stubRuntimeUsageStatsByPid(
service: TeamProvisioningService,
entries: readonly (readonly [number, number])[]
): void {
const statsByPid = createRuntimeUsageStatsMap(entries);
const target = service as unknown as RuntimeUsageStatsStubTarget;
target.readRuntimeProcessRowsForUsageSnapshot = async () => null;
target.readProcessUsageStatsByPid = async (pids: readonly number[]) => {
const requestedPids = new Set(pids);
return new Map([...statsByPid].filter(([pid]) => requestedPids.has(pid)));
};
}
function createRuntimeUsageStatsByPid(
pids: readonly number[]
): Map<number, RuntimeUsageStatsForTest> {
@ -4121,8 +4140,7 @@ describe('Team agent launch matrix safe e2e', () => {
},
],
]);
(svc as any).readProcessUsageStatsByPid = async () =>
createRuntimeUsageStatsMap([[sharedHostPid, 183.9 * 1024 * 1024]]);
stubRuntimeUsageStatsByPid(svc, [[sharedHostPid, 183.9 * 1024 * 1024]]);
await waitForCondition(async () => {
const snapshot = await svc.getTeamAgentRuntimeSnapshot(teamName);
@ -4227,8 +4245,7 @@ describe('Team agent launch matrix safe e2e', () => {
},
],
]);
(svc as any).readProcessUsageStatsByPid = async () =>
createRuntimeUsageStatsMap([[sharedHostPid, sharedRssBytes]]);
stubRuntimeUsageStatsByPid(svc, [[sharedHostPid, sharedRssBytes]]);
const runtimeSnapshot = await svc.getTeamAgentRuntimeSnapshot(teamName);
@ -4329,8 +4346,7 @@ describe('Team agent launch matrix safe e2e', () => {
},
],
]);
(svc as any).readProcessUsageStatsByPid = async () =>
createRuntimeUsageStatsMap([[sharedHostPid, sharedRssBytes]]);
stubRuntimeUsageStatsByPid(svc, [[sharedHostPid, sharedRssBytes]]);
const runtimeSnapshot = await svc.getTeamAgentRuntimeSnapshot(teamName);
@ -4451,8 +4467,7 @@ describe('Team agent launch matrix safe e2e', () => {
},
],
]);
(svc as any).readProcessUsageStatsByPid = async () =>
createRuntimeUsageStatsMap([[sharedHostPid, sharedRssBytes]]);
stubRuntimeUsageStatsByPid(svc, [[sharedHostPid, sharedRssBytes]]);
const runtimeSnapshot = await svc.getTeamAgentRuntimeSnapshot(teamName);
@ -4535,8 +4550,7 @@ describe('Team agent launch matrix safe e2e', () => {
},
],
]);
(restartedService as any).readProcessUsageStatsByPid = async () =>
createRuntimeUsageStatsMap([[sharedHostPid, 188.4 * 1024 * 1024]]);
stubRuntimeUsageStatsByPid(restartedService, [[sharedHostPid, 188.4 * 1024 * 1024]]);
const runtimeSnapshot = await restartedService.getTeamAgentRuntimeSnapshot(teamName);
@ -5294,8 +5308,7 @@ describe('Team agent launch matrix safe e2e', () => {
},
],
]);
(svc as any).readProcessUsageStatsByPid = async () =>
createRuntimeUsageStatsMap([[sharedHostPid, sharedRssBytes]]);
stubRuntimeUsageStatsByPid(svc, [[sharedHostPid, sharedRssBytes]]);
const runtimeSnapshot = await svc.getTeamAgentRuntimeSnapshot(teamName);