| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef EXTENSIONS_BROWSER_VALUE_STORE_TESTING_VALUE_STORE_H_ | 5 #ifndef EXTENSIONS_BROWSER_VALUE_STORE_TESTING_VALUE_STORE_H_ |
| 6 #define EXTENSIONS_BROWSER_VALUE_STORE_TESTING_VALUE_STORE_H_ | 6 #define EXTENSIONS_BROWSER_VALUE_STORE_TESTING_VALUE_STORE_H_ |
| 7 | 7 |
| 8 #include "base/compiler_specific.h" | 8 #include "base/compiler_specific.h" |
| 9 #include "extensions/browser/value_store/value_store.h" | 9 #include "extensions/browser/value_store/value_store.h" |
| 10 | 10 |
| 11 // ValueStore for testing, with an in-memory storage but the ability to | 11 // ValueStore for testing, with an in-memory storage but the ability to |
| 12 // optionally fail all operations. | 12 // optionally fail all operations. |
| 13 class TestingValueStore : public ValueStore { | 13 class TestingValueStore : public ValueStore { |
| 14 public: | 14 public: |
| 15 TestingValueStore(); | 15 TestingValueStore(); |
| 16 virtual ~TestingValueStore(); | 16 ~TestingValueStore() override; |
| 17 | 17 |
| 18 // Sets the error code for requests. If OK, errors won't be thrown. | 18 // Sets the error code for requests. If OK, errors won't be thrown. |
| 19 // Defaults to OK. | 19 // Defaults to OK. |
| 20 void set_error_code(ErrorCode error_code) { error_code_ = error_code; } | 20 void set_error_code(ErrorCode error_code) { error_code_ = error_code; } |
| 21 | 21 |
| 22 // Accessors for the number of reads/writes done by this value store. Each | 22 // Accessors for the number of reads/writes done by this value store. Each |
| 23 // Get* operation (except for the BytesInUse ones) counts as one read, and | 23 // Get* operation (except for the BytesInUse ones) counts as one read, and |
| 24 // each Set*/Remove/Clear operation counts as one write. This is useful in | 24 // each Set*/Remove/Clear operation counts as one write. This is useful in |
| 25 // tests seeking to assert that some number of reads/writes to their | 25 // tests seeking to assert that some number of reads/writes to their |
| 26 // underlying value store have (or have not) happened. | 26 // underlying value store have (or have not) happened. |
| 27 int read_count() { return read_count_; } | 27 int read_count() { return read_count_; } |
| 28 int write_count() { return write_count_; } | 28 int write_count() { return write_count_; } |
| 29 | 29 |
| 30 // ValueStore implementation. | 30 // ValueStore implementation. |
| 31 virtual size_t GetBytesInUse(const std::string& key) override; | 31 size_t GetBytesInUse(const std::string& key) override; |
| 32 virtual size_t GetBytesInUse(const std::vector<std::string>& keys) override; | 32 size_t GetBytesInUse(const std::vector<std::string>& keys) override; |
| 33 virtual size_t GetBytesInUse() override; | 33 size_t GetBytesInUse() override; |
| 34 virtual ReadResult Get(const std::string& key) override; | 34 ReadResult Get(const std::string& key) override; |
| 35 virtual ReadResult Get(const std::vector<std::string>& keys) override; | 35 ReadResult Get(const std::vector<std::string>& keys) override; |
| 36 virtual ReadResult Get() override; | 36 ReadResult Get() override; |
| 37 virtual WriteResult Set( | 37 WriteResult Set(WriteOptions options, |
| 38 WriteOptions options, | 38 const std::string& key, |
| 39 const std::string& key, | 39 const base::Value& value) override; |
| 40 const base::Value& value) override; | 40 WriteResult Set(WriteOptions options, |
| 41 virtual WriteResult Set( | 41 const base::DictionaryValue& values) override; |
| 42 WriteOptions options, const base::DictionaryValue& values) override; | 42 WriteResult Remove(const std::string& key) override; |
| 43 virtual WriteResult Remove(const std::string& key) override; | 43 WriteResult Remove(const std::vector<std::string>& keys) override; |
| 44 virtual WriteResult Remove(const std::vector<std::string>& keys) override; | 44 WriteResult Clear() override; |
| 45 virtual WriteResult Clear() override; | |
| 46 // TestingValueStores can't get corrupted (they're all in-memory), so these | 45 // TestingValueStores can't get corrupted (they're all in-memory), so these |
| 47 // just return true. | 46 // just return true. |
| 48 virtual bool Restore() override; | 47 bool Restore() override; |
| 49 virtual bool RestoreKey(const std::string& key) override; | 48 bool RestoreKey(const std::string& key) override; |
| 50 | 49 |
| 51 private: | 50 private: |
| 52 scoped_ptr<ValueStore::Error> TestingError(); | 51 scoped_ptr<ValueStore::Error> TestingError(); |
| 53 | 52 |
| 54 base::DictionaryValue storage_; | 53 base::DictionaryValue storage_; |
| 55 int read_count_; | 54 int read_count_; |
| 56 int write_count_; | 55 int write_count_; |
| 57 ErrorCode error_code_; | 56 ErrorCode error_code_; |
| 58 | 57 |
| 59 DISALLOW_COPY_AND_ASSIGN(TestingValueStore); | 58 DISALLOW_COPY_AND_ASSIGN(TestingValueStore); |
| 60 }; | 59 }; |
| 61 | 60 |
| 62 #endif // EXTENSIONS_BROWSER_VALUE_STORE_TESTING_VALUE_STORE_H_ | 61 #endif // EXTENSIONS_BROWSER_VALUE_STORE_TESTING_VALUE_STORE_H_ |
| OLD | NEW |