Promotes a handful of members from WmShell to Shell

ash/common/system/chromeos/palette/tools/screenshot_unittest.cc

 // Copyright 2016 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/common/system/chromeos/palette/mock_palette_tool_delegate.h"
 #include "ash/common/system/chromeos/palette/palette_ids.h"
 #include "ash/common/system/chromeos/palette/palette_tool.h"
 #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"
 
 namespace ash {
 
 namespace {
 
 // Base class for all create note ash tests.
 class ScreenshotToolTest : public test::AshTestBase {
  public:
   ScreenshotToolTest() {}
   ~ScreenshotToolTest() override {}
 
   void SetUp() override {
     test::AshTestBase::SetUp();
 
-    WmShell::Get()->SetPaletteDelegateForTesting(
+    test::ShellTestApi().SetPaletteDelegate(
         base::MakeUnique<TestPaletteDelegate>());
 
     palette_tool_delegate_ = base::MakeUnique<MockPaletteToolDelegate>();
   }
 
   TestPaletteDelegate* test_palette_delegate() {
     return static_cast<TestPaletteDelegate*>(
-        WmShell::Get()->palette_delegate());
+        Shell::GetInstance()->palette_delegate());
   }
 
  protected:
   std::unique_ptr<MockPaletteToolDelegate> palette_tool_delegate_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(ScreenshotToolTest);
 };
 
 }  // namespace
@@ skipping 24 matching lines @@
   std::unique_ptr<PaletteTool> tool =
       base::MakeUnique<CaptureScreenAction>(palette_tool_delegate_.get());
   EXPECT_CALL(*palette_tool_delegate_.get(),
               DisableTool(PaletteToolId::CAPTURE_SCREEN));
   EXPECT_CALL(*palette_tool_delegate_.get(), HidePaletteImmediately());
   tool->OnEnable();
   EXPECT_EQ(1, test_palette_delegate()->take_screenshot_count());
 }
 
 }  // namespace ash