blob: e925e0b8fbcfa86211777111bff0fb7776a7414f [file] [log] [blame]
// 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_
#pragma once
#include <map>
#include <vector>
#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_