1 files changed, 42 insertions, 0 deletions
diff --git a/chrome/browser/mock_browsing_data_quota_helper.cc b/chrome/browser/mock_browsing_data_quota_helper.cc
new file mode 100644
index 0000000..45d2c41
--- /dev/null
+++ b/chrome/browser/mock_browsing_data_quota_helper.cc
@@ -0,0 +1,42 @@
+// Copyright (c) 2011 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/mock_browsing_data_quota_helper.h"
+
+MockBrowsingDataQuotaHelper::MockBrowsingDataQuotaHelper(Profile* profile)
+    : BrowsingDataQuotaHelper(BrowserThread::GetMessageLoopProxyForThread(
+        BrowserThread::IO)) {}
+
+MockBrowsingDataQuotaHelper::~MockBrowsingDataQuotaHelper() {}
+
+void MockBrowsingDataQuotaHelper::StartFetching(
+    FetchResultCallback* callback) {
+  callback_.reset(callback);
+}
+
+void MockBrowsingDataQuotaHelper::CancelNotification() {
+  callback_.reset(NULL);
+}
+
+void MockBrowsingDataQuotaHelper::AddHost(
+    const std::string& host,
+    int64 temporary_usage,
+    int64 persistent_usage) {
+  response_.push_back(QuotaInfo(
+      host,
+      temporary_usage,
+      persistent_usage));
+}
+
+void MockBrowsingDataQuotaHelper::AddQuotaSamples() {
+  AddHost("quotahost1", 1, 2);
+  AddHost("quotahost2", 10, 20);
+}
+
+void MockBrowsingDataQuotaHelper::Notify() {
+  CHECK(callback_.get());
+  callback_->Run(response_);
+  callback_.reset();
+  response_.clear();
+}