Updates the display message in the uber tray.

ash/extended_desktop_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/display/display_controller.h"
 #include "ash/display/display_manager.h"
+#include "ash/root_window_controller.h"
 #include "ash/screen_ash.h"
 #include "ash/shell.h"
 #include "ash/shell_window_ids.h"
 #include "ash/system/tray/system_tray.h"
 #include "ash/test/ash_test_base.h"
 #include "ash/wm/coordinate_conversion.h"
 #include "ash/wm/property_util.h"
 #include "ash/wm/window_cycle_controller.h"
 #include "ash/wm/window_properties.h"
 #include "ash/wm/window_util.h"
@@ skipping 783 matching lines @@
 }
 
 TEST_F(ExtendedDesktopTest, PassiveGrab) {
   if (!SupportsMultipleDisplays())
     return;
 
   EventLocationRecordingEventHandler event_handler;
   ash::Shell::GetInstance()->AddPreTargetHandler(&event_handler);
 
   UpdateDisplay("300x300,200x200");
+  // close all notification for display change. Otherwise some event may
+  // happen on the notification bubble, which breaks the test expectation.
+  // TODO(mukai): remove this when the display change notification has moved
+  // to the message_center.
+  Shell::RootWindowControllerList root_controllers =
+      Shell::GetAllRootWindowControllers();
+  for (size_t i = 0; i < root_controllers.size(); ++i)
+    root_controllers[i]->GetSystemTray()->CloseNotificationBubbleForTest();
 
   views::Widget* widget = CreateTestWidget(gfx::Rect(50, 50, 200, 200));
   widget->Show();
   ASSERT_EQ("50,50 200x200", widget->GetWindowBoundsInScreen().ToString());
 
   aura::test::EventGenerator& generator(GetEventGenerator());
   generator.MoveMouseTo(150, 150);
   EXPECT_EQ("100,100 150,150", event_handler.GetLocationsAndReset());
 
   generator.PressLeftButton();
   generator.MoveMouseTo(400, 150);
 
   EXPECT_EQ("350,100 400,150", event_handler.GetLocationsAndReset());
 
   generator.ReleaseLeftButton();
   EXPECT_EQ("-999,-999 -999,-999", event_handler.GetLocationsAndReset());
 
   generator.MoveMouseTo(400, 150);
   EXPECT_EQ("100,150 100,150", event_handler.GetLocationsAndReset());
 
   ash::Shell::GetInstance()->RemovePreTargetHandler(&event_handler);
 }
 
 }  // namespace internal
 }  // namespace ash