chrome/browser/browsing_data/mock_browsing_data_database_helper.h

   1 // Copyright (c) 2012 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_DATABASE_HELPER_H_
   6 #define CHROME_BROWSER_BROWSING_DATA_MOCK_BROWSING_DATA_DATABASE_HELPER_H_
   7
   8 #include <list>
   9 #include <map>
  10 #include <string>
  11
  12 #include "base/callback.h"
  13 #include "chrome/browser/browsing_data/browsing_data_database_helper.h"
  14
  15 // Mock for BrowsingDataDatabaseHelper.
  16 // Use AddDatabaseSamples() or add directly to response_ list, then call
  17 // Notify().
  18 class MockBrowsingDataDatabaseHelper : public BrowsingDataDatabaseHelper {
  19  public:
  20   explicit MockBrowsingDataDatabaseHelper(Profile* profile);
  21
  22   void StartFetching(const FetchCallback& callback) override;
  23
  24   void DeleteDatabase(const std::string& origin,
  25                       const std::string& name) override;
  26
  27   // Adds some DatabaseInfo samples.
  28   void AddDatabaseSamples();
  29
  30   // Notifies the callback.
  31   void Notify();
  32
  33   // Marks all databases as existing.
  34   void Reset();
  35
  36   // Returns true if all databases since the last Reset() invokation were
  37   // deleted.
  38   bool AllDeleted();
  39
  40   std::string last_deleted_origin_;
  41
  42   std::string last_deleted_db_;
  43
  44  private:
  45   ~MockBrowsingDataDatabaseHelper() override;
  46
  47   FetchCallback callback_;
  48
  49   // Stores which databases exist.
  50   std::map<const std::string, bool> databases_;
  51
  52   std::list<DatabaseInfo> response_;
  53
  54   DISALLOW_COPY_AND_ASSIGN(MockBrowsingDataDatabaseHelper);
  55 };
  56
  57 #endif  // CHROME_BROWSER_BROWSING_DATA_MOCK_BROWSING_DATA_DATABASE_HELPER_H_