TestCookieStoreSubscriptionService.cpp

Enable keyboard shortcuts

/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "gtest/gtest.h"

#include "mozilla/dom/CookieStoreSubscriptionService.h"

#include "mozilla/dom/PCookieStore.h"

#include "mozilla/dom/ServiceWorkerRegistrarTypes.h"

#include "mozilla/ipc/PBackgroundSharedTypes.h"

using namespace mozilla;

using namespace mozilla::dom;

namespace {

ServiceWorkerRegistrationData Registration(const nsACString& aScope) {

  const nsCString scope(aScope);

  ServiceWorkerRegistrationData data;

  data.scope() = scope;

  data.principal() = mozilla::ipc::ContentPrincipalInfo(

      OriginAttributes(), scope, scope, Nothing(), scope);

  return data;

// The service is a singleton and keeps the loaded registrations forever, so

// every test has to use its own scope to stay independent from the others.

nsTArray<CookieSubscription> LoadAndGet(const nsACString& aScope,

                                        const nsACString& aValue) {

  ServiceWorkerRegistrationData registration = Registration(aScope);

  CookieStoreSubscriptionService* service =

      CookieStoreSubscriptionService::Instance();

  service->Load(registration, aValue);

  nsTArray<CookieSubscription> subscriptions;

  service->GetSubscriptions(registration.principal(), aScope, subscriptions);

  return subscriptions;

}  // namespace

TEST(CookieStoreSubscriptionService, RestoreNameAndURL)

  nsTArray<CookieSubscription> subscriptions = LoadAndGet(

      "https://example.com/named/"_ns,

      R"([{"name":"cookieName","url":"https://example.com/named/a"}])"_ns);

  ASSERT_EQ(1u, subscriptions.Length());

  ASSERT_TRUE(subscriptions[0].name().isSome());

  EXPECT_TRUE(subscriptions[0].name()->EqualsLiteral("cookieName"));

  EXPECT_TRUE(

      subscriptions[0].url().EqualsLiteral("https://example.com/named/a"));

TEST(CookieStoreSubscriptionService, RestoreURLOnly)

  nsTArray<CookieSubscription> subscriptions =

      LoadAndGet("https://example.com/nameless/"_ns,

                 R"([{"url":"https://example.com/nameless/a"}])"_ns);

  ASSERT_EQ(1u, subscriptions.Length());

  EXPECT_TRUE(subscriptions[0].name().isNothing());

  EXPECT_TRUE(

      subscriptions[0].url().EqualsLiteral("https://example.com/nameless/a"));

TEST(CookieStoreSubscriptionService, RestoreMultiple)

  nsTArray<CookieSubscription> subscriptions =

      LoadAndGet("https://example.com/multiple/"_ns,

                 R"([{"name":"a","url":"https://example.com/multiple/a"},)"

                 R"({"name":"b","url":"https://example.com/multiple/b"}])"_ns);

  ASSERT_EQ(2u, subscriptions.Length());

  EXPECT_TRUE(subscriptions[0].name()->EqualsLiteral("a"));

  EXPECT_TRUE(subscriptions[1].name()->EqualsLiteral("b"));

TEST(CookieStoreSubscriptionService, MalformedDocument)

  EXPECT_TRUE(

      LoadAndGet("https://example.com/broken/"_ns, "[{\"url\":"_ns).IsEmpty());

  EXPECT_TRUE(LoadAndGet("https://example.com/empty/"_ns, ""_ns).IsEmpty());

  EXPECT_TRUE(LoadAndGet("https://example.com/object/"_ns,

                         R"({"url":"https://example.com/object/a"})"_ns)

                  .IsEmpty());

  EXPECT_TRUE(

      LoadAndGet("https://example.com/emptyarray/"_ns, "[]"_ns).IsEmpty());

TEST(CookieStoreSubscriptionService, InvalidEntriesAreSkipped)

  nsTArray<CookieSubscription> subscriptions =

      LoadAndGet("https://example.com/invalid/"_ns,

                 R"([42,)"

                 R"({},)"

                 R"({"url":42},)"

                 R"({"name":42,"url":"https://example.com/invalid/a"},)"

                 R"({"url":"https://example.com/invalid/b"}])"_ns);

  ASSERT_EQ(1u, subscriptions.Length());

  EXPECT_TRUE(

      subscriptions[0].url().EqualsLiteral("https://example.com/invalid/b"));