| OLD | NEW |
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. | 1 // Copyright 2017 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 COMPONENTS_DOWNLOAD_INTERNAL_TEST_TEST_STORE_H_ | 5 #ifndef COMPONENTS_DOWNLOAD_INTERNAL_TEST_TEST_STORE_H_ |
| 6 #define COMPONENTS_DOWNLOAD_INTERNAL_TEST_TEST_STORE_H_ | 6 #define COMPONENTS_DOWNLOAD_INTERNAL_TEST_TEST_STORE_H_ |
| 7 | 7 |
| 8 #include <string> | 8 #include <string> |
| 9 #include <vector> | 9 #include <vector> |
| 10 | 10 |
| 11 #include "base/callback.h" |
| 11 #include "base/macros.h" | 12 #include "base/macros.h" |
| 12 #include "components/download/internal/store.h" | 13 #include "components/download/internal/store.h" |
| 13 | 14 |
| 14 namespace download { | 15 namespace download { |
| 15 | 16 |
| 16 struct Entry; | 17 struct Entry; |
| 17 | 18 |
| 18 namespace test { | 19 namespace test { |
| 19 | 20 |
| 20 class TestStore : public Store { | 21 class TestStore : public Store { |
| 21 public: | 22 public: |
| 22 TestStore(); | 23 TestStore(); |
| 23 ~TestStore() override; | 24 ~TestStore() override; |
| 24 | 25 |
| 25 // Store implementation. | 26 // Store implementation. |
| 26 bool IsInitialized() override; | 27 bool IsInitialized() override; |
| 27 void Initialize(InitCallback callback) override; | 28 void Initialize(InitCallback callback) override; |
| 28 void Destroy(StoreCallback callback) override; | |
| 29 void Update(const Entry& entry, StoreCallback callback) override; | 29 void Update(const Entry& entry, StoreCallback callback) override; |
| 30 void Remove(const std::string& guid, StoreCallback callback) override; | 30 void Remove(const std::string& guid, StoreCallback callback) override; |
| 31 | 31 |
| 32 // Callback trigger methods. | 32 // Callback trigger methods. |
| 33 void TriggerInit(bool success, std::unique_ptr<std::vector<Entry>> entries); | 33 void TriggerInit(bool success, std::unique_ptr<std::vector<Entry>> entries); |
| 34 void TriggerDestroy(bool success); | |
| 35 void TriggerUpdate(bool success); | 34 void TriggerUpdate(bool success); |
| 36 void TriggerRemove(bool success); | 35 void TriggerRemove(bool success); |
| 37 | 36 |
| 38 // Parameter access methods. | 37 // Parameter access methods. |
| 39 const Entry* LastUpdatedEntry() const; | 38 const Entry* LastUpdatedEntry() const; |
| 40 std::string LastRemovedEntry() const; | 39 std::string LastRemovedEntry() const; |
| 41 bool init_called() const { return init_called_; } | 40 bool init_called() const { return init_called_; } |
| 42 bool destroy_called() const { return destroy_called_; } | |
| 43 const std::vector<Entry>& updated_entries() const { return updated_entries_; } | 41 const std::vector<Entry>& updated_entries() const { return updated_entries_; } |
| 44 const std::vector<std::string>& removed_entries() const { | 42 const std::vector<std::string>& removed_entries() const { |
| 45 return removed_entries_; | 43 return removed_entries_; |
| 46 } | 44 } |
| 47 | 45 |
| 48 private: | 46 private: |
| 49 bool ready_; | 47 bool ready_; |
| 50 | 48 |
| 51 bool init_called_; | 49 bool init_called_; |
| 52 bool destroy_called_; | |
| 53 | 50 |
| 54 std::vector<Entry> updated_entries_; | 51 std::vector<Entry> updated_entries_; |
| 55 std::vector<std::string> removed_entries_; | 52 std::vector<std::string> removed_entries_; |
| 56 | 53 |
| 57 InitCallback init_callback_; | 54 InitCallback init_callback_; |
| 58 StoreCallback destroy_callback_; | |
| 59 StoreCallback update_callback_; | 55 StoreCallback update_callback_; |
| 60 StoreCallback remove_callback_; | 56 StoreCallback remove_callback_; |
| 61 | 57 |
| 62 DISALLOW_COPY_AND_ASSIGN(TestStore); | 58 DISALLOW_COPY_AND_ASSIGN(TestStore); |
| 63 }; | 59 }; |
| 64 | 60 |
| 65 } // namespace test | 61 } // namespace test |
| 66 } // namespace download | 62 } // namespace download |
| 67 | 63 |
| 68 #endif // COMPONENTS_DOWNLOAD_INTERNAL_TEST_TEST_STORE_H_ | 64 #endif // COMPONENTS_DOWNLOAD_INTERNAL_TEST_TEST_STORE_H_ |
| OLD | NEW |