Chromium Code Reviews| Index: chrome/browser/safe_browsing/srt_chrome_prompt_impl.h |
| diff --git a/chrome/browser/safe_browsing/srt_chrome_prompt_impl.h b/chrome/browser/safe_browsing/srt_chrome_prompt_impl.h |
| index bf14b30ff80655a3d21a277e5e7072eeae42b919..9d03171371329074be9d45e506d774958a81f3c5 100644 |
| --- a/chrome/browser/safe_browsing/srt_chrome_prompt_impl.h |
| +++ b/chrome/browser/safe_browsing/srt_chrome_prompt_impl.h |
| @@ -13,6 +13,13 @@ namespace safe_browsing { |
| // Implementation of the ChromePrompt Mojo interface. |
| class ChromePromptImpl : public chrome_cleaner::mojom::ChromePrompt { |
| public: |
| + // Implements specific mocking actions for tests. |
|
grt (UTC plus 2)
2017/04/21 11:10:58
this is an interface for such, not an implementati
ftirelo
2017/04/24 15:47:45
Done.
|
| + class TestDelegate { |
| + public: |
| + virtual void OnPromptUser() = 0; |
| + virtual void OnConnectionError() = 0; |
| + }; |
| + |
| explicit ChromePromptImpl(chrome_cleaner::mojom::ChromePromptRequest request); |
| ~ChromePromptImpl() override; |
| @@ -22,8 +29,19 @@ class ChromePromptImpl : public chrome_cleaner::mojom::ChromePrompt { |
| const chrome_cleaner::mojom::ChromePrompt::PromptUserCallback& callback) |
| override; |
| + // Handles the case when the remote end has been closed, by performing the |
| + // necessary cleanups if the prompt dialog is being shown to the user. |
| + void OnConnectionClosed(); |
| + |
| + // Handles the case when a mojo::ReportBadMessage has been explicitely |
| + // reported. |
| + void OnConnectionError(const std::string& message); |
| + |
| + static void SetTestDelegate(TestDelegate* test_delegate); |
| + |
| private: |
| mojo::Binding<chrome_cleaner::mojom::ChromePrompt> binding_; |
| + static TestDelegate* test_delegate_; |
|
grt (UTC plus 2)
2017/04/21 11:10:58
is it possible to inject the delegate without usin
ftirelo
2017/04/24 15:47:45
Good idea. Changed as suggested in srt_fetcher_win
|
| }; |
|
grt (UTC plus 2)
2017/04/21 11:10:58
DISALLOW_COPY_AND_ASSIGN?
ftirelo
2017/04/24 15:47:45
Done.
|
| } // namespace safe_browsing |