base/prefs/testing_pref_store.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 BASE_PREFS_TESTING_PREF_STORE_H_
   6 #define BASE_PREFS_TESTING_PREF_STORE_H_
   7
   8 #include <string>
   9
  10 #include "base/basictypes.h"
  11 #include "base/compiler_specific.h"
  12 #include "base/observer_list.h"
  13 #include "base/prefs/persistent_pref_store.h"
  14 #include "base/prefs/pref_value_map.h"
  15
  16 // |TestingPrefStore| is a preference store implementation that allows tests to
  17 // explicitly manipulate the contents of the store, triggering notifications
  18 // where appropriate.
  19 class TestingPrefStore : public PersistentPrefStore {
  20  public:
  21   TestingPrefStore();
  22
  23   // Overriden from PrefStore.
  24   bool GetValue(const std::string& key,
  25                 const base::Value** result) const override;
  26   void AddObserver(PrefStore::Observer* observer) override;
  27   void RemoveObserver(PrefStore::Observer* observer) override;
  28   bool HasObservers() const override;
  29   bool IsInitializationComplete() const override;
  30
  31   // PersistentPrefStore overrides:
  32   bool GetMutableValue(const std::string& key, base::Value** result) override;
  33   void ReportValueChanged(const std::string& key, uint32 flags) override;
  34   void SetValue(const std::string& key,
  35                 base::Value* value,
  36                 uint32 flags) override;
  37   void SetValueSilently(const std::string& key,
  38                         base::Value* value,
  39                         uint32 flags) override;
  40   void RemoveValue(const std::string& key, uint32 flags) override;
  41   bool ReadOnly() const override;
  42   PrefReadError GetReadError() const override;
  43   PersistentPrefStore::PrefReadError ReadPrefs() override;
  44   void ReadPrefsAsync(ReadErrorDelegate* error_delegate) override;
  45   void CommitPendingWrite() override;
  46
  47   // Marks the store as having completed initialization.
  48   void SetInitializationCompleted();
  49
  50   // Used for tests to trigger notifications explicitly.
  51   void NotifyPrefValueChanged(const std::string& key);
  52   void NotifyInitializationCompleted();
  53
  54   // Some convenience getters/setters.
  55   void SetString(const std::string& key, const std::string& value);
  56   void SetInteger(const std::string& key, int value);
  57   void SetBoolean(const std::string& key, bool value);
  58
  59   bool GetString(const std::string& key, std::string* value) const;
  60   bool GetInteger(const std::string& key, int* value) const;
  61   bool GetBoolean(const std::string& key, bool* value) const;
  62
  63   // Determines whether ReadPrefsAsync completes immediately. Defaults to false
  64   // (non-blocking). To block, invoke this with true (blocking) before the call
  65   // to ReadPrefsAsync. To unblock, invoke again with false (non-blocking) after
  66   // the call to ReadPrefsAsync.
  67   void SetBlockAsyncRead(bool block_async_read);
  68
  69   // Getter and Setter methods for setting and getting the state of the
  70   // |TestingPrefStore|.
  71   virtual void set_read_only(bool read_only);
  72   void set_read_success(bool read_success);
  73   void set_read_error(PersistentPrefStore::PrefReadError read_error);
  74   bool committed() { return committed_; }
  75
  76  protected:
  77   ~TestingPrefStore() override;
  78
  79  private:
  80   // Stores the preference values.
  81   PrefValueMap prefs_;
  82
  83   // Flag that indicates if the PrefStore is read-only
  84   bool read_only_;
  85
  86   // The result to pass to PrefStore::Observer::OnInitializationCompleted
  87   bool read_success_;
  88
  89   // The result to return from ReadPrefs or ReadPrefsAsync.
  90   PersistentPrefStore::PrefReadError read_error_;
  91
  92   // Whether a call to ReadPrefsAsync should block.
  93   bool block_async_read_;
  94
  95   // Whether there is a pending call to ReadPrefsAsync.
  96   bool pending_async_read_;
  97
  98   // Whether initialization has been completed.
  99   bool init_complete_;
 100
 101   // Whether the store contents have been committed to disk since the last
 102   // mutation.
 103   bool committed_;
 104
 105   scoped_ptr<ReadErrorDelegate> error_delegate_;
 106   ObserverList<PrefStore::Observer, true> observers_;
 107
 108   DISALLOW_COPY_AND_ASSIGN(TestingPrefStore);
 109 };
 110
 111 #endif  // BASE_PREFS_TESTING_PREF_STORE_H_