chromium/src/third_party/blink/web_tests/external/wpt/fetch/metadata/resources/redirectTestHelper.sub.js

function createVideoElement() {
  let el = document.createElement('video');
  el.src = '/media/movie_5.mp4';
  el.setAttribute('controls', '');
  el.setAttribute('crossorigin', '');
  return el;
}

function createTrack() {
  let el = document.createElement('track');
  el.setAttribute('default', '');
  el.setAttribute('kind', 'captions');
  el.setAttribute('srclang', 'en');
  return el;
}

let secureRedirectURL = 'https://{{host}}:{{ports[https][0]}}/fetch/api/resources/redirect.py?location=';
let insecureRedirectURL = 'http://{{host}}:{{ports[http][0]}}/fetch/api/resources/redirect.py?location=';
let secureTestURL = 'https://{{host}}:{{ports[https][0]}}/fetch/metadata/';
let insecureTestURL = 'http://{{host}}:{{ports[http][0]}}/fetch/metadata/';

// Helper to craft an URL that will go from HTTPS => HTTP => HTTPS to
// simulate us downgrading then upgrading again during the same redirect chain.
function MultipleRedirectTo(partialPath) {
  let finalURL = insecureRedirectURL + encodeURIComponent(secureTestURL + partialPath);
  return secureRedirectURL + encodeURIComponent(finalURL);
}

// Helper to craft an URL that will go from HTTP => HTTPS to simulate upgrading a
// given request.
function upgradeRedirectTo(partialPath) {
  return insecureRedirectURL + encodeURIComponent(secureTestURL + partialPath);
}

// Helper to craft an URL that will go from HTTPS => HTTP to simulate downgrading a
// given request.
function downgradeRedirectTo(partialPath) {
  return secureRedirectURL + encodeURIComponent(insecureTestURL + partialPath);
}

// Helper to test the behavior of the `prefetch` Link type [1]. Because the the
// behavior under test is optional [2], this function should only be used in
// tests which have been denoted as "optional" [3].
//
// [1] https://html.spec.whatwg.org/#link-type-prefetch
// [2] https://w3c.github.io/resource-hints/#load-and-error-events
// [3] https://web-platform-tests.org/writing-tests/file-names.html
function testPrefetch(nonce, testNamePrefix, urlHelperMethod, expectedResults) {
  async_test(t => {
    let key = 'prefetch' + nonce;
    let e = document.createElement('link');
    e.rel = 'prefetch';
    e.crossOrigin = 'anonymous';
    e.href = urlHelperMethod('resources/record-header.py?file=' + key) + '&simple=true';
    e.onload = t.step_func(e => {
      let expectation = { ...expectedResults };
      if (expectation['mode'] != '')
        expectation['mode'] = 'cors';
      fetch('/fetch/metadata/resources/record-header.py?retrieve=true&file=' + key)
        .then(t.step_func(response => response.text()))
        .then(t.step_func_done(text => assert_header_equals(text, expectation, testNamePrefix + ' prefetch => No headers')))
        .catch(t.unreached_func('Fetching and verifying the results should succeed.'));
    });
    e.onerror = t.unreached_func();
    document.head.appendChild(e);
  }, testNamePrefix + ' prefetch => No headers');
}

// Helper to run common redirect test cases that don't require special setup on
// the test page itself.
function RunCommonRedirectTests(testNamePrefix, urlHelperMethod, expectedResults) {
  async_test(t => {
    let i = document.createElement('iframe');
    i.src = urlHelperMethod('resources/post-to-owner.py?iframe-navigation' + nonce);
    window.addEventListener('message', t.step_func(e => {
      if (e.source != i.contentWindow) {
        return;
      }
      let expectation = { ...expectedResults };
      if (expectation['mode'] != '')
        expectation['mode'] = 'navigate';
      if (expectation['dest'] == 'font')
        expectation['dest'] = 'iframe';
      assert_header_equals(e.data, expectation, testNamePrefix + ' iframe');
      t.done();
    }));

    document.body.appendChild(i);
  }, testNamePrefix + ' iframe');

  async_test(t => {
    let testWindow = window.open(urlHelperMethod('resources/post-to-owner.py?top-level-navigation' + nonce));
    t.add_cleanup(_ => testWindow.close());
    window.addEventListener('message', t.step_func(e => {
      if (e.source != testWindow) {
        return;
      }

      let expectation = { ...expectedResults };
      if (expectation['mode'] != '')
        expectation['mode'] = 'navigate';
      if (expectation['dest'] == 'font')
        expectation['dest'] = 'document';
      assert_header_equals(e.data, expectation, testNamePrefix + ' top level navigation');
      t.done();
    }));
  }, testNamePrefix + ' top level navigation');

  promise_test(t => {
    return new Promise((resolve, reject) => {
      let key = 'embed-https-redirect' + nonce;
      let e = document.createElement('embed');
      e.src = urlHelperMethod('resources/record-header.py?file=' + key);
      e.onload = e => {
        let expectation = { ...expectedResults };
        if (expectation['mode'] != '')
          expectation['mode'] = 'navigate';
        if (expectation['dest'] == 'font')
          expectation['dest'] = 'embed';
        fetch('/fetch/metadata/resources/record-header.py?retrieve=true&file=' + key)
          .then(response => response.text())
          .then(t.step_func(text => assert_header_equals(text, expectation, testNamePrefix + ' embed')))
          .then(resolve)
          .catch(e => reject(e));
      };
      document.body.appendChild(e);
    });
  }, testNamePrefix + ' embed');

  promise_test(t => {
    let key = 'fetch-redirect' + nonce;
    let expectation = { ...expectedResults };
    if (expectation['mode'] != '')
      expectation['mode'] = 'cors';
    if (expectation['dest'] == 'font')
      expectation['dest'] = 'empty';
    return fetch(urlHelperMethod('resources/echo-as-json.py?' + key))
      .then(r => r.json())
      .then(j => {assert_header_equals(j, expectation, testNamePrefix + ' fetch() api');});
  }, testNamePrefix + ' fetch() api');

  promise_test(t => {
    return new Promise((resolve, reject) => {
      let key = 'object-https-redirect' + nonce;
      let e = document.createElement('object');
      e.data = urlHelperMethod('resources/record-header.py?file=' + key);
      e.onload = e => {
        let expectation = { ...expectedResults };
        if (expectation['mode'] != '')
          expectation['mode'] = 'navigate';
        if (expectation['dest'] == 'font')
          expectation['dest'] = 'object';
        fetch('/fetch/metadata/resources/record-header.py?retrieve=true&file=' + key)
          .then(response => response.text())
          .then(t.step_func(text => assert_header_equals(text, expectation, testNamePrefix + ' object')))
          .then(resolve)
          .catch(e => reject(e));
      };
      document.body.appendChild(e);
    });
  }, testNamePrefix + ' object');

  if (document.createElement('link').relList.supports('preload')) {
    async_test(t => {
      let key = 'preload' + nonce;
      let e = document.createElement('link');
      e.rel = 'preload';
      e.href = urlHelperMethod('resources/record-header.py?file=' + key);
      e.setAttribute('as', 'track');
      e.onload = e.onerror = t.step_func_done(e => {
        let expectation = { ...expectedResults };
        if (expectation['mode'] != '')
          expectation['mode'] = 'cors';
        fetch('/fetch/metadata/resources/record-header.py?retrieve=true&file=' + key)
          .then(t.step_func(response => response.text()))
          .then(t.step_func_done(text => assert_header_equals(text, expectation, testNamePrefix + ' preload')))
          .catch(t.unreached_func());
      });
      document.head.appendChild(e);
    }, testNamePrefix + ' preload');
  }

  promise_test(t => {
    return new Promise((resolve, reject) => {
      let key = 'style-https-redirect' + nonce;
      let e = document.createElement('link');
      e.rel = 'stylesheet';
      e.href = urlHelperMethod('resources/record-header.py?file=' + key);
      e.onload = e => {
        let expectation = { ...expectedResults };
        if (expectation['mode'] != '')
          expectation['mode'] = 'no-cors';
        if (expectation['dest'] == 'font')
          expectation['dest'] = 'style';
        fetch('/fetch/metadata/resources/record-header.py?retrieve=true&file=' + key)
          .then(response => response.text())
          .then(t.step_func(text => assert_header_equals(text, expectation, testNamePrefix + ' stylesheet')))
          .then(resolve)
          .catch(e => reject(e));
      };
      document.body.appendChild(e);
    });
  }, testNamePrefix + ' stylesheet');

  promise_test(t => {
    return new Promise((resolve, reject) => {
      let key = 'track-https-redirect' + nonce;
      let video = createVideoElement();
      let el = createTrack();
      el.src = urlHelperMethod('resources/record-header.py?file=' + key);
      el.onload = t.step_func(_ => {
        let expectation = { ...expectedResults };
        if (expectation['mode'] != '')
          expectation['mode'] = 'cors';
        if (expectation['dest'] == 'font')
          expectation['dest'] = 'track';
        fetch('/fetch/metadata/resources/record-header.py?retrieve=true&file=' + key)
          .then(response => response.text())
          .then(t.step_func(text => assert_header_equals(text, expectation, testNamePrefix + ' track')))
          .then(resolve);
      });
      video.appendChild(el);
      document.body.appendChild(video);
    });
  }, testNamePrefix + ' track');
}