Refactor LauncherDelegate

ash/test/test_launcher_delegate.cc

 // Copyright (c) 2012 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/test/test_launcher_delegate.h"
 
+#include "ash/launcher/launcher_item_delegate_manager.h"
 #include "ash/launcher/launcher_model.h"
 #include "ash/launcher/launcher_util.h"
+#include "ash/shell.h"
 #include "ash/wm/window_util.h"
 #include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "grit/ash_resources.h"
 #include "ui/aura/window.h"
 
 namespace ash {
 namespace test {
 
 TestLauncherDelegate* TestLauncherDelegate::instance_ = NULL;
 
 TestLauncherDelegate::TestLauncherDelegate(LauncherModel* model)
     : model_(model) {
   CHECK(!instance_);
   instance_ = this;
+
+  if (Shell::HasInstance()) {
+    ash::LauncherItemDelegateManager* manager =
+        ash::Shell::GetInstance()->launcher_item_delegate_manager();
+    manager->RegisterLauncherItemDelegate(ash::TYPE_APP_PANEL, this);
+    manager->RegisterLauncherItemDelegate(ash::TYPE_TABBED, this);
+    manager->RegisterLauncherItemDelegate(ash::TYPE_APP_SHORTCUT, this);
+    manager->RegisterLauncherItemDelegate(ash::TYPE_BROWSER_SHORTCUT, this);
+    manager->RegisterLauncherItemDelegate(ash::TYPE_PLATFORM_APP, this);
+    manager->RegisterLauncherItemDelegate(ash::TYPE_WINDOWED_APP, this);
+  }
 }
 
 TestLauncherDelegate::~TestLauncherDelegate() {
+  if (Shell::HasInstance()) {
+    ash::LauncherItemDelegateManager* manager =
+        ash::Shell::GetInstance()->launcher_item_delegate_manager();
+    manager->UnregisterLauncherItemDelegate(ash::TYPE_APP_PANEL);
+    manager->UnregisterLauncherItemDelegate(ash::TYPE_TABBED);
+    manager->UnregisterLauncherItemDelegate(ash::TYPE_APP_SHORTCUT);
+    manager->UnregisterLauncherItemDelegate(ash::TYPE_BROWSER_SHORTCUT);
+    manager->UnregisterLauncherItemDelegate(ash::TYPE_PLATFORM_APP);
+    manager->UnregisterLauncherItemDelegate(ash::TYPE_WINDOWED_APP);
+  }
+
   instance_ = NULL;
 }
 
 void TestLauncherDelegate::AddLauncherItem(aura::Window* window) {
   AddLauncherItem(window, STATUS_CLOSED);
 }
 
 void TestLauncherDelegate::AddLauncherItem(
     aura::Window* window,
     LauncherItemStatus status) {
@@ skipping 104 matching lines @@
 
 bool TestLauncherDelegate::IsAppPinned(const std::string& app_id) {
   return false;
 }
 
 void TestLauncherDelegate::UnpinAppWithID(const std::string& app_id) {
 }
 
 }  // namespace test
 }  // namespace ash