| Index: ash/root_window_controller_unittest.cc
|
| diff --git a/ash/root_window_controller_unittest.cc b/ash/root_window_controller_unittest.cc
|
| index 074b48e9bc6ca8bf86e7036c34dcb165427e5739..401d3badffcfab3de612c3a17042ee7e0e970ded 100644
|
| --- a/ash/root_window_controller_unittest.cc
|
| +++ b/ash/root_window_controller_unittest.cc
|
| @@ -11,7 +11,6 @@
|
| #include "ash/common/system/tray/system_tray_delegate.h"
|
| #include "ash/common/wm/system_modal_container_layout_manager.h"
|
| #include "ash/common/wm/window_state.h"
|
| -#include "ash/common/wm_lookup.h"
|
| #include "ash/common/wm_shell.h"
|
| #include "ash/common/wm_window.h"
|
| #include "ash/public/cpp/shell_window_ids.h"
|
| @@ -388,7 +387,7 @@ TEST_F(RootWindowControllerTest, ModalContainer) {
|
| CreateModalWidget(gfx::Rect(300, 10, 100, 100));
|
| EXPECT_EQ(GetLayoutManager(controller, kShellWindowId_SystemModalContainer),
|
| controller->GetSystemModalLayoutManager(
|
| - WmLookup::Get()->GetWindowForWidget(session_modal_widget)));
|
| + WmWindow::Get(session_modal_widget->GetNativeWindow())));
|
|
|
| wm_shell->GetSessionStateDelegate()->LockScreen();
|
| EXPECT_EQ(LoginStatus::LOCKED,
|
| @@ -404,10 +403,10 @@ TEST_F(RootWindowControllerTest, ModalContainer) {
|
| EXPECT_EQ(
|
| GetLayoutManager(controller, kShellWindowId_LockSystemModalContainer),
|
| controller->GetSystemModalLayoutManager(
|
| - WmLookup::Get()->GetWindowForWidget(lock_modal_widget)));
|
| + WmWindow::Get(lock_modal_widget->GetNativeWindow())));
|
| EXPECT_EQ(GetLayoutManager(controller, kShellWindowId_SystemModalContainer),
|
| controller->GetSystemModalLayoutManager(
|
| - WmLookup::Get()->GetWindowForWidget(session_modal_widget)));
|
| + WmWindow::Get(session_modal_widget->GetNativeWindow())));
|
|
|
| wm_shell->GetSessionStateDelegate()->UnlockScreen();
|
| }
|
| @@ -437,7 +436,7 @@ TEST_F(RootWindowControllerTest, ModalContainerNotLoggedInLoggedIn) {
|
| EXPECT_EQ(
|
| GetLayoutManager(controller, kShellWindowId_LockSystemModalContainer),
|
| controller->GetSystemModalLayoutManager(
|
| - WmLookup::Get()->GetWindowForWidget(login_modal_widget)));
|
| + WmWindow::Get(login_modal_widget->GetNativeWindow())));
|
| login_modal_widget->Close();
|
|
|
| // Configure user session environment.
|
| @@ -454,7 +453,7 @@ TEST_F(RootWindowControllerTest, ModalContainerNotLoggedInLoggedIn) {
|
| CreateModalWidget(gfx::Rect(300, 10, 100, 100));
|
| EXPECT_EQ(GetLayoutManager(controller, kShellWindowId_SystemModalContainer),
|
| controller->GetSystemModalLayoutManager(
|
| - WmLookup::Get()->GetWindowForWidget(session_modal_widget)));
|
| + WmWindow::Get(session_modal_widget->GetNativeWindow())));
|
| }
|
|
|
| TEST_F(RootWindowControllerTest, ModalContainerBlockedSession) {
|
| @@ -469,7 +468,7 @@ TEST_F(RootWindowControllerTest, ModalContainerBlockedSession) {
|
| CreateModalWidget(gfx::Rect(300, 10, 100, 100));
|
| EXPECT_EQ(GetLayoutManager(controller, kShellWindowId_SystemModalContainer),
|
| controller->GetSystemModalLayoutManager(
|
| - WmLookup::Get()->GetWindowForWidget(session_modal_widget)));
|
| + WmWindow::Get(session_modal_widget->GetNativeWindow())));
|
| EXPECT_EQ(GetLayoutManager(controller, kShellWindowId_SystemModalContainer),
|
| controller->GetSystemModalLayoutManager(NULL));
|
| session_modal_widget->Close();
|
| @@ -485,12 +484,12 @@ TEST_F(RootWindowControllerTest, ModalContainerBlockedSession) {
|
| EXPECT_EQ(
|
| GetLayoutManager(controller, kShellWindowId_LockSystemModalContainer),
|
| controller->GetSystemModalLayoutManager(
|
| - WmLookup::Get()->GetWindowForWidget(lock_modal_widget)));
|
| + WmWindow::Get(lock_modal_widget->GetNativeWindow())));
|
|
|
| session_modal_widget = CreateModalWidget(gfx::Rect(300, 10, 100, 100));
|
| EXPECT_EQ(GetLayoutManager(controller, kShellWindowId_SystemModalContainer),
|
| controller->GetSystemModalLayoutManager(
|
| - WmLookup::Get()->GetWindowForWidget(session_modal_widget)));
|
| + WmWindow::Get(session_modal_widget->GetNativeWindow())));
|
| session_modal_widget->Close();
|
|
|
| lock_modal_widget->Close();
|
|
|