chromeos: Introduce ash::SystemTrayTestApi

chrome/browser/ui/ash/system_tray_client_browsertest.cc

 // Copyright 2016 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 "chrome/browser/ui/ash/system_tray_client.h"
 
 #include "ash/root_window_controller.h"
 #include "ash/shell.h"
 #include "ash/system/enterprise/tray_enterprise.h"
 #include "ash/system/tray/label_tray_view.h"
 #include "ash/system/tray/system_tray.h"
+#include "ash/system/tray/system_tray_test_api.h"
 #include "ash/system/update/tray_update.h"
 #include "chrome/browser/chromeos/policy/device_policy_cros_browser_test.h"
 #include "chrome/browser/upgrade_detector.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "content/public/test/test_utils.h"
 
 namespace {
 
 // TODO(jamescook): Add a test-only mojo API to get system tray details.
 ash::SystemTray* GetSystemTray() {
@@ skipping 40 matching lines @@
   // Mark the device as enterprise managed.
   policy::DevicePolicyCrosTestHelper::MarkAsEnterpriseOwnedBy("example.com");
   content::RunAllPendingInMessageLoop();
 
   // Open the system tray menu.
   ash::SystemTray* system_tray = GetSystemTray();
   system_tray->ShowDefaultView(ash::BUBBLE_CREATE_NEW);
 
   // Managed devices show an item in the menu.
   ash::TrayEnterprise* tray_enterprise =
-      system_tray->GetTrayEnterpriseForTesting();
+      ash::SystemTrayTestApi(system_tray).tray_enterprise();
   ASSERT_TRUE(tray_enterprise->tray_view());
   EXPECT_TRUE(tray_enterprise->tray_view()->visible());
 
   system_tray->CloseSystemBubble();
 }