Chromium Code Reviews| 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 EXTENSIONS_BROWSER_PRELOAD_CHECK_TEST_UTIL_H_ | 5 #ifndef EXTENSIONS_BROWSER_PRELOAD_CHECK_TEST_UTIL_H_ |
| 6 #define EXTENSIONS_BROWSER_PRELOAD_CHECK_TEST_UTIL_H_ | 6 #define EXTENSIONS_BROWSER_PRELOAD_CHECK_TEST_UTIL_H_ |
| 7 | 7 |
| 8 #include <memory> | 8 #include <memory> |
| 9 | 9 |
| 10 #include "base/macros.h" | 10 #include "base/macros.h" |
| (...skipping 12 matching lines...) Expand all Loading... | |
| 23 public: | 23 public: |
| 24 PreloadCheckRunner(); | 24 PreloadCheckRunner(); |
| 25 virtual ~PreloadCheckRunner(); | 25 virtual ~PreloadCheckRunner(); |
| 26 | 26 |
| 27 // Starts the check, providing OnCheckComplete as the callback. | 27 // Starts the check, providing OnCheckComplete as the callback. |
| 28 void Run(PreloadCheck* check); | 28 void Run(PreloadCheck* check); |
| 29 | 29 |
| 30 // Starts the check and waits for its callback to execute. | 30 // Starts the check and waits for its callback to execute. |
| 31 void RunUntilComplete(PreloadCheck* check); | 31 void RunUntilComplete(PreloadCheck* check); |
| 32 | 32 |
| 33 // Runs the message loop until OnCheckComplete is called. | |
| 34 void WaitForCompletion(); | |
| 35 | |
| 36 // Runs the message loop until idle. Useful to see whether OnCheckComplete is | 33 // Runs the message loop until idle. Useful to see whether OnCheckComplete is |
| 37 // called without waiting indefinitely. | 34 // called without waiting indefinitely. |
| 38 void WaitForIdle(); | 35 void WaitForIdle(); |
| 39 | 36 |
| 37 // Runs the message loop until OnCheckComplete is called. | |
| 38 void WaitForComplete(); | |
| 39 | |
| 40 PreloadCheck::ResultCallback GetCallback(); | 40 PreloadCheck::ResultCallback GetCallback(); |
| 41 | 41 |
| 42 const PreloadCheck::Errors& errors() const { return errors_; } | 42 const PreloadCheck::Errors& errors() const { return errors_; } |
| 43 bool called() const { return called_; } | 43 bool called() const { return called_; } |
| 44 | 44 |
| 45 private: | 45 private: |
| 46 void OnCheckComplete(PreloadCheck::Errors errors); | 46 void OnCheckComplete(PreloadCheck::Errors errors); |
| 47 | 47 |
| 48 PreloadCheck::Errors errors_; | 48 PreloadCheck::Errors errors_; |
| 49 bool called_; | 49 bool called_; |
| 50 | 50 |
| 51 // Using a RunLoop data member would trigger tricky timing troubles. | 51 // Using a RunLoop data member would trigger tricky timing troubles. |
| 52 std::unique_ptr<base::RunLoop> run_loop_; | 52 std::unique_ptr<base::RunLoop> run_loop_; |
| 53 | 53 |
| 54 DISALLOW_COPY_AND_ASSIGN(PreloadCheckRunner); | 54 DISALLOW_COPY_AND_ASSIGN(PreloadCheckRunner); |
| 55 }; | 55 }; |
| 56 | 56 |
| 57 // Stub for a PreloadCheck that returns the desired error(s). | 57 // Stub for a PreloadCheck that calls the callback with the given error(s). |
| 58 class PreloadCheckStub : public PreloadCheck { | 58 class PreloadCheckStub : public PreloadCheck { |
| 59 public: | 59 public: |
| 60 PreloadCheckStub(); | 60 explicit PreloadCheckStub(const Errors& errors = Errors()); |
|
Devlin
2017/04/06 02:03:53
nit: I'd prefer we don't use default arguments on
michaelpg
2017/04/06 22:31:16
Done.
| |
| 61 ~PreloadCheckStub() override; | 61 ~PreloadCheckStub() override; |
| 62 | 62 |
| 63 void AddError(Error error); | |
| 64 void set_error_message(const base::string16& message) { message_ = message; } | |
| 65 | |
| 66 bool is_async() const { return is_async_; } | |
| 67 void set_is_async(bool is_async) { is_async_ = is_async; } | 63 void set_is_async(bool is_async) { is_async_ = is_async; } |
| 64 bool started() const { return started_; } | |
| 68 | 65 |
| 69 // PreloadCheck: | 66 // PreloadCheck: |
| 70 void Start(ResultCallback callback) override; | 67 void Start(ResultCallback callback) override; |
| 71 base::string16 GetErrorMessage() const override; | |
| 72 | 68 |
| 73 private: | 69 private: |
| 74 void RunCallback(ResultCallback callback); | 70 void RunCallback(ResultCallback callback); |
| 75 | 71 |
| 76 bool is_async_ = false; | 72 bool is_async_ = false; |
| 73 bool started_ = false; | |
| 77 Errors errors_; | 74 Errors errors_; |
| 78 base::string16 message_; | |
| 79 | 75 |
| 80 base::WeakPtrFactory<PreloadCheckStub> weak_ptr_factory_; | 76 base::WeakPtrFactory<PreloadCheckStub> weak_ptr_factory_; |
| 81 | 77 |
| 82 DISALLOW_COPY_AND_ASSIGN(PreloadCheckStub); | 78 DISALLOW_COPY_AND_ASSIGN(PreloadCheckStub); |
| 83 }; | 79 }; |
| 84 | 80 |
| 85 } // namespace extensions | 81 } // namespace extensions |
| 86 | 82 |
| 87 #endif // EXTENSIONS_BROWSER_PRELOAD_CHECK_TEST_UTIL_H_ | 83 #endif // EXTENSIONS_BROWSER_PRELOAD_CHECK_TEST_UTIL_H_ |
| OLD | NEW |