Index: ash/system/network/tray_network_unittest.cc |
diff --git a/ash/system/network/tray_network_unittest.cc b/ash/system/network/tray_network_unittest.cc |
index f4a2259655e7de02bce99bd2cc225b0651ce43fb..32f23bd0881e64add22892318a71ffee07d7e8c8 100644 |
--- a/ash/system/network/tray_network_unittest.cc |
+++ b/ash/system/network/tray_network_unittest.cc |
@@ -7,6 +7,7 @@ |
#include "ash/login_status.h" |
#include "ash/system/network/network_list.h" |
#include "ash/system/tray/system_tray.h" |
+#include "ash/system/tray/system_tray_test_api.h" |
#include "ash/test/ash_test_base.h" |
#include "base/macros.h" |
#include "chromeos/dbus/dbus_thread_manager.h" |
@@ -59,7 +60,7 @@ TEST_F(TrayNetworkTest, Basics) { |
RunAllPendingInMessageLoop(); |
// Show network details. |
- TrayNetwork* tray_network = system_tray->GetTrayNetworkForTesting(); |
+ TrayNetwork* tray_network = SystemTrayTestApi(system_tray).tray_network(); |
const int close_delay_in_seconds = 0; |
bool activate = true; |
system_tray->ShowDetailedView(tray_network, close_delay_in_seconds, activate, |
@@ -74,7 +75,7 @@ TEST_F(TrayNetworkTest, Basics) { |
// Verifies that toggling Wi-Fi (usually via keyboard) shows a notification. |
TEST_F(TrayNetworkTest, ToggleWifi) { |
TrayNetwork* tray_network = |
- GetPrimarySystemTray()->GetTrayNetworkForTesting(); |
+ SystemTrayTestApi(GetPrimarySystemTray()).tray_network(); |
// No notifications at startup. |
ASSERT_EQ(0u, MessageCenter::Get()->NotificationCount()); |