Index: ash/shell_unittest.cc |
diff --git a/ash/shell_unittest.cc b/ash/shell_unittest.cc |
index 826c1b0d7e599410b16151f6cea5d1aaa3c80fe6..5ff61ff2981fc8698b0f9f60510e805303821906 100644 |
--- a/ash/shell_unittest.cc |
+++ b/ash/shell_unittest.cc |
@@ -9,9 +9,9 @@ |
#include "ash/common/ash_switches.h" |
#include "ash/common/session/session_state_delegate.h" |
-#include "ash/common/shelf/shelf.h" |
#include "ash/common/shelf/shelf_layout_manager.h" |
#include "ash/common/shelf/shelf_widget.h" |
+#include "ash/common/shelf/wm_shelf.h" |
#include "ash/common/shell_window_ids.h" |
#include "ash/common/wm_shell.h" |
#include "ash/desktop_background/desktop_background_widget_controller.h" |
@@ -371,7 +371,7 @@ TEST_F(ShellTest, ManagedWindowModeBasics) { |
// We start with the usual window containers. |
ExpectAllContainers(); |
// Shelf is visible. |
- ShelfWidget* shelf_widget = Shelf::ForPrimaryDisplay()->shelf_widget(); |
+ ShelfWidget* shelf_widget = GetPrimaryShelf()->GetShelfWidgetForTesting(); |
EXPECT_TRUE(shelf_widget->IsVisible()); |
// Shelf is at bottom-left of screen. |
EXPECT_EQ(0, shelf_widget->GetWindowBoundsInScreen().x()); |
@@ -441,7 +441,7 @@ TEST_F(ShellTest, ToggleAutoHide) { |
window->Show(); |
wm::ActivateWindow(window.get()); |
- Shelf* shelf = Shelf::ForPrimaryDisplay(); |
+ WmShelf* shelf = GetPrimaryShelf(); |
shelf->SetAutoHideBehavior(SHELF_AUTO_HIDE_BEHAVIOR_ALWAYS); |
EXPECT_EQ(SHELF_AUTO_HIDE_BEHAVIOR_ALWAYS, shelf->auto_hide_behavior()); |