| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "ash/test/status_area_widget_test_helper.h" | 5 #include "ash/test/status_area_widget_test_helper.h" |
| 6 | 6 |
| 7 #include "ash/root_window_controller.h" | 7 #include "ash/root_window_controller.h" |
| 8 #include "ash/shell.h" | 8 #include "ash/shell.h" |
| 9 #include "ash/system/status_area_widget.h" | 9 #include "ash/system/status_area_widget.h" |
| 10 #include "ash/system/tray/system_tray_delegate.h" | 10 #include "ash/system/tray/system_tray_delegate.h" |
| 11 #include "ash/wm_shell.h" | |
| 12 | 11 |
| 13 namespace ash { | 12 namespace ash { |
| 14 | 13 |
| 15 LoginStatus StatusAreaWidgetTestHelper::GetUserLoginStatus() { | 14 LoginStatus StatusAreaWidgetTestHelper::GetUserLoginStatus() { |
| 16 return Shell::Get()->system_tray_delegate()->GetUserLoginStatus(); | 15 return Shell::Get()->system_tray_delegate()->GetUserLoginStatus(); |
| 17 } | 16 } |
| 18 | 17 |
| 19 StatusAreaWidget* StatusAreaWidgetTestHelper::GetStatusAreaWidget() { | 18 StatusAreaWidget* StatusAreaWidgetTestHelper::GetStatusAreaWidget() { |
| 20 return Shell::GetPrimaryRootWindowController()->GetStatusAreaWidget(); | 19 return Shell::GetPrimaryRootWindowController()->GetStatusAreaWidget(); |
| 21 } | 20 } |
| 22 | 21 |
| 23 StatusAreaWidget* StatusAreaWidgetTestHelper::GetSecondaryStatusAreaWidget() { | 22 StatusAreaWidget* StatusAreaWidgetTestHelper::GetSecondaryStatusAreaWidget() { |
| 24 RootWindowController* primary_controller = | 23 RootWindowController* primary_controller = |
| 25 Shell::GetPrimaryRootWindowController(); | 24 Shell::GetPrimaryRootWindowController(); |
| 26 Shell::RootWindowControllerList controllers = | 25 Shell::RootWindowControllerList controllers = |
| 27 Shell::GetAllRootWindowControllers(); | 26 Shell::GetAllRootWindowControllers(); |
| 28 for (size_t i = 0; i < controllers.size(); ++i) { | 27 for (size_t i = 0; i < controllers.size(); ++i) { |
| 29 if (controllers[i] != primary_controller) | 28 if (controllers[i] != primary_controller) |
| 30 return controllers[i]->GetStatusAreaWidget(); | 29 return controllers[i]->GetStatusAreaWidget(); |
| 31 } | 30 } |
| 32 | 31 |
| 33 return nullptr; | 32 return nullptr; |
| 34 } | 33 } |
| 35 | 34 |
| 36 } // namespace ash | 35 } // namespace ash |
| OLD | NEW |