Domain Reliability: Remove browsing data when requested.

chrome/browser/browsing_data/browsing_data_remover_unittest.cc

 // Copyright (c) 2012 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 "chrome/browser/browsing_data/browsing_data_remover.h"
 
 #include <set>
 #include <vector>
 
 #include "base/bind.h"
@@ skipping 25 matching lines @@
 #if defined(OS_CHROMEOS)
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "chromeos/dbus/fake_dbus_thread_manager.h"
 #include "chromeos/dbus/mock_cryptohome_client.h"
 #endif
 #include "components/autofill/core/browser/autofill_profile.h"
 #include "components/autofill/core/browser/autofill_test_utils.h"
 #include "components/autofill/core/browser/credit_card.h"
 #include "components/autofill/core/browser/personal_data_manager.h"
 #include "components/autofill/core/browser/personal_data_manager_observer.h"
+#include "components/domain_reliability/clear_mode.h"
+#include "components/domain_reliability/monitor.h"
 #include "content/public/browser/cookie_store_factory.h"
 #include "content/public/browser/dom_storage_context.h"
 #include "content/public/browser/local_storage_usage_info.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/storage_partition.h"
 #include "content/public/test/test_browser_thread.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "content/public/test/test_utils.h"
 #include "net/cookies/cookie_store.h"
 #include "net/ssl/server_bound_cert_service.h"
 #include "net/ssl/server_bound_cert_store.h"
 #include "net/ssl/ssl_client_cert_type.h"
 #include "net/url_request/url_request_context.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using content::BrowserThread;
 using content::StoragePartition;
+using domain_reliability::CLEAR_BEACONS;
+using domain_reliability::CLEAR_CONTEXTS;
+using domain_reliability::DomainReliabilityClearMode;
+using domain_reliability::DomainReliabilityMonitor;
 using testing::_;
 using testing::Invoke;
 using testing::WithArgs;
 
 namespace {
 
 const char kTestOrigin1[] = "http://host1:1/";
 const char kTestOrigin2[] = "http://host2:1/";
 const char kTestOrigin3[] = "http://host3:1/";
 const char kTestOriginExt[] = "chrome-extension://abcdefghijklmnopqrstuvwxyz/";
@@ skipping 492 matching lines @@
   // We don't own these pointers.
   TestingProfile* profile_;
   content::DOMStorageContext* dom_storage_context_;
 
   std::vector<content::LocalStorageUsageInfo> infos_;
   base::Closure quit_closure_;
 
   DISALLOW_COPY_AND_ASSIGN(RemoveLocalStorageTester);
 };
 
+class TestingProfileWithDomainReliabilityMonitor : public TestingProfile {
+ public:
+  TestingProfileWithDomainReliabilityMonitor() :
+      TestingProfile(),
+      upload_reporter_string_("test-reporter"),
+      monitor_(GetRequestContext()->GetURLRequestContext(),
+               upload_reporter_string_) {}
+
+  virtual void ClearDomainReliabilityMonitor(
+      DomainReliabilityClearMode mode,
+      const base::Closure& completion) OVERRIDE {
+    monitor_.ClearBrowsingData(mode);
+    completion.Run();
+  }
+
+  DomainReliabilityMonitor* monitor() { return &monitor_; }
+
+ private:
+  std::string upload_reporter_string_;
+  DomainReliabilityMonitor monitor_;
+};
+
 // Test Class ----------------------------------------------------------------
 
 class BrowsingDataRemoverTest : public testing::Test,
                                 public content::NotificationObserver {
  public:
   BrowsingDataRemoverTest()
       : profile_(new TestingProfile()) {
     registrar_.Add(this, chrome::NOTIFICATION_BROWSING_DATA_REMOVED,
                    content::Source<Profile>(profile_.get()));
   }
@@ skipping 85 matching lines @@
 
     // We're not taking ownership of the details object, but storing a copy of
     // it locally.
     called_with_details_.reset(new BrowsingDataRemover::NotificationDetails(
         *content::Details<BrowsingDataRemover::NotificationDetails>(
             details).ptr()));
 
     registrar_.RemoveAll();
   }
 
