Integrate SiteEngagementEvictionPolicy with QuotaManager.

chrome/browser/engagement/site_engagement_eviction_policy_unittest.cc

 // Copyright 2015 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 "base/files/scoped_temp_dir.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/memory/weak_ptr.h"
+#include "base/run_loop.h"
+#include "base/thread_task_runner_handle.h"
 #include "chrome/browser/engagement/site_engagement_eviction_policy.h"
 #include "chrome/browser/engagement/site_engagement_service.h"
 #include "content/public/test/mock_special_storage_policy.h"
+#include "content/public/test/mock_storage_client.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "storage/browser/quota/quota_manager.h"
+#include "storage/browser/quota/quota_manager_proxy.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
+const char* kFooOrigin = "http://foo.com/";
+
 const int64 kGlobalQuota = 25 * 1024;
 
-}  // namespace
+// Returns a deterministic value for the amount of available disk space.
+int64 GetAvailableDiskSpaceForTest(const base::FilePath&) {
+  return kGlobalQuota;
+}
 
 class TestSiteEngagementScoreProvider : public SiteEngagementScoreProvider {
  public:
   TestSiteEngagementScoreProvider() {}
 
   virtual ~TestSiteEngagementScoreProvider() {}
 
   double GetScore(const GURL& url) override {
     return engagement_score_map_[url];
   }
 
   double GetTotalEngagementPoints() override {
     double total = 0;
     for (const auto& site : engagement_score_map_)
       total += site.second;
     return total;
   }
 
   void SetScore(const GURL& origin, double score) {
     engagement_score_map_[origin] = score;
   }
 
  private:
   std::map<GURL, double> engagement_score_map_;
 
   DISALLOW_COPY_AND_ASSIGN(TestSiteEngagementScoreProvider);
 };
 
+SiteEngagementScoreProvider* GetTestSiteEngagementScoreProvider(
+    content::BrowserContext*) {
+  CR_DEFINE_STATIC_LOCAL(TestSiteEngagementScoreProvider,
+                         kGlobalTestScoreProvider, ());
+  return &kGlobalTestScoreProvider;
+}
+
+}  // namespace
+
+class SiteEngagementEvictionPolicyWithQuotaManagerTest : public testing::Test {
+ public:
+  SiteEngagementEvictionPolicyWithQuotaManagerTest() : weak_factory_(this) {}
+
+  void SetUp() override {
+    ASSERT_TRUE(data_dir_.CreateUniqueTempDir());
+    mock_special_storage_policy_ = new content::MockSpecialStoragePolicy;
+
+    quota_manager_ = new storage::QuotaManager(
+        false, data_dir_.path(), base::ThreadTaskRunnerHandle::Get().get(),
+        base::ThreadTaskRunnerHandle::Get().get(),
+        mock_special_storage_policy_.get());
+
+    // Populate the QuotaManager with some data.
+    static const content::MockOriginData kData[] = {
+        {kFooOrigin, storage::kStorageTypeTemporary, 1},
+    };
+    quota_manager_->proxy()->RegisterClient(new content::MockStorageClient(
+        quota_manager_->proxy(), kData, storage::QuotaClient::kFileSystem,
+        arraysize(kData)));
+
+    // Set the eviction policy.
+    quota_manager_->SetQuotaEvictionPolicy(
+        storage::kStorageTypeTemporary,
+        make_scoped_ptr(new SiteEngagementEvictionPolicy(nullptr)));
+
+    // Don't (automatically) start the eviction for testing.
+    quota_manager_->eviction_disabled_ = true;
+    // Don't query the hard disk for remaining capacity.
+    quota_manager_->get_disk_space_fn_ = &GetAvailableDiskSpaceForTest;
+
+    score_provider_callback_ = base::Bind(&GetTestSiteEngagementScoreProvider);
+
+    // Use a TestSiteEngagementScoreProvider when retrieving the engagement
+    // scores.
+    SiteEngagementEvictionPolicy::
+        SetSiteEngagementScoreProviderCallbackForTests(
+            &score_provider_callback_);
+  }
+
+  void TearDown() override {
+    SiteEngagementEvictionPolicy::
+        SetSiteEngagementScoreProviderCallbackForTests(nullptr);
+    // Make sure the quota manager cleans up correctly.
+    quota_manager_ = nullptr;
+    base::RunLoop().RunUntilIdle();
+  }
+
+ protected:
+  void GetEvictionOrigin() {
+    quota_manager_->GetEvictionOrigin(
+        storage::kStorageTypeTemporary,
+        base::Bind(&SiteEngagementEvictionPolicyWithQuotaManagerTest::
+                       OnDidGetEvictionOrigin,
+                   weak_factory_.GetWeakPtr()));
+  }
+
+  void OnDidGetEvictionOrigin(const GURL& origin) { eviction_origin_ = origin; }
+
+  const GURL& eviction_origin() { return eviction_origin_; }
+
+ private:
+  content::TestBrowserThreadBundle thread_bundle_;
+  base::ScopedTempDir data_dir_;
+
+  scoped_refptr<storage::QuotaManager> quota_manager_;
+  scoped_refptr<content::MockSpecialStoragePolicy> mock_special_storage_policy_;
+  base::Callback<SiteEngagementScoreProvider*(content::BrowserContext*)>
+      score_provider_callback_;
+
+  GURL eviction_origin_;
+
+  base::WeakPtrFactory<SiteEngagementEvictionPolicyWithQuotaManagerTest>
+      weak_factory_;
+
+  DISALLOW_COPY_AND_ASSIGN(SiteEngagementEvictionPolicyWithQuotaManagerTest);
+};
+
+// Ensure the SiteEngagementEvictionPolicy completes successfully
+// when used as a QuotaManager's eviction policy.
+TEST_F(SiteEngagementEvictionPolicyWithQuotaManagerTest, GetEvictionOrigin) {
+  GetEvictionOrigin();
+  base::RunLoop().RunUntilIdle();
+
+  EXPECT_EQ(GURL(kFooOrigin), eviction_origin());
+}
+
 class SiteEngagementEvictionPolicyTest : public testing::Test {
  public:
   SiteEngagementEvictionPolicyTest()
       : score_provider_(new TestSiteEngagementScoreProvider()),
         storage_policy_(new content::MockSpecialStoragePolicy()) {}
 
   ~SiteEngagementEvictionPolicyTest() override {}
 
   GURL CalculateEvictionOrigin(const std::map<GURL, int64>& usage) {
-    return SiteEngagementEvictionPolicy::CalculateEvictionOrigin(
+    return SiteEngagementEvictionPolicy::CalculateEvictionOriginForTests(
         storage_policy_, score_provider_.get(), usage, kGlobalQuota);
   }
 
   TestSiteEngagementScoreProvider* score_provider() {
     return score_provider_.get();
   }
 
   content::MockSpecialStoragePolicy* storage_policy() {
     return storage_policy_.get();
   }
@@ skipping 58 matching lines @@
   EXPECT_EQ(url2, CalculateEvictionOrigin(usage));
 
   // Durable storage doesn't get evicted.
   storage_policy()->AddDurable(url2);
   EXPECT_EQ(url1, CalculateEvictionOrigin(usage));
 
   // Unlimited storage doesn't get evicted.
   storage_policy()->AddUnlimited(url1);
   EXPECT_EQ(GURL(), CalculateEvictionOrigin(usage));
 }