Filip/playwright
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

test: override dns lookup to support subdomains (#8419)

Browse source
This commit is contained in:
Yury Semikhatsky 2021-08-24 16:55:10 -07:00 committed by GitHub
parent 712a36649b
commit 338d3c8fd4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 26 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
tests/browsercontext-fetch.spec.ts
View file

@ -14,8 +14,27 @@
* limitations under the License. * limitations under the License.
*/ */
import http from 'http';
import { contextTest as it, expect } from './config/browserTest'; import { contextTest as it, expect } from './config/browserTest';
let prevAgent: http.Agent;
it.beforeAll(() => {
prevAgent = http.globalAgent;
http.globalAgent = new http.Agent({
// @ts-expect-error
lookup: (hostname, options, callback) => {
if (hostname === 'localhost' || hostname.endsWith('playwright.dev'))
callback(null, '127.0.0.1', 4);
else
throw new Error(`Failed to resolve hostname: ${hostname}`);
}
});
});
it.afterAll(() => {
http.globalAgent = prevAgent;
});
it('should work', async ({context, server}) => { it('should work', async ({context, server}) => {
// @ts-expect-error // @ts-expect-error
const response = await context._fetch(server.PREFIX + '/simple.json'); const response = await context._fetch(server.PREFIX + '/simple.json');
@ -28,11 +47,11 @@ it('should work', async ({context, server}) => {
expect(await response.text()).toBe('{"foo": "bar"}\n'); expect(await response.text()).toBe('{"foo": "bar"}\n');
}); });
it('should add session cookies to request', async ({context, server, isLinux}) => { it('should add session cookies to request', async ({context, server}) => {
await context.addCookies([{ await context.addCookies([{
name: 'username', name: 'username',
value: 'John Doe', value: 'John Doe',
domain: isLinux ? '.my.localhost' : 'localhost', domain: '.my.playwright.dev',
path: '/', path: '/',
expires: -1, expires: -1,
httpOnly: false, httpOnly: false,
@ -42,18 +61,18 @@ it('should add session cookies to request', async ({context, server, isLinux}) =
const [req] = await Promise.all([ const [req] = await Promise.all([
server.waitForRequest('/simple.json'), server.waitForRequest('/simple.json'),
// @ts-expect-error // @ts-expect-error
context._fetch(`http://${isLinux ? 'www.my.localhost' : 'localhost'}:${server.PORT}/simple.json`), context._fetch(`http://www.my.playwright.dev:${server.PORT}/simple.json`),
]); ]);
expect(req.headers.cookie).toEqual('username=John Doe'); expect(req.headers.cookie).toEqual('username=John Doe');
}); });
it('should follow redirects', async ({context, server, isLinux}) => { it('should follow redirects', async ({context, server}) => {
server.setRedirect('/redirect1', '/redirect2'); server.setRedirect('/redirect1', '/redirect2');
server.setRedirect('/redirect2', '/simple.json'); server.setRedirect('/redirect2', '/simple.json');
await context.addCookies([{ await context.addCookies([{
name: 'username', name: 'username',
value: 'John Doe', value: 'John Doe',
domain: isLinux ? '.my.localhost' : 'localhost', domain: '.my.playwright.dev',
path: '/', path: '/',
expires: -1, expires: -1,
httpOnly: false, httpOnly: false,
@ -63,10 +82,10 @@ it('should follow redirects', async ({context, server, isLinux}) => {
const [req, response] = await Promise.all([ const [req, response] = await Promise.all([
server.waitForRequest('/simple.json'), server.waitForRequest('/simple.json'),
// @ts-expect-error // @ts-expect-error
context._fetch(`http://${isLinux ? 'www.my.localhost' : 'localhost'}:${server.PORT}/redirect1`), context._fetch(`http://www.my.playwright.dev:${server.PORT}/redirect1`),
]); ]);
expect(req.headers.cookie).toEqual('username=John Doe'); expect(req.headers.cookie).toEqual('username=John Doe');
expect(response.url()).toBe(`http://${isLinux ? 'www.my.localhost' : 'localhost'}:${server.PORT}/simple.json`); expect(response.url()).toBe(`http://www.my.playwright.dev:${server.PORT}/simple.json`);
expect(await response.json()).toEqual({foo: 'bar'}); expect(await response.json()).toEqual({foo: 'bar'});
}); });