chromeos: Introduce ash::SystemTrayTestApi

ash/system/network/tray_network_unittest.cc

 // Copyright 2017 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/system/network/tray_network.h"
 
 #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"
 #include "chromeos/network/network_handler.h"
 #include "components/prefs/testing_pref_service.h"
 #include "ui/message_center/message_center.h"
 
 using message_center::MessageCenter;
 
 namespace ash {
@@ skipping 32 matching lines @@
 };
 
 // Verifies that the network views can be created.
 TEST_F(TrayNetworkTest, Basics) {
   // Open the system tray menu.
   SystemTray* system_tray = GetPrimarySystemTray();
   system_tray->ShowDefaultView(BUBBLE_CREATE_NEW);
   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,
                                 BUBBLE_USE_EXISTING);
   RunAllPendingInMessageLoop();
 
   // Network details view was created.
   ASSERT_TRUE(tray_network->detailed());
   EXPECT_TRUE(tray_network->detailed()->visible());
 }
 
 // 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());
 
   // Simulate a user action to toggle Wi-Fi.
   tray_network->RequestToggleWifi();
 
   // Notification was shown.
   EXPECT_EQ(1u, MessageCenter::Get()->NotificationCount());
   EXPECT_TRUE(MessageCenter::Get()->HasPopupNotifications());
   EXPECT_TRUE(MessageCenter::Get()->FindVisibleNotificationById("wifi-toggle"));
 }
 
 }  // namespace
 }  // namespace ash