Display "Restart to update Adobe Flash Player" for Flash updates.

ash/system/update/tray_update.h

Index: ash/system/update/tray_update.h
diff --git a/ash/system/update/tray_update.h b/ash/system/update/tray_update.h
index 70c970e2f69ccae348780fa122aa08ab30820cd6..b4456209c8653ac06c777b433ce43c88580efab3 100644
--- a/ash/system/update/tray_update.h
+++ b/ash/system/update/tray_update.h
@@ -7,9 +7,12 @@
 
 #include "ash/ash_export.h"
 #include "ash/system/tray/tray_image_item.h"
+#include "base/gtest_prod_util.h"
 #include "base/macros.h"
+#include "base/strings/string16.h"
 
 namespace views {
+class Label;
 class View;
 }
 
@@ -17,6 +20,7 @@ namespace ash {
 
 namespace mojom {
 enum class UpdateSeverity;
+enum class UpdateType;
 }
 
 // The system update tray item. The tray icon stays visible once an update
@@ -31,21 +35,30 @@ class ASH_EXPORT TrayUpdate : public TrayImageItem {
   // available. Once shown the icon persists until reboot. |severity| and
   // |factory_reset_required| are used to set the icon, color, and tooltip.
   void ShowUpdateIcon(mojom::UpdateSeverity severity,
-                      bool factory_reset_required);
+                      bool factory_reset_required,
+                      mojom::UpdateType update_type);
 
  private:
+  FRIEND_TEST_ALL_PREFIXES(TrayUpdateTest, VisibilityAfterUpdate);
+  FRIEND_TEST_ALL_PREFIXES(TrayUpdateTest, VisibilityAfterFlashUpdate);
+
   class UpdateView;
 
   // Overridden from TrayImageItem.
   bool GetInitialVisibility() override;
   views::View* CreateDefaultView(LoginStatus status) override;
 
+  // Expose label information for testing.
+  base::string16 GetLabelForTesting();
+
   // If an external monitor is connected then the system tray may be created
   // after the update data is sent from chrome, so share the update info between
   // all instances.
   static bool update_required_;
   static mojom::UpdateSeverity severity_;
   static bool factory_reset_required_;
+  static mojom::UpdateType update_type_;
+  static views::Label* update_label_;
 
   DISALLOW_COPY_AND_ASSIGN(TrayUpdate);
 };