Skip to content

Commit 6263584

Browse files
committed
universe mgr added to ctx
Change-Id: Id8ff5120b5f02cd57ed42427b182763afbf65dff
1 parent ebc87c1 commit 6263584

2 files changed

Lines changed: 12 additions & 7 deletions

File tree

src/McpContext.ts

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,11 @@ import fs from 'node:fs/promises';
88
import os from 'node:os';
99
import path from 'node:path';
1010

11-
import {extractUrlLikeFromDevToolsTitle, urlsEqual} from './DevtoolsUtils.js';
11+
import {
12+
extractUrlLikeFromDevToolsTitle,
13+
urlsEqual,
14+
UniverseManager,
15+
} from './DevtoolsUtils.js';
1216
import type {ListenerMap} from './PageCollector.js';
1317
import {NetworkCollector, ConsoleCollector} from './PageCollector.js';
1418
import {Locator} from './third_party/index.js';
@@ -116,6 +120,7 @@ export class McpContext implements Context {
116120

117121
#nextSnapshotId = 1;
118122
#traceResults: TraceResult[] = [];
123+
#universeManager: UniverseManager;
119124

120125
#locatorClass: typeof Locator;
121126
#options: McpContextOptions;
@@ -130,6 +135,7 @@ export class McpContext implements Context {
130135
this.logger = logger;
131136
this.#locatorClass = locatorClass;
132137
this.#options = options;
138+
this.#universeManager = new UniverseManager(this.browser);
133139

134140
this.#networkCollector = new NetworkCollector(this.browser);
135141

@@ -156,11 +162,13 @@ export class McpContext implements Context {
156162

157163
async #init() {
158164
const pages = await this.createPagesSnapshot();
165+
await this.#universeManager.init(pages);
159166
await this.#networkCollector.init(pages);
160167
await this.#consoleCollector.init(pages);
161168
}
162169

163170
dispose() {
171+
this.#universeManager.dispose();
164172
this.#networkCollector.dispose();
165173
this.#consoleCollector.dispose();
166174
}

src/manual-perf-test.ts

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -93,12 +93,9 @@ async function run() {
9393

9494
// Check trace content (basic check)
9595
const traceResult = traces[0];
96-
// We assume traceResult is valid if we got here and storeTraceRecording was called.
97-
// startTrace.handler calls storeTraceRecording only on success.
98-
console.log(
99-
'Trace result summary:',
100-
lines.find(l => l.includes('Trace duration')),
101-
);
96+
console.log('--- Response Lines ---');
97+
console.log(lines.join('\n'));
98+
console.log('----------------------');
10299

103100
console.log('SUCCESS: Trace recorded and stopped automatically.');
104101
} catch (err) {

0 commit comments

Comments
 (0)