[Prototype] Delete Google service worker caches on Android signout

chrome/browser/android/signin/signin_manager_android_unittest.cc

+// Copyright 2017 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "signin_manager_android.h"
+
+#include <memory>
+#include <set>
+
+#include "base/memory/ref_counted.h"
+#include "base/run_loop.h"
+#include "chrome/browser/browsing_data/browsing_data_service_worker_helper.h"
+#include "chrome/test/base/testing_browser_process.h"
+#include "chrome/test/base/testing_profile_manager.h"
+#include "content/public/browser/browser_context.h"
+#include "content/public/browser/browsing_data_remover.h"
+#include "content/public/browser/storage_partition.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+class SigninManagerAndroidTest : public ::testing::Test {
+ public:
+  SigninManagerAndroidTest()
+      : profile_manager_(TestingBrowserProcess::GetGlobal()) {}
+  ~SigninManagerAndroidTest() override {};
+
+  void SetUp() override {
+    ASSERT_TRUE(profile_manager_.SetUp());
+    profile_ = profile_manager_.CreateTestingProfile("Testing Profile");
+  }
+
+  TestingProfile* profile() { return profile_; }
+
+ private:
+  content::TestBrowserThreadBundle thread_bundle_;
+  TestingProfileManager profile_manager_;
+  TestingProfile* profile_;  // Owned by |profile_manager_|.
+
+  DISALLOW_COPY_AND_ASSIGN(SigninManagerAndroidTest);
+};
+
+TEST_F(SigninManagerAndroidTest, DeleteGoogleServiceWorkers) {
+  struct TestCase {
+    std::string worker_url;
+    bool should_be_deleted;
+  } kTestCases[] = {
+      // A Google domain.
+      {"https://google.com/foo/bar", true},
+
+      // A Google domain with long TLD.
+      {"https://plus.google.co.uk/?query_params", true},
+
+      // Youtube.
+      {"https://youtube.com", false},
+
+      // A random domain.
+      {"https://a.b.c.example.com", false},
+
+      // Another Google domain.
+      {"https://www.google.de/worker.html", true},
+
+      // Ports don't matter, only TLDs.
+      {"https://google.com:8444/worker.html", true},
+  };
+
+  // Add service workers.
+  scoped_refptr<CannedBrowsingDataServiceWorkerHelper> helper(
+      new CannedBrowsingDataServiceWorkerHelper(
+          content::BrowserContext::GetDefaultStoragePartition(profile())
+              ->GetServiceWorkerContext()));
+
+  for (const TestCase& test_case : kTestCases) {
+    std::vector<GURL> scopes = {GURL("/")};  // the entire origin
+    helper->AddServiceWorker(GURL(test_case.worker_url), scopes);
+  }
+
+  ASSERT_EQ(arraysize(kTestCases), helper->GetServiceWorkerCount());
+
+  // Delete service workers and wait for completion.
+  base::RunLoop run_loop;
+  SigninManagerAndroid::WipeData(profile(), false /* only service workers */,
+                                 run_loop.QuitClosure());
+  run_loop.Run();
+
+  // Test whether the correct service workers were deleted.
+  std::set<std::string> remaining_workers;
+  for (const auto& info : helper->GetServiceWorkerUsageInfo())
+    remaining_workers.insert(info.origin.spec());
+
+  for (const TestCase& test_case : kTestCases) {
+    EXPECT_EQ(test_case.should_be_deleted,
+              base::ContainsKey(remaining_workers, test_case.worker_url))
+        << test_case.worker_url << " should "
+        << (test_case.should_be_deleted ? "" : "NOT ")
+        << "be deleted, but it was"
+        << (test_case.should_be_deleted ? "NOT" : "") << ".";
+  }
+}