MockColorChooser

components/test_runner/mock_color_chooser.h

Index: components/test_runner/mock_color_chooser.h
diff --git a/components/test_runner/mock_color_chooser.h b/components/test_runner/mock_color_chooser.h
index 8e6d87ae817bf10fe25f07262e47cd4e3fb65105..6f5746ea559012bf10fff7d05bc3e06418ac319c 100644
--- a/components/test_runner/mock_color_chooser.h
+++ b/components/test_runner/mock_color_chooser.h
@@ -17,8 +17,8 @@ class WebTestDelegate;
 
 class MockColorChooser : public blink::WebColorChooser {
  public:
-  // Caller has to guarantee that |client| and |delegate| are alive
-  // until |WebColorChooserClient::didEndChooser| is called.
+  // Caller has to guarantee that |client| is alive until
+  // |WebColorChooserClient::didEndChooser| is called.
   // Caller has to guarantee that |test_runner| lives longer
   // than MockColorChooser.
   MockColorChooser(blink::WebColorChooserClient* client,
@@ -31,6 +31,10 @@ class MockColorChooser : public blink::WebColorChooser {
   void endChooser() override;
 
   void InvokeDidEndChooser();
+  // MockColorChooser is owned by ColorChooserUIController which is garbage
+  // collected. That means this might be destroyed after WebTestDelegate is
+  // destryed, so we use this to clear |delegate_|.
+  void SetDelegate(WebTestDelegate*);
 
  private:
   blink::WebColorChooserClient* client_;