| Index: ash/common/system/chromeos/palette/tools/screenshot_unittest.cc
|
| diff --git a/ash/common/system/chromeos/palette/tools/screenshot_unittest.cc b/ash/common/system/chromeos/palette/tools/screenshot_unittest.cc
|
| index faf7eaea089d7c80f864788d06b6754fe3554e0c..f66426d17370df0e00c34b46c68c375c3ac03bd5 100644
|
| --- a/ash/common/system/chromeos/palette/tools/screenshot_unittest.cc
|
| +++ b/ash/common/system/chromeos/palette/tools/screenshot_unittest.cc
|
| @@ -8,8 +8,9 @@
|
| #include "ash/common/system/chromeos/palette/tools/capture_region_mode.h"
|
| #include "ash/common/system/chromeos/palette/tools/capture_screen_action.h"
|
| #include "ash/common/test/test_palette_delegate.h"
|
| -#include "ash/common/wm_shell.h"
|
| +#include "ash/shell.h"
|
| #include "ash/test/ash_test_base.h"
|
| +#include "ash/test/shell_test_api.h"
|
| #include "base/macros.h"
|
| #include "base/memory/ptr_util.h"
|
|
|
| @@ -26,7 +27,7 @@ class ScreenshotToolTest : public test::AshTestBase {
|
| void SetUp() override {
|
| test::AshTestBase::SetUp();
|
|
|
| - WmShell::Get()->SetPaletteDelegateForTesting(
|
| + test::ShellTestApi().SetPaletteDelegate(
|
| base::MakeUnique<TestPaletteDelegate>());
|
|
|
| palette_tool_delegate_ = base::MakeUnique<MockPaletteToolDelegate>();
|
| @@ -34,7 +35,7 @@ class ScreenshotToolTest : public test::AshTestBase {
|
|
|
| TestPaletteDelegate* test_palette_delegate() {
|
| return static_cast<TestPaletteDelegate*>(
|
| - WmShell::Get()->palette_delegate());
|
| + Shell::GetInstance()->palette_delegate());
|
| }
|
|
|
| protected:
|
|
|