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/common/system/status_area_widget.h" |
7 #include "ash/common/system/tray/system_tray_delegate.h" | 8 #include "ash/common/system/tray/system_tray_delegate.h" |
8 #include "ash/common/wm_shell.h" | 9 #include "ash/common/wm_shell.h" |
9 #include "ash/root_window_controller.h" | 10 #include "ash/root_window_controller.h" |
10 #include "ash/shelf/shelf_widget.h" | 11 #include "ash/shelf/shelf_widget.h" |
11 #include "ash/shell.h" | 12 #include "ash/shell.h" |
12 #include "ash/system/status_area_widget.h" | |
13 | 13 |
14 namespace ash { | 14 namespace ash { |
15 | 15 |
16 LoginStatus StatusAreaWidgetTestHelper::GetUserLoginStatus() { | 16 LoginStatus StatusAreaWidgetTestHelper::GetUserLoginStatus() { |
17 return WmShell::Get()->system_tray_delegate()->GetUserLoginStatus(); | 17 return WmShell::Get()->system_tray_delegate()->GetUserLoginStatus(); |
18 } | 18 } |
19 | 19 |
20 StatusAreaWidget* StatusAreaWidgetTestHelper::GetStatusAreaWidget() { | 20 StatusAreaWidget* StatusAreaWidgetTestHelper::GetStatusAreaWidget() { |
21 return Shell::GetPrimaryRootWindowController() | 21 return Shell::GetPrimaryRootWindowController() |
22 ->shelf_widget() | 22 ->shelf_widget() |
23 ->status_area_widget(); | 23 ->status_area_widget(); |
24 } | 24 } |
25 | 25 |
26 StatusAreaWidget* StatusAreaWidgetTestHelper::GetSecondaryStatusAreaWidget() { | 26 StatusAreaWidget* StatusAreaWidgetTestHelper::GetSecondaryStatusAreaWidget() { |
27 RootWindowController* primary_controller = | 27 RootWindowController* primary_controller = |
28 Shell::GetPrimaryRootWindowController(); | 28 Shell::GetPrimaryRootWindowController(); |
29 Shell::RootWindowControllerList controllers = | 29 Shell::RootWindowControllerList controllers = |
30 Shell::GetAllRootWindowControllers(); | 30 Shell::GetAllRootWindowControllers(); |
31 for (size_t i = 0; i < controllers.size(); ++i) { | 31 for (size_t i = 0; i < controllers.size(); ++i) { |
32 if (controllers[i] != primary_controller) | 32 if (controllers[i] != primary_controller) |
33 return controllers[i]->shelf_widget()->status_area_widget(); | 33 return controllers[i]->shelf_widget()->status_area_widget(); |
34 } | 34 } |
35 | 35 |
36 return NULL; | 36 return NULL; |
37 } | 37 } |
38 | 38 |
39 } // namespace ash | 39 } // namespace ash |
OLD | NEW |