1 files changed, 118 insertions, 0 deletions
diff --git a/services/sync/tests/unit/test_service_verifyLogin.js b/services/sync/tests/unit/test_service_verifyLogin.js
new file mode 100644
index 0000000000..b99b5c692c
--- /dev/null
+++ b/services/sync/tests/unit/test_service_verifyLogin.js
@@ -0,0 +1,118 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+const { Service } = ChromeUtils.importESModule(
+  "resource://services-sync/service.sys.mjs"
+);
+
+function login_handling(handler) {
+  return function (request, response) {
+    if (has_hawk_header(request)) {
+      handler(request, response);
+    } else {
+      let body = "Unauthorized";
+      response.setStatusLine(request.httpVersion, 401, "Unauthorized");
+      response.bodyOutputStream.write(body, body.length);
+    }
+  };
+}
+
+function service_unavailable(request, response) {
+  let body = "Service Unavailable";
+  response.setStatusLine(request.httpVersion, 503, "Service Unavailable");
+  response.setHeader("Retry-After", "42");
+  response.bodyOutputStream.write(body, body.length);
+}
+
+function run_test() {
+  Log.repository.rootLogger.addAppender(new Log.DumpAppender());
+
+  run_next_test();
+}
+
+add_task(async function test_verifyLogin() {
+  // This test expects a clean slate -- no saved passphrase.
+  Services.logins.removeAllUserFacingLogins();
+  let johnHelper = track_collections_helper();
+  let johnU = johnHelper.with_updated_collection;
+
+  do_test_pending();
+
+  let server = httpd_setup({
+    "/1.1/johndoe/info/collections": login_handling(johnHelper.handler),
+    "/1.1/janedoe/info/collections": service_unavailable,
+
+    "/1.1/johndoe/storage/crypto/keys": johnU(
+      "crypto",
+      new ServerWBO("keys").handler()
+    ),
+    "/1.1/johndoe/storage/meta/global": johnU(
+      "meta",
+      new ServerWBO("global").handler()
+    ),
+  });
+
+  try {
+    _("Force the initial state.");
+    Service.status.service = STATUS_OK;
+    Assert.equal(Service.status.service, STATUS_OK);
+
+    _("Credentials won't check out because we're not configured yet.");
+    Service.status.resetSync();
+    Assert.equal(false, await Service.verifyLogin());
+    Assert.equal(Service.status.service, CLIENT_NOT_CONFIGURED);
+    Assert.equal(Service.status.login, LOGIN_FAILED_NO_USERNAME);
+
+    _("Success if syncBundleKey is set.");
+    Service.status.resetSync();
+    await configureIdentity({ username: "johndoe" }, server);
+    Assert.ok(await Service.verifyLogin());
+    Assert.equal(Service.status.service, STATUS_OK);
+    Assert.equal(Service.status.login, LOGIN_SUCCEEDED);
+
+    _(
+      "If verifyLogin() encounters a server error, it flips on the backoff flag and notifies observers on a 503 with Retry-After."
+    );
+    Service.status.resetSync();
+    await configureIdentity({ username: "janedoe" }, server);
+    Service._updateCachedURLs();
+    Assert.ok(!Service.status.enforceBackoff);
+    let backoffInterval;
+    Svc.Obs.add(
+      "weave:service:backoff:interval",
+      function observe(subject, data) {
+        Svc.Obs.remove("weave:service:backoff:interval", observe);
+        backoffInterval = subject;
+      }
+    );
+    Assert.equal(false, await Service.verifyLogin());
+    Assert.ok(Service.status.enforceBackoff);
+    Assert.equal(backoffInterval, 42);
+    Assert.equal(Service.status.service, LOGIN_FAILED);
+    Assert.equal(Service.status.login, SERVER_MAINTENANCE);
+
+    _(
+      "Ensure a network error when finding the cluster sets the right Status bits."
+    );
+    Service.status.resetSync();
+    Service.clusterURL = "";
+    Service.identity._findCluster = () => "http://localhost:12345/";
+    Assert.equal(false, await Service.verifyLogin());
+    Assert.equal(Service.status.service, LOGIN_FAILED);
+    Assert.equal(Service.status.login, LOGIN_FAILED_NETWORK_ERROR);
+
+    _(
+      "Ensure a network error when getting the collection info sets the right Status bits."
+    );
+    Service.status.resetSync();
+    Service.clusterURL = "http://localhost:12345/";
+    Assert.equal(false, await Service.verifyLogin());
+    Assert.equal(Service.status.service, LOGIN_FAILED);
+    Assert.equal(Service.status.login, LOGIN_FAILED_NETWORK_ERROR);
+  } finally {
+    for (const pref of Svc.PrefBranch.getChildList("")) {
+      Svc.PrefBranch.clearUserPref(pref);
+    }
+    server.stop(do_test_finished);
+  }
+});