| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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 CHROME_BROWSER_VALUE_STORE_TESTING_VALUE_STORE_H_ | 5 #ifndef CHROME_BROWSER_VALUE_STORE_TESTING_VALUE_STORE_H_ |
| 6 #define CHROME_BROWSER_VALUE_STORE_TESTING_VALUE_STORE_H_ | 6 #define CHROME_BROWSER_VALUE_STORE_TESTING_VALUE_STORE_H_ |
| 7 | 7 |
| 8 #include "base/compiler_specific.h" | 8 #include "base/compiler_specific.h" |
| 9 #include "chrome/browser/value_store/value_store.h" | 9 #include "chrome/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 virtual ~TestingValueStore(); |
| 17 | 17 |
| 18 // Sets whether to fail all requests (default is false). | 18 // Sets the error code for requests. If OK, errors won't be thrown. |
| 19 void SetFailAllRequests(bool fail_all_requests); | 19 // Defaults to OK. |
| 20 void set_error_code(ErrorCode error_code) { error_code_ = error_code; } |
| 20 | 21 |
| 21 // 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 |
| 22 // 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 |
| 23 // 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 |
| 24 // 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 |
| 25 // underlying value store have (or have not) happened. | 26 // underlying value store have (or have not) happened. |
| 26 int read_count() { return read_count_; } | 27 int read_count() { return read_count_; } |
| 27 int write_count() { return write_count_; } | 28 int write_count() { return write_count_; } |
| 28 | 29 |
| 29 // ValueStore implementation. | 30 // ValueStore implementation. |
| 30 virtual size_t GetBytesInUse(const std::string& key) OVERRIDE; | 31 virtual size_t GetBytesInUse(const std::string& key) OVERRIDE; |
| 31 virtual size_t GetBytesInUse(const std::vector<std::string>& keys) OVERRIDE; | 32 virtual size_t GetBytesInUse(const std::vector<std::string>& keys) OVERRIDE; |
| 32 virtual size_t GetBytesInUse() OVERRIDE; | 33 virtual size_t GetBytesInUse() OVERRIDE; |
| 33 virtual ReadResult Get(const std::string& key) OVERRIDE; | 34 virtual ReadResult Get(const std::string& key) OVERRIDE; |
| 34 virtual ReadResult Get(const std::vector<std::string>& keys) OVERRIDE; | 35 virtual ReadResult Get(const std::vector<std::string>& keys) OVERRIDE; |
| 35 virtual ReadResult Get() OVERRIDE; | 36 virtual ReadResult Get() OVERRIDE; |
| 36 virtual WriteResult Set( | 37 virtual WriteResult Set( |
| 37 WriteOptions options, | 38 WriteOptions options, |
| 38 const std::string& key, | 39 const std::string& key, |
| 39 const Value& value) OVERRIDE; | 40 const Value& value) OVERRIDE; |
| 40 virtual WriteResult Set( | 41 virtual WriteResult Set( |
| 41 WriteOptions options, const DictionaryValue& values) OVERRIDE; | 42 WriteOptions options, const DictionaryValue& values) OVERRIDE; |
| 42 virtual WriteResult Remove(const std::string& key) OVERRIDE; | 43 virtual WriteResult Remove(const std::string& key) OVERRIDE; |
| 43 virtual WriteResult Remove(const std::vector<std::string>& keys) OVERRIDE; | 44 virtual WriteResult Remove(const std::vector<std::string>& keys) OVERRIDE; |
| 44 virtual WriteResult Clear() OVERRIDE; | 45 virtual WriteResult Clear() OVERRIDE; |
| 45 | 46 |
| 46 private: | 47 private: |
| 48 scoped_ptr<ValueStore::Error> TestingError(); |
| 49 |
| 47 DictionaryValue storage_; | 50 DictionaryValue storage_; |
| 48 int read_count_; | 51 int read_count_; |
| 49 int write_count_; | 52 int write_count_; |
| 50 bool fail_all_requests_; | 53 ErrorCode error_code_; |
| 51 | 54 |
| 52 DISALLOW_COPY_AND_ASSIGN(TestingValueStore); | 55 DISALLOW_COPY_AND_ASSIGN(TestingValueStore); |
| 53 }; | 56 }; |
| 54 | 57 |
| 55 #endif // CHROME_BROWSER_VALUE_STORE_TESTING_VALUE_STORE_H_ | 58 #endif // CHROME_BROWSER_VALUE_STORE_TESTING_VALUE_STORE_H_ |
| OLD | NEW |