Fix "modal isn't modal in multi displays" issue

ash/wm/system_modal_container_layout_manager_unittest.cc

 // Copyright (c) 2012 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 "ash/wm/system_modal_container_layout_manager.h"
 
+#include <memory>
+
 #include "ash/common/session/session_state_delegate.h"
 #include "ash/common/shell_window_ids.h"
 #include "ash/common/wm_shell.h"
 #include "ash/root_window_controller.h"
 #include "ash/shell.h"
 #include "ash/test/ash_test_base.h"
 #include "ash/wm/window_util.h"
 #include "base/command_line.h"
 #include "base/compiler_specific.h"
 #include "base/run_loop.h"
 #include "ui/aura/client/aura_constants.h"
+#include "ui/aura/test/test_window_delegate.h"
 #include "ui/aura/window.h"
 #include "ui/aura/window_event_dispatcher.h"
 #include "ui/compositor/layer.h"
 #include "ui/compositor/scoped_animation_duration_scale_mode.h"
 #include "ui/compositor/test/layer_animator_test_controller.h"
 #include "ui/events/test/event_generator.h"
 #include "ui/keyboard/keyboard_controller.h"
 #include "ui/keyboard/keyboard_switches.h"
 #include "ui/keyboard/keyboard_ui.h"
 #include "ui/keyboard/keyboard_util.h"
@@ skipping 641 matching lines @@
   window->SetProperty(aura::client::kModalKey, ui::MODAL_TYPE_NONE);
   EXPECT_FALSE(WmShell::Get()->IsSystemModalWindowOpen());
 
   window->SetProperty(aura::client::kModalKey, ui::MODAL_TYPE_SYSTEM);
   EXPECT_TRUE(WmShell::Get()->IsSystemModalWindowOpen());
 
   widget->Close();
   EXPECT_FALSE(WmShell::Get()->IsSystemModalWindowOpen());
 }
 
+namespace {
+
+class InputTestDelegate : public aura::test::TestWindowDelegate {
+ public:
+  InputTestDelegate() {}
+  ~InputTestDelegate() override {}
+
+  // ui::EventHandler:
+  void OnMouseEvent(ui::MouseEvent* event) override { mouse_event_count_++; }
+
+  int mouse_event_count() const { return mouse_event_count_; }
+
+ private:
+  int mouse_event_count_ = 0;
+
+  DISALLOW_COPY_AND_ASSIGN(InputTestDelegate);
+};
+
+}  // namespace
+
+// Make sure that events are properly blocked in multi displays environment.
+TEST_F(SystemModalContainerLayoutManagerTest, BlockEvent) {
+  UpdateDisplay("500x500, 500x500");
+  InputTestDelegate delegate;
+  std::unique_ptr<aura::Window> window(CreateTestWindowInShellWithDelegate(
+      &delegate, 0, gfx::Rect(0, 0, 100, 100)));
+  window->Show();
+
+  ui::test::EventGenerator event_generator(Shell::GetPrimaryRootWindow(),
+                                           window.get());
+  event_generator.ClickLeftButton();
+  EXPECT_EQ(2, delegate.mouse_event_count());
+
+  views::Widget* widget = views::Widget::CreateWindowWithContextAndBounds(
+      new TestWindow(true), Shell::GetPrimaryRootWindow(),
+      gfx::Rect(200, 200, 100, 100));
+  widget->Show();
+  EXPECT_TRUE(WmShell::Get()->IsSystemModalWindowOpen());
+
+  // Events should be blocked.
+  event_generator.ClickLeftButton();
+  EXPECT_EQ(2, delegate.mouse_event_count());
+  widget->Close();
+
+  EXPECT_FALSE(WmShell::Get()->IsSystemModalWindowOpen());
+  event_generator.ClickLeftButton();
+  EXPECT_EQ(4, delegate.mouse_event_count());
+
+  // We need to delete window here because delegate is on the stack.
+  window.reset();
+}
+
 }  // namespace test
 }  // namespace ash