mirror of
https://github.com/microsoft/playwright.git
synced 2025-06-26 21:40:17 +00:00
aeba083da0
Fixes #24144. Previously, we only matched by url, which confuses GET and HEAD requests where the latter is usually zero-sized. Also make sure that resources are sorted by their monotonicTime, since that's not always the case in the trace file, where they are sorted by the "response body retrieved" time.
65 lines
2.3 KiB
TypeScript
65 lines
2.3 KiB
TypeScript
/**
|
|
* Copyright (c) Microsoft Corporation.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
import type { FrameSnapshot, ResourceSnapshot } from '@trace/snapshot';
|
|
import { rewriteURLForCustomProtocol, SnapshotRenderer } from './snapshotRenderer';
|
|
|
|
export class SnapshotStorage {
|
|
private _resources: ResourceSnapshot[] = [];
|
|
private _frameSnapshots = new Map<string, {
|
|
raw: FrameSnapshot[],
|
|
renderers: SnapshotRenderer[]
|
|
}>();
|
|
|
|
addResource(resource: ResourceSnapshot): void {
|
|
resource.request.url = rewriteURLForCustomProtocol(resource.request.url);
|
|
this._resources.push(resource);
|
|
}
|
|
|
|
addFrameSnapshot(snapshot: FrameSnapshot) {
|
|
for (const override of snapshot.resourceOverrides)
|
|
override.url = rewriteURLForCustomProtocol(override.url);
|
|
let frameSnapshots = this._frameSnapshots.get(snapshot.frameId);
|
|
if (!frameSnapshots) {
|
|
frameSnapshots = {
|
|
raw: [],
|
|
renderers: [],
|
|
};
|
|
this._frameSnapshots.set(snapshot.frameId, frameSnapshots);
|
|
if (snapshot.isMainFrame)
|
|
this._frameSnapshots.set(snapshot.pageId, frameSnapshots);
|
|
}
|
|
frameSnapshots.raw.push(snapshot);
|
|
const renderer = new SnapshotRenderer(this._resources, frameSnapshots.raw, frameSnapshots.raw.length - 1);
|
|
frameSnapshots.renderers.push(renderer);
|
|
return renderer;
|
|
}
|
|
|
|
snapshotByName(pageOrFrameId: string, snapshotName: string): SnapshotRenderer | undefined {
|
|
const snapshot = this._frameSnapshots.get(pageOrFrameId);
|
|
return snapshot?.renderers.find(r => r.snapshotName === snapshotName);
|
|
}
|
|
|
|
snapshotsForTest() {
|
|
return [...this._frameSnapshots.keys()];
|
|
}
|
|
|
|
finalize() {
|
|
// Resources are not necessarily sorted in the trace file, so sort them now.
|
|
this._resources.sort((a, b) => (a._monotonicTime || 0) - (b._monotonicTime || 0));
|
|
}
|
|
}
|