ash: Use system tray mojo interface to show system update tray icon

chrome/browser/ui/ash/system_tray_client_browsertest.cc

Index: chrome/browser/ui/ash/system_tray_client_browsertest.cc
diff --git a/chrome/browser/ui/ash/system_tray_client_browsertest.cc b/chrome/browser/ui/ash/system_tray_client_browsertest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..2f37af4542983efe541e284ea0b5be6e887473d7
--- /dev/null
+++ b/chrome/browser/ui/ash/system_tray_client_browsertest.cc
@@ -0,0 +1,52 @@
+// 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/common/system/tray/system_tray.h"
+#include "ash/common/system/update/tray_update.h"
+#include "ash/common/wm_root_window_controller.h"
+#include "ash/common/wm_shell.h"
+#include "chrome/browser/upgrade_detector.h"
+#include "chrome/test/base/in_process_browser_test.h"
+#include "content/public/test/test_utils.h"
+
+ash::TrayUpdate* GetTrayUpdate() {
+  return ash::WmShell::Get()
+      ->GetPrimaryRootWindowController()
+      ->GetSystemTray()
+      ->tray_update();
+}
+
+using SystemTrayClientTest = InProcessBrowserTest;
+
+// Test that a chrome update shows the update icon in the system menu.
+IN_PROC_BROWSER_TEST_F(SystemTrayClientTest, UpdateTrayIcon) {
+  ash::TrayUpdate* tray_update = GetTrayUpdate();
+
+  // When no update is pending, the icon isn't visible.
+  EXPECT_FALSE(tray_update->tray_view()->visible());
+
+  // Simulate an upgrade. This sends a mojo message to ash.
+  UpgradeDetector::GetInstance()->NotifyUpgradeRecommended();
+  content::RunAllPendingInMessageLoop();
+
+  // Tray icon is now visible.
+  EXPECT_TRUE(tray_update->tray_view()->visible());
+}
+
+// Test that a flash update causes the update UI to show in the system menu.
+IN_PROC_BROWSER_TEST_F(SystemTrayClientTest, FlashUpdateTrayIcon) {
+  ash::TrayUpdate* tray_update = GetTrayUpdate();
+
+  // When no update is pending, the icon isn't visible.
+  EXPECT_FALSE(tray_update->tray_view()->visible());
+
+  // Simulate a Flash update. This sends a mojo message to ash.
+  SystemTrayClient::Get()->SetFlashUpdateAvailable();
+  content::RunAllPendingInMessageLoop();
+
+  // Tray icon is now visible.
+  EXPECT_TRUE(tray_update->tray_view()->visible());
+}