chrome/browser/browsing_data/mock_browsing_data_channel_id_helper.h

   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.
  14 class MockBrowsingDataChannelIDHelper
  15     : public BrowsingDataChannelIDHelper {
  16  public:
  17   MockBrowsingDataChannelIDHelper();
  18
  19   // BrowsingDataChannelIDHelper methods.
  20   void StartFetching(const FetchResultCallback& callback) override;
  21   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   ~MockBrowsingDataChannelIDHelper() override;
  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_