/* Any copyright is dedicated to the Public Domain.
   http://creativecommons.org/publicdomain/zero/1.0/ */

"use strict";

const { HttpServer } = ChromeUtils.import("resource://testing-common/httpd.js");

var httpServer = null;

XPCOMUtils.defineLazyGetter(this, "serverPort", function () {
  return httpServer.identity.primaryPort;
});

var handlerDone;
var handlerPromise = new Promise(r => (handlerDone = after(2, r)));

function resubscribeHandler(metadata, response) {
  ok(true, "Ask for new subscription");
  handlerDone();
  response.setHeader(
    "Location",
    "http://localhost:" + serverPort + "/newSubscription"
  );
  response.setHeader(
    "Link",
    '</newPushEndpoint>; rel="urn:ietf:params:push", ' +
      '</newReceiptPushEndpoint>; rel="urn:ietf:params:push:receipt"'
  );
  response.setStatusLine(metadata.httpVersion, 201, "OK");
}

function listenSuccessHandler(metadata, response) {
  Assert.ok(true, "New listener point");
  httpServer.stop(handlerDone);
  response.setStatusLine(metadata.httpVersion, 204, "Try again");
}

httpServer = new HttpServer();
httpServer.registerPathHandler("/subscribe", resubscribeHandler);
httpServer.registerPathHandler("/newSubscription", listenSuccessHandler);
httpServer.start(-1);

function run_test() {
  do_get_profile();
  setPrefs({
    "testing.allowInsecureServerURL": true,
    "http2.retryInterval": 1000,
    "http2.maxRetries": 2,
  });

  run_next_test();
}

add_task(async function test1() {
  let db = PushServiceHttp2.newPushDB();
  registerCleanupFunction(() => {
    return db.drop().then(_ => db.close());
  });

  var serverURL = "http://localhost:" + httpServer.identity.primaryPort;

  let records = [
    {
      subscriptionUri: "http://localhost/subscriptionNotExist",
      pushEndpoint: serverURL + "/pushEndpoint",
      pushReceiptEndpoint: serverURL + "/pushReceiptEndpoint",
      scope: "https://example.com/page",
      p256dhPublicKey:
        "BPCd4gNQkjwRah61LpdALdzZKLLnU5UAwDztQ5_h0QsT26jk0IFbqcK6-JxhHAm-rsHEwy0CyVJjtnfOcqc1tgA",
      p256dhPrivateKey: {
        crv: "P-256",
        d: "1jUPhzVsRkzV0vIzwL4ZEsOlKdNOWm7TmaTfzitJkgM",
        ext: true,
        key_ops: ["deriveBits"],
        kty: "EC",
        x: "8J3iA1CSPBFqHrUul0At3NkosudTlQDAPO1Dn-HRCxM",
        y: "26jk0IFbqcK6-JxhHAm-rsHEwy0CyVJjtnfOcqc1tgA",
      },
      originAttributes: "",
      quota: Infinity,
    },
  ];

  for (let record of records) {
    await db.put(record);
  }

  PushService.init({
    serverURI: serverURL + "/subscribe",
    db,
  });

  await handlerPromise;

  let record = await db.getByIdentifiers({
    scope: "https://example.com/page",
    originAttributes: "",
  });
  equal(
    record.keyID,
    serverURL + "/newSubscription",
    "Should update subscription URL"
  );
  equal(
    record.pushEndpoint,
    serverURL + "/newPushEndpoint",
    "Should update push endpoint"
  );
  equal(
    record.pushReceiptEndpoint,
    serverURL + "/newReceiptPushEndpoint",
    "Should update push receipt endpoint"
  );
});