+  DomainReliabilityMonitor *UseProfileWithDomainReliabilityMonitor() {
+    TestingProfileWithDomainReliabilityMonitor* new_profile =
+        new TestingProfileWithDomainReliabilityMonitor();
+    DomainReliabilityMonitor* monitor = new_profile->monitor();
+    profile_.reset(new_profile);
+    return monitor;
+  }
+
  protected:
   scoped_ptr<BrowsingDataRemover::NotificationDetails> called_with_details_;
 
  private:
   content::NotificationRegistrar registrar_;
 
   content::TestBrowserThreadBundle thread_bundle_;
   scoped_ptr<TestingProfile> profile_;
 
   StoragePartitionRemovalData storage_partition_removal_data_;
@@ skipping 919 matching lines @@
   EXPECT_CALL(*cryptohome_client, TpmAttestationDeleteKeys(_, _, _, _))
       .WillOnce(WithArgs<3>(Invoke(FakeDBusCall)));
 
   BlockUntilBrowsingDataRemoved(
       BrowsingDataRemover::EVERYTHING,
       BrowsingDataRemover::REMOVE_CONTENT_LICENSES, false);
 
   chromeos::DBusThreadManager::Shutdown();
 }
 #endif
+
+TEST_F(BrowsingDataRemoverTest, DomainReliability_Null) {
+  DomainReliabilityMonitor* monitor = UseProfileWithDomainReliabilityMonitor();
+  EXPECT_FALSE(monitor->was_cleared_for_testing());
+}
+
+TEST_F(BrowsingDataRemoverTest, DomainReliability_Beacons) {
+  DomainReliabilityMonitor* monitor = UseProfileWithDomainReliabilityMonitor();
+  BlockUntilBrowsingDataRemoved(
+      BrowsingDataRemover::EVERYTHING,
+      BrowsingDataRemover::REMOVE_HISTORY, false);
+  EXPECT_TRUE(monitor->was_cleared_for_testing());
+  EXPECT_EQ(CLEAR_BEACONS, monitor->cleared_mode_for_testing());
+}
+
+TEST_F(BrowsingDataRemoverTest, DomainReliability_Beacons_ProtectedOrigins) {
+  DomainReliabilityMonitor* monitor = UseProfileWithDomainReliabilityMonitor();
+  BlockUntilBrowsingDataRemoved(
+      BrowsingDataRemover::EVERYTHING,
+      BrowsingDataRemover::REMOVE_HISTORY, true);
+  EXPECT_TRUE(monitor->was_cleared_for_testing());
+  EXPECT_EQ(CLEAR_BEACONS, monitor->cleared_mode_for_testing());
+}
+
+TEST_F(BrowsingDataRemoverTest, DomainReliability_Contexts) {
+  DomainReliabilityMonitor* monitor = UseProfileWithDomainReliabilityMonitor();
+  BlockUntilBrowsingDataRemoved(
+      BrowsingDataRemover::EVERYTHING,
+      BrowsingDataRemover::REMOVE_COOKIES, false);
+  EXPECT_TRUE(monitor->was_cleared_for_testing());
+  EXPECT_EQ(CLEAR_CONTEXTS, monitor->cleared_mode_for_testing());
+}
+
+TEST_F(BrowsingDataRemoverTest, DomainReliability_Contexts_ProtectedOrigins) {
+  DomainReliabilityMonitor* monitor = UseProfileWithDomainReliabilityMonitor();
+  BlockUntilBrowsingDataRemoved(
+      BrowsingDataRemover::EVERYTHING,
+      BrowsingDataRemover::REMOVE_COOKIES, true);
+  EXPECT_TRUE(monitor->was_cleared_for_testing());
+  EXPECT_EQ(CLEAR_CONTEXTS, monitor->cleared_mode_for_testing());
+}