// META: script=helpers.js
// META: script=/cookies/resources/cookie-helper.sub.js
// META: script=/resources/testdriver.js
// META: script=/resources/testdriver-vendor.js

'use strict';

const altRoot = "https://{{hosts[alt][]}}:{{ports[https][0]}}";
const altRootWss = "wss://{{hosts[alt][]}}:{{ports[wss][0]}}";

const responderPath = "/storage-access-api/resources/script-with-cookie-header.py?script=embedded_responder.js";
const altRootResponder = `${altRoot}${responderPath}`;

async function SetUpResponderFrame(t, url) {
  const frame = await CreateFrame(url);

  await SetPermissionInFrame(frame, [{ name: 'storage-access' }, 'granted']);
  t.add_cleanup(async () => {
    await test_driver.delete_all_cookies();
    await SetPermissionInFrame(frame, [{ name: 'storage-access' }, 'prompt']);
    await MaybeSetStorageAccess("*", "*", "allowed");
  });

  if (await FrameHasStorageAccess(frame)) {
    // Nothing to test here, since cookie access is not blocked.
    // See https://github.com/privacycg/storage-access/issues/162.
    return null;
  }

  return frame;
}

promise_test(async (t) => {
  await MaybeSetStorageAccess("*", "*", "blocked");
  await SetFirstPartyCookie(altRoot);

  const frame = await SetUpResponderFrame(t, altRootResponder);
  if (!frame) {
    return;
  }

  assert_true(await RequestStorageAccessInFrame(frame), "requestStorageAccess resolves without requiring a gesture.");
  assert_true(await FrameHasStorageAccess(frame), "frame has storage access after request.");
  assert_true(await HasUnpartitionedCookie(frame), "frame has access to cookies after request.");

  assert_true(cookieStringHasCookie("cookie", "unpartitioned",
              await ReadCookiesFromWebSocketConnection(frame, altRootWss)),
          "WebSocket handshake should include unpartitioned cookie");
}, "WebSocket inherits storage access");

promise_test(async (t) => {

  await MaybeSetStorageAccess("*", "*", "blocked");
  await SetFirstPartyCookie(altRoot);
  const frame = await SetUpResponderFrame(t, altRootResponder);
  if (!frame) {
    return;
  }

  assert_false(cookieStringHasCookie("cookie", "unpartitioned",
              await ReadCookiesFromWebSocketConnection(frame, altRootWss)),
          "request should not contain cookies");
}, "WebSocket omits unpartitioned cookies without storage access");