Index: ash/display/display_controller_unittest.cc |
diff --git a/ash/display/display_controller_unittest.cc b/ash/display/display_controller_unittest.cc |
index c07a9b7ab323009b4780bc1689acdc69cf1f6763..2531e53ea34f3ea0c5172f63ad64b6bf8a16bafd 100644 |
--- a/ash/display/display_controller_unittest.cc |
+++ b/ash/display/display_controller_unittest.cc |
@@ -7,6 +7,7 @@ |
#include "ash/display/display_manager.h" |
#include "ash/launcher/launcher.h" |
#include "ash/screen_ash.h" |
+#include "ash/shelf/shelf_widget.h" |
#include "ash/shell.h" |
#include "ash/test/ash_test_base.h" |
#include "ash/test/cursor_manager_test_api.h" |
@@ -250,7 +251,7 @@ TEST_F(DisplayControllerTest, SwapPrimary) { |
display_controller->GetRootWindowForDisplayId(secondary_display.id()); |
EXPECT_NE(primary_root, secondary_root); |
aura::Window* launcher_window = |
- Launcher::ForPrimaryDisplay()->widget()->GetNativeView(); |
+ Launcher::ForPrimaryDisplay()->shelf_widget()->GetNativeView(); |
EXPECT_TRUE(primary_root->Contains(launcher_window)); |
EXPECT_FALSE(secondary_root->Contains(launcher_window)); |
EXPECT_EQ(primary_display.id(), |
@@ -341,7 +342,7 @@ TEST_F(DisplayControllerTest, SwapPrimaryById) { |
aura::RootWindow* secondary_root = |
display_controller->GetRootWindowForDisplayId(secondary_display.id()); |
aura::Window* launcher_window = |
- Launcher::ForPrimaryDisplay()->widget()->GetNativeView(); |
+ Launcher::ForPrimaryDisplay()->shelf_widget()->GetNativeView(); |
EXPECT_TRUE(primary_root->Contains(launcher_window)); |
EXPECT_FALSE(secondary_root->Contains(launcher_window)); |
EXPECT_NE(primary_root, secondary_root); |