1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
/**
* @license
* Copyright 2022 Google Inc.
* SPDX-License-Identifier: Apache-2.0
*/
import expect from 'expect';
import type {CdpBrowser} from 'puppeteer-core/internal/cdp/Browser.js';
import {getTestState, launch} from '../mocha-utils.js';
import {attachFrame} from '../utils.js';
describe('TargetManager', () => {
/* We use a special browser for this test as we need the --site-per-process flag */
let state: Awaited<ReturnType<typeof launch>> & {
browser: CdpBrowser;
};
beforeEach(async () => {
const {defaultBrowserOptions} = await getTestState({
skipLaunch: true,
});
state = (await launch(
Object.assign({}, defaultBrowserOptions, {
args: (defaultBrowserOptions.args || []).concat([
'--site-per-process',
'--remote-debugging-port=21222',
'--host-rules=MAP * 127.0.0.1',
]),
}),
{createPage: false}
)) as Awaited<ReturnType<typeof launch>> & {
browser: CdpBrowser;
};
});
afterEach(async () => {
await state.close();
});
// CDP-specific test.
it('should handle targets', async () => {
const {server, context, browser} = state;
const targetManager = browser._targetManager();
expect(targetManager.getAvailableTargets().size).toBe(3);
expect(await context.pages()).toHaveLength(0);
expect(targetManager.getAvailableTargets().size).toBe(3);
const page = await context.newPage();
expect(await context.pages()).toHaveLength(1);
expect(targetManager.getAvailableTargets().size).toBe(5);
await page.goto(server.EMPTY_PAGE);
expect(await context.pages()).toHaveLength(1);
expect(targetManager.getAvailableTargets().size).toBe(5);
// attach a local iframe.
let framePromise = page.waitForFrame(frame => {
return frame.url().endsWith('/empty.html');
});
await attachFrame(page, 'frame1', server.EMPTY_PAGE);
await framePromise;
expect(await context.pages()).toHaveLength(1);
expect(targetManager.getAvailableTargets().size).toBe(5);
expect(page.frames()).toHaveLength(2);
// // attach a remote frame iframe.
framePromise = page.waitForFrame(frame => {
return frame.url() === server.CROSS_PROCESS_PREFIX + '/empty.html';
});
await attachFrame(
page,
'frame2',
server.CROSS_PROCESS_PREFIX + '/empty.html'
);
await framePromise;
expect(await context.pages()).toHaveLength(1);
expect(targetManager.getAvailableTargets().size).toBe(6);
expect(page.frames()).toHaveLength(3);
framePromise = page.waitForFrame(frame => {
return frame.url() === server.CROSS_PROCESS_PREFIX + '/empty.html';
});
await attachFrame(
page,
'frame3',
server.CROSS_PROCESS_PREFIX + '/empty.html'
);
await framePromise;
expect(await context.pages()).toHaveLength(1);
expect(targetManager.getAvailableTargets().size).toBe(7);
expect(page.frames()).toHaveLength(4);
});
});
|