Adds error handling support for the SwReporter launcher.

chrome/browser/safe_browsing/srt_chrome_prompt_impl.h

 // Copyright 2017 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROME_BROWSER_SAFE_BROWSING_SRT_CHROME_PROMPT_IMPL_H_
 #define CHROME_BROWSER_SAFE_BROWSING_SRT_CHROME_PROMPT_IMPL_H_
 
 #include "components/chrome_cleaner/public/interfaces/chrome_prompt.mojom.h"
 #include "mojo/public/cpp/bindings/binding.h"
 
 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 implementation. :-)

[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;
 
   void PromptUser(
       std::vector<chrome_cleaner::mojom::UwSPtr> removable_uws_found,
       chrome_cleaner::mojom::ElevationStatus elevation_status,
       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 using a global? perhaps by making
the methods in ChromePromptImpl you care about virtual, and allowing the test to
provide a factory function that will create its test-specific subclass of it?
using SwReporterTestingDelegate somehow?

[ftirelo, 2017/04/24 15:47:45]

Good idea. Changed as suggested in srt_fetcher_win.cc to have the
SwReporterTestingDelegate set it.

 };

[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
 
 #endif  // CHROME_BROWSER_SAFE_BROWSING_SRT_CHROME_PROMPT_IMPL_H_