MockColorChooser

components/test_runner/test_runner.cc

 // Copyright 2014 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.
 
 #include "components/test_runner/test_runner.h"
 
 #include <stddef.h>
 #include <limits>
 #include <utility>
 
 #include "base/command_line.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/strings/nullable_string16.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "build/build_config.h"
 #include "components/test_runner/layout_and_paint_async_then.h"
 #include "components/test_runner/layout_dump.h"
+#include "components/test_runner/mock_color_chooser.h"
 #include "components/test_runner/mock_content_settings_client.h"
 #include "components/test_runner/mock_credential_manager_client.h"
 #include "components/test_runner/mock_screen_orientation_client.h"
 #include "components/test_runner/mock_web_document_subresource_filter.h"
 #include "components/test_runner/mock_web_speech_recognizer.h"
 #include "components/test_runner/mock_web_user_media_client.h"
 #include "components/test_runner/pixel_dump.h"
 #include "components/test_runner/spell_check_client.h"
 #include "components/test_runner/test_common.h"
 #include "components/test_runner/test_interfaces.h"
@@ skipping 1579 matching lines @@
       web_history_item_count_(0),
       test_interfaces_(interfaces),
       delegate_(nullptr),
       main_view_(nullptr),
       mock_content_settings_client_(
           new MockContentSettingsClient(&layout_test_runtime_flags_)),
       will_navigate_(false),
       credential_manager_client_(new MockCredentialManagerClient),
       mock_screen_orientation_client_(new MockScreenOrientationClient),
       spellcheck_(new SpellCheckClient(this)),
-      chooser_count_(0),
       previously_focused_view_(nullptr),
       is_web_platform_tests_mode_(false),
       effective_connection_type_(
           blink::WebEffectiveConnectionType::TypeUnknown),
       weak_factory_(this) {}
 
 TestRunner::~TestRunner() {}
 
 void TestRunner::Install(
     WebLocalFrame* frame,
     base::WeakPtr<TestRunnerForSpecificView> view_test_runner) {
   TestRunnerBindings::Install(weak_factory_.GetWeakPtr(), view_test_runner,
                               frame);
 }
 
 void TestRunner::SetDelegate(WebTestDelegate* delegate) {
   delegate_ = delegate;
   mock_content_settings_client_->SetDelegate(delegate);
   spellcheck_->SetDelegate(delegate);
   if (speech_recognizer_)
     speech_recognizer_->SetDelegate(delegate);
+  for (auto& chooser : active_color_choosers_) {
+    chooser->SetDelegate(delegate);
+  }
 }
 
 void TestRunner::SetMainView(WebView* web_view) {
   main_view_ = web_view;
   if (disable_v8_cache_)
     SetV8CacheDisabled(true);
 }
 
 void TestRunner::Reset() {
   is_web_platform_tests_mode_ = false;
@@ skipping 665 matching lines @@
       mock_screen_orientation_client_->UpdateDeviceOrientation(
           main_frame->toWebLocalFrame(), orientation);
     }
   }
 }
 
 void TestRunner::DisableMockScreenOrientation() {
   mock_screen_orientation_client_->SetDisabled(true);
 }
 
-void TestRunner::DidOpenChooser() {
-  chooser_count_++;
+void TestRunner::DidOpenChooser(MockColorChooser* color_chooser) {
+  DCHECK(active_color_choosers_.count(color_chooser) == 0);
+  active_color_choosers_.insert(color_chooser);
 }
 
-void TestRunner::DidCloseChooser() {
-  chooser_count_--;
-  DCHECK_LE(0, chooser_count_);
+void TestRunner::DidCloseChooser(MockColorChooser* color_chooser) {
+  active_color_choosers_.erase(color_chooser);
+  DCHECK_LE(0U, active_color_choosers_.size());
 }
 
 void TestRunner::SetPopupBlockingEnabled(bool block_popups) {
   delegate_->Preferences()->java_script_can_open_windows_automatically =
       !block_popups;
   delegate_->ApplyPreferences();
 }
 
 void TestRunner::SetJavaScriptCanAccessClipboard(bool can_access) {
   delegate_->Preferences()->java_script_can_access_clipboard = can_access;
@@ skipping 311 matching lines @@
 
 bool TestRunner::ShouldDumpJavaScriptDialogs() const {
   return layout_test_runtime_flags_.dump_javascript_dialogs();
 }
 
 void TestRunner::CloseWebInspector() {
   delegate_->CloseDevTools();
 }
 
 bool TestRunner::IsChooserShown() {
-  return 0 < chooser_count_;
+  return 0 < active_color_choosers_.size();
 }
 
 void TestRunner::EvaluateInWebInspector(int call_id,
                                         const std::string& script) {
   delegate_->EvaluateInWebInspector(call_id, script);
 }
 
 std::string TestRunner::EvaluateInWebInspectorOverlay(
     const std::string& script) {
   return delegate_->EvaluateInWebInspectorOverlay(script);
@@ skipping 137 matching lines @@
 
 void TestRunner::NotifyDone() {
   if (layout_test_runtime_flags_.wait_until_done() && !topLoadingFrame() &&
       !will_navigate_ && work_queue_.is_empty())
     delegate_->TestFinished();
   layout_test_runtime_flags_.set_wait_until_done(false);
   OnLayoutTestRuntimeFlagsChanged();
 }
 
 }  // namespace test_runner