84 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			TypeScript
		
	
	
	
			
		
		
	
	
			84 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			TypeScript
		
	
	
	
/**
 | 
						|
 * Copyright 2018 Google Inc. All rights reserved.
 | 
						|
 * Modifications 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 './playwright.fixtures';
 | 
						|
 | 
						|
it('should work', async({page, server}) => {
 | 
						|
  await page.setExtraHTTPHeaders({
 | 
						|
    foo: 'bar',
 | 
						|
    baz: undefined,
 | 
						|
  });
 | 
						|
  const [request] = await Promise.all([
 | 
						|
    server.waitForRequest('/empty.html'),
 | 
						|
    page.goto(server.EMPTY_PAGE),
 | 
						|
  ]);
 | 
						|
  expect(request.headers['foo']).toBe('bar');
 | 
						|
  expect(request.headers['baz']).toBe(undefined);
 | 
						|
});
 | 
						|
 | 
						|
it('should work with redirects', async({page, server}) => {
 | 
						|
  server.setRedirect('/foo.html', '/empty.html');
 | 
						|
  await page.setExtraHTTPHeaders({
 | 
						|
    foo: 'bar'
 | 
						|
  });
 | 
						|
  const [request] = await Promise.all([
 | 
						|
    server.waitForRequest('/empty.html'),
 | 
						|
    page.goto(server.PREFIX + '/foo.html'),
 | 
						|
  ]);
 | 
						|
  expect(request.headers['foo']).toBe('bar');
 | 
						|
});
 | 
						|
 | 
						|
it('should work with extra headers from browser context', async({browser, server}) => {
 | 
						|
  const context = await browser.newContext();
 | 
						|
  await context.setExtraHTTPHeaders({
 | 
						|
    'foo': 'bar',
 | 
						|
  });
 | 
						|
  const page = await context.newPage();
 | 
						|
  const [request] = await Promise.all([
 | 
						|
    server.waitForRequest('/empty.html'),
 | 
						|
    page.goto(server.EMPTY_PAGE),
 | 
						|
  ]);
 | 
						|
  await context.close();
 | 
						|
  expect(request.headers['foo']).toBe('bar');
 | 
						|
});
 | 
						|
 | 
						|
it('should override extra headers from browser context', async({browser, server}) => {
 | 
						|
  const context = await browser.newContext({
 | 
						|
    extraHTTPHeaders: { 'fOo': 'bAr', 'baR': 'foO' },
 | 
						|
  });
 | 
						|
  const page = await context.newPage();
 | 
						|
  await page.setExtraHTTPHeaders({
 | 
						|
    'Foo': 'Bar'
 | 
						|
  });
 | 
						|
  const [request] = await Promise.all([
 | 
						|
    server.waitForRequest('/empty.html'),
 | 
						|
    page.goto(server.EMPTY_PAGE),
 | 
						|
  ]);
 | 
						|
  await context.close();
 | 
						|
  expect(request.headers['foo']).toBe('Bar');
 | 
						|
  expect(request.headers['bar']).toBe('foO');
 | 
						|
});
 | 
						|
 | 
						|
it('should throw for non-string header values', async({browser, page}) => {
 | 
						|
  const error1 = await page.setExtraHTTPHeaders({ 'foo': 1 as any }).catch(e => e);
 | 
						|
  expect(error1.message).toContain('Expected value of header "foo" to be String, but "number" is found.');
 | 
						|
  const error2 = await page.context().setExtraHTTPHeaders({ 'foo': true as any }).catch(e => e);
 | 
						|
  expect(error2.message).toContain('Expected value of header "foo" to be String, but "boolean" is found.');
 | 
						|
  const error3 = await browser.newContext({ extraHTTPHeaders: { 'foo': null as any } }).catch(e => e);
 | 
						|
  expect(error3.message).toContain('Expected value of header "foo" to be String, but "object" is found.');
 | 
						|
});
 |