Integrate SiteEngagementEvictionPolicy with QuotaManager.

chrome/browser/engagement/site_engagement_eviction_policy.h

 // 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.
 
 #ifndef CHROME_BROWSER_ENGAGEMENT_SITE_ENGAGEMENT_EVICTION_POLICY_H_
 #define CHROME_BROWSER_ENGAGEMENT_SITE_ENGAGEMENT_EVICTION_POLICY_H_
 
 #include <map>
 
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
+#include "base/memory/weak_ptr.h"
 #include "storage/browser/quota/quota_manager.h"
 #include "url/gurl.h"
 
 namespace content {
 class BrowserContext;
 }
 
 class SiteEngagementScoreProvider;
 
 class SiteEngagementEvictionPolicy : public storage::QuotaEvictionPolicy {
  public:
-  explicit SiteEngagementEvictionPolicy(
-      content::BrowserContext* browser_context);
-  virtual ~SiteEngagementEvictionPolicy();
+  SiteEngagementEvictionPolicy(storage::StorageType type,
+                               storage::QuotaManager* manager,
+                               content::BrowserContext* browser_context);
+  ~SiteEngagementEvictionPolicy() override;
 
   // Overridden from storage::QuotaEvictionPolicy:
   void GetEvictionOrigin(const scoped_refptr<storage::SpecialStoragePolicy>&
                              special_storage_policy,
-                         const std::map<GURL, int64>& usage_map,
-                         int64 global_quota,
                          const storage::GetOriginCallback& callback) override;
 
  private:
   friend class SiteEngagementEvictionPolicyTest;
+  friend class SiteEngagementEvictionPolicyWithQuotaManagerTest;
+
+  static void SetSiteEngagementScoreProviderCallback(
+      base::Callback<SiteEngagementScoreProvider*(content::BrowserContext*)>
+          callback);
 
   static GURL CalculateEvictionOrigin(
       const scoped_refptr<storage::SpecialStoragePolicy>&
           special_storage_policy,
       SiteEngagementScoreProvider* score_provider,
       const std::map<GURL, int64>& usage_map,
       int64 global_quota);
 
-  content::BrowserContext* browser_context_;
+  void DidGetGlobalUsage(storage::StorageType type, int64, int64);
+  void DidGetGlobalQuota(storage::QuotaStatusCode status, int64 quota);
+  void OnQuotaTasksCompleted();
+
+  const storage::StorageType type_;
+  storage::QuotaManager* const manager_;
+  content::BrowserContext* const browser_context_;
+
+  scoped_refptr<storage::SpecialStoragePolicy> special_storage_policy_;
+  std::map<GURL, int64> usage_map_;
+  int64 global_quota_;
+
+  int remaining_tasks_;
+
+  storage::GetOriginCallback eviction_origin_callback_;
+
+  base::WeakPtrFactory<SiteEngagementEvictionPolicy> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(SiteEngagementEvictionPolicy);
 };
 
 #endif  // CHROME_BROWSER_ENGAGEMENT_SITE_ENGAGEMENT_EVICTION_POLICY_H_