1 files changed, 160 insertions, 0 deletions
diff --git a/dom/quota/test/gtest/QuotaManagerDependencyFixture.cpp b/dom/quota/test/gtest/QuotaManagerDependencyFixture.cpp
new file mode 100644
index 0000000000..2e52b4b69a
--- /dev/null
+++ b/dom/quota/test/gtest/QuotaManagerDependencyFixture.cpp
@@ -0,0 +1,160 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* 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 "QuotaManagerDependencyFixture.h"
+
+#include "mozIStorageService.h"
+#include "mozStorageCID.h"
+#include "mozilla/dom/ScriptSettings.h"
+#include "mozilla/dom/quota/QuotaManagerService.h"
+#include "mozilla/gtest/MozAssertions.h"
+#include "nsIPrefBranch.h"
+#include "nsIPrefService.h"
+#include "nsIQuotaCallbacks.h"
+#include "nsIQuotaRequests.h"
+#include "nsIVariant.h"
+#include "nsScriptSecurityManager.h"
+
+namespace mozilla::dom::quota::test {
+
+namespace {
+
+class RequestResolver final : public nsIQuotaCallback {
+ public:
+  RequestResolver() : mDone(false) {}
+
+  bool Done() const { return mDone; }
+
+  NS_DECL_ISUPPORTS
+
+  NS_IMETHOD OnComplete(nsIQuotaRequest* aRequest) override {
+    mDone = true;
+
+    return NS_OK;
+  }
+
+ private:
+  ~RequestResolver() = default;
+
+  bool mDone;
+};
+
+}  // namespace
+
+NS_IMPL_ISUPPORTS(RequestResolver, nsIQuotaCallback)
+
+// static
+void QuotaManagerDependencyFixture::InitializeFixture() {
+  // Some QuotaManagerService methods fail if the testing pref is not set.
+  nsCOMPtr<nsIPrefBranch> prefs = do_GetService(NS_PREFSERVICE_CONTRACTID);
+
+  prefs->SetBoolPref("dom.quotaManager.testing", true);
+
+  // The first initialization of storage service must be done on the main
+  // thread.
+  nsCOMPtr<mozIStorageService> storageService =
+      do_GetService(MOZ_STORAGE_SERVICE_CONTRACTID);
+  ASSERT_TRUE(storageService);
+
+  nsIObserver* observer = QuotaManager::GetObserver();
+  ASSERT_TRUE(observer);
+
+  nsresult rv = observer->Observe(nullptr, "profile-do-change", nullptr);
+  ASSERT_NS_SUCCEEDED(rv);
+
+  ASSERT_NO_FATAL_FAILURE(StorageInitialized());
+
+  QuotaManager* quotaManager = QuotaManager::Get();
+  ASSERT_TRUE(quotaManager);
+
+  ASSERT_TRUE(quotaManager->OwningThread());
+
+  sBackgroundTarget = quotaManager->OwningThread();
+}
+
+// static
+void QuotaManagerDependencyFixture::ShutdownFixture() {
+  nsCOMPtr<nsIPrefBranch> prefs = do_GetService(NS_PREFSERVICE_CONTRACTID);
+
+  prefs->SetBoolPref("dom.quotaManager.testing", false);
+
+  nsIObserver* observer = QuotaManager::GetObserver();
+  ASSERT_TRUE(observer);
+
+  nsresult rv = observer->Observe(nullptr, "profile-before-change-qm", nullptr);
+  ASSERT_NS_SUCCEEDED(rv);
+
+  PerformOnBackgroundThread([]() { QuotaManager::Reset(); });
+
+  sBackgroundTarget = nullptr;
+}
+
+// static
+void QuotaManagerDependencyFixture::StorageInitialized(bool* aResult) {
+  AutoJSAPI jsapi;
+
+  bool ok = jsapi.Init(xpc::PrivilegedJunkScope());
+  ASSERT_TRUE(ok);
+
+  nsCOMPtr<nsIQuotaManagerService> qms = QuotaManagerService::GetOrCreate();
+  ASSERT_TRUE(qms);
+
+  nsCOMPtr<nsIQuotaRequest> request;
+  nsresult rv = qms->StorageInitialized(getter_AddRefs(request));
+  ASSERT_NS_SUCCEEDED(rv);
+
+  RefPtr<RequestResolver> resolver = new RequestResolver();
+
+  rv = request->SetCallback(resolver);
+  ASSERT_NS_SUCCEEDED(rv);
+
+  SpinEventLoopUntil("Promise is fulfilled"_ns,
+                     [&resolver]() { return resolver->Done(); });
+
+  if (aResult) {
+    nsCOMPtr<nsIVariant> result;
+    rv = request->GetResult(getter_AddRefs(result));
+    ASSERT_NS_SUCCEEDED(rv);
+
+    rv = result->GetAsBool(aResult);
+    ASSERT_NS_SUCCEEDED(rv);
+  }
+}
+
+// static
+void QuotaManagerDependencyFixture::ClearStoragesForOrigin(
+    const OriginMetadata& aOriginMetadata) {
+  nsCOMPtr<nsIQuotaManagerService> qms = QuotaManagerService::GetOrCreate();
+  ASSERT_TRUE(qms);
+
+  nsCOMPtr<nsIScriptSecurityManager> ssm =
+      nsScriptSecurityManager::GetScriptSecurityManager();
+  ASSERT_TRUE(ssm);
+
+  nsCOMPtr<nsIPrincipal> principal;
+  nsresult rv = ssm->CreateContentPrincipalFromOrigin(
+      aOriginMetadata.mOrigin, getter_AddRefs(principal));
+  ASSERT_NS_SUCCEEDED(rv);
+
+  nsCOMPtr<nsIQuotaRequest> request;
+  rv = qms->ClearStoragesForPrincipal(principal, VoidCString(), VoidString(),
+                                      /* aClearAll */ false,
+                                      getter_AddRefs(request));
+  ASSERT_NS_SUCCEEDED(rv);
+
+  RefPtr<RequestResolver> resolver = new RequestResolver();
+  ASSERT_TRUE(resolver);
+
+  rv = request->SetCallback(resolver);
+  ASSERT_NS_SUCCEEDED(rv);
+
+  SpinEventLoopUntil("Promise is fulfilled"_ns,
+                     [&resolver]() { return resolver->Done(); });
+}
+
+nsCOMPtr<nsISerialEventTarget> QuotaManagerDependencyFixture::sBackgroundTarget;
+
+}  // namespace mozilla::dom::quota::test