1 files changed, 60 insertions, 0 deletions
diff --git a/chrome/browser/mock_browsing_data_database_helper.h b/chrome/browser/mock_browsing_data_database_helper.h
new file mode 100644
index 0000000..8e075db
--- /dev/null
+++ b/chrome/browser/mock_browsing_data_database_helper.h
@@ -0,0 +1,60 @@
+// Copyright (c) 2010 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_MOCK_BROWSING_DATA_DATABASE_HELPER_H_
+#define CHROME_BROWSER_MOCK_BROWSING_DATA_DATABASE_HELPER_H_
+
+#include <map>
+
+#include "base/callback.h"
+
+#include "chrome/browser/browsing_data_database_helper.h"
+
+// Mock for BrowsingDataDatabaseHelper.
+// Use AddDatabaseSamples() or add directly to response_ vector, then call
+// Notify().
+class MockBrowsingDataDatabaseHelper : public BrowsingDataDatabaseHelper {
+ public:
+  explicit MockBrowsingDataDatabaseHelper(Profile* profile);
+
+  virtual void StartFetching(
+      Callback1<const std::vector<DatabaseInfo>& >::Type* callback);
+
+  virtual void CancelNotification();
+
+  virtual void DeleteDatabase(const std::string& origin,
+      const std::string& name);
+
+  // Adds some DatabaseInfo samples.
+  void AddDatabaseSamples();
+
+  // Notifies the callback.
+  void Notify();
+
+  // Marks all databases as existing.
+  void Reset();
+
+  // Returns true if all databases since the last Reset() invokation were
+  // deleted.
+  bool AllDeleted();
+
+  std::string last_deleted_origin_;
+
+  std::string last_deleted_db_;
+
+ private:
+  virtual ~MockBrowsingDataDatabaseHelper();
+
+  Profile* profile_;
+
+  scoped_ptr<Callback1<const std::vector<DatabaseInfo>& >::Type >
+      callback_;
+
+  // Stores which databases exist.
+  std::map<const std::string, bool> databases_;
+
+  std::vector<DatabaseInfo> response_;
+};
+
+#endif  // CHROME_BROWSER_MOCK_BROWSING_DATA_DATABASE_HELPER_H_