var RESOURCES_DIR = "/beacon/resources/";

var referrerOrigin = self.location.origin + '/';
var referrerUrl = self.location.href;

function testReferrerHeader(testBase, expectedReferrer, mayBeBlockedAsMixedContent = false) {
  var id = self.token();
  var testUrl = testBase + "inspect-header.py?header=referer&cmd=put&id=" + id;

  promise_test(function(test) {
    const sentBeacon = navigator.sendBeacon(testUrl);
    if (mayBeBlockedAsMixedContent && !sentBeacon)
      return Promise.resolve();
    assert_true(sentBeacon, "SendBeacon Succeeded");
    return pollResult(expectedReferrer, id) .then(result => {
      assert_equals(result, expectedReferrer, "Correct referrer header result");
    });
  }, "Test referer header " + testBase);
}

function testOriginHeader(testBase, expectedOrigin, addBody) {
  var id = self.token();
  var testUrl = testBase + "inspect-header.py?header=origin&cmd=put&id=" + id;

  promise_test(function(test) {
    const sentBeacon = navigator.sendBeacon(testUrl, addBody ? "data" : undefined);
    assert_true(sentBeacon, "SendBeacon Succeeded");
    return pollResult(expectedOrigin, id, "origin") .then(result => {
      assert_equals(result, expectedOrigin, "Correct origin header result");
    });
  }, "Test origin header " + testBase + (addBody ? " - with body" : " - without body"));
}

// SendBeacon is an asynchronous and non-blocking request to a web server.
// We may have to create a poll loop to get result from server
function pollResult(expectedReferrer, id, headerName) {
  if (!headerName)
    headerName = "referer";
  var checkUrl = RESOURCES_DIR + "inspect-header.py?header=" + headerName + "&cmd=get&id=" + id;

  return new Promise(resolve => {
    function checkResult() {
      fetch(checkUrl).then(
        function(response) {
          assert_equals(response.status, 200, "Inspect header response's status is 200");
          let result = response.headers.get("x-request-" + headerName);

          if (result != undefined) {
            resolve(result);
          } else {
            step_timeout(checkResult.bind(this), 100);
          }
        });
    }

    checkResult();
  });

}