1// Copyright 2014 The Chromium Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style license that can be 3// found in the LICENSE file. 4 5#ifndef CHROME_BROWSER_BROWSING_DATA_MOCK_BROWSING_DATA_CHANNEL_ID_HELPER_H_ 6#define CHROME_BROWSER_BROWSING_DATA_MOCK_BROWSING_DATA_CHANNEL_ID_HELPER_H_ 7 8#include <map> 9#include <string> 10 11#include "chrome/browser/browsing_data/browsing_data_channel_id_helper.h" 12 13// Mock for BrowsingDataChannelIDHelper. 14class MockBrowsingDataChannelIDHelper 15 : public BrowsingDataChannelIDHelper { 16 public: 17 MockBrowsingDataChannelIDHelper(); 18 19 // BrowsingDataChannelIDHelper methods. 20 virtual void StartFetching(const FetchResultCallback& callback) OVERRIDE; 21 virtual void DeleteChannelID(const std::string& server_id) OVERRIDE; 22 23 // Adds a channel_id sample. 24 void AddChannelIDSample(const std::string& server_id); 25 26 // Notifies the callback. 27 void Notify(); 28 29 // Marks all channel_ids as existing. 30 void Reset(); 31 32 // Returns true if all channel_ids since the last Reset() invocation 33 // were deleted. 34 bool AllDeleted(); 35 36 private: 37 virtual ~MockBrowsingDataChannelIDHelper(); 38 39 FetchResultCallback callback_; 40 41 net::ChannelIDStore::ChannelIDList channel_id_list_; 42 43 // Stores which channel_ids exist. 44 std::map<const std::string, bool> channel_ids_; 45 46 DISALLOW_COPY_AND_ASSIGN(MockBrowsingDataChannelIDHelper); 47}; 48 49#endif // CHROME_BROWSER_BROWSING_DATA_MOCK_BROWSING_DATA_CHANNEL_ID_HELPER_H_ 50