chromeos: Convert system tray session length limit to mojo

ash/system/session/tray_session_length_limit_unittest.cc

 // Copyright 2014 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/session/tray_session_length_limit.h"
 
+#include "ash/session/session_controller.h"
+#include "ash/shell.h"
+#include "ash/system/tray/label_tray_view.h"
 #include "ash/system/tray/system_tray.h"
 #include "ash/test/ash_test_base.h"
-#include "ash/test/test_system_tray_delegate.h"
 #include "base/memory/ptr_util.h"
 #include "base/time/time.h"
 #include "ui/message_center/message_center.h"
 #include "ui/message_center/notification.h"
 #include "ui/message_center/notification_types.h"
 
 namespace ash {
 namespace test {
 
 class TraySessionLengthLimitTest : public AshTestBase {
  public:
-  TraySessionLengthLimitTest() {}
-  ~TraySessionLengthLimitTest() override {}
+  TraySessionLengthLimitTest() = default;
+  ~TraySessionLengthLimitTest() override = default;
 
-  void SetUp() override {
-    AshTestBase::SetUp();
-    SystemTray* system_tray = GetPrimarySystemTray();
-    tray_session_length_limit_ = new TraySessionLengthLimit(system_tray);
-    system_tray->AddTrayItem(base::WrapUnique(tray_session_length_limit_));
+ protected:
+  LabelTrayView* GetTrayBubbleView() {

[msw, 2017/06/06 19:01:04]

nit: maybe add some session length limit context to this helper's name?

[James Cook, 2017/06/07 00:48:13]

Done.

+    return GetPrimarySystemTray()
+        ->GetTraySessionLengthLimitForTesting()
+        ->tray_bubble_view_;
   }
 
-  void TearDown() override {
-    ClearSessionLengthLimit();
-    AshTestBase::TearDown();
-  }
-
- protected:
   void UpdateSessionLengthLimitInMin(int mins) {
-    GetSystemTrayDelegate()->SetSessionLengthLimitForTest(
-        base::TimeDelta::FromMinutes(mins));
-    tray_session_length_limit_->OnSessionLengthLimitChanged();
+    Shell::Get()->session_controller()->SetSessionLengthLimit(
+        base::TimeDelta::FromMinutes(mins), base::TimeTicks::Now());
   }
 
   message_center::Notification* GetNotification() {
     const message_center::NotificationList::Notifications& notifications =
         message_center::MessageCenter::Get()->GetVisibleNotifications();
     for (message_center::NotificationList::Notifications::const_iterator iter =
              notifications.begin();
          iter != notifications.end(); ++iter) {
       if ((*iter)->id() == TraySessionLengthLimit::kNotificationId)
         return *iter;
     }
     return nullptr;
   }
 
   void ClearSessionLengthLimit() {
-    GetSystemTrayDelegate()->ClearSessionLengthLimit();
-    tray_session_length_limit_->OnSessionLengthLimitChanged();
+    Shell::Get()->session_controller()->SetSessionLengthLimit(
+        base::TimeDelta(), base::TimeTicks());
   }
 
   void RemoveNotification() {
     message_center::MessageCenter::Get()->RemoveNotification(
         TraySessionLengthLimit::kNotificationId, false /* by_user */);
   }
 
-  TraySessionLengthLimit* tray_session_length_limit() {
-    return tray_session_length_limit_;
-  }
-
  private:
-  // Weak reference, owned by the SystemTray.
-  TraySessionLengthLimit* tray_session_length_limit_;
-
   DISALLOW_COPY_AND_ASSIGN(TraySessionLengthLimitTest);
 };
 
+TEST_F(TraySessionLengthLimitTest, Visibility) {
+  SystemTray* system_tray = GetPrimarySystemTray();
+
+  // By default there is no session length limit item.
+  system_tray->ShowDefaultView(BUBBLE_CREATE_NEW);
+  EXPECT_FALSE(GetTrayBubbleView());
+  system_tray->CloseSystemBubble();
+
+  // Setting a length limit shows an item in the system tray menu.
+  UpdateSessionLengthLimitInMin(10);
+  system_tray->ShowDefaultView(BUBBLE_CREATE_NEW);
+  ASSERT_TRUE(GetTrayBubbleView());
+  EXPECT_TRUE(GetTrayBubbleView()->visible());
+  system_tray->CloseSystemBubble();
+
+  // Removing the session length limit removes the tray menu item.
+  UpdateSessionLengthLimitInMin(0);
+  system_tray->ShowDefaultView(BUBBLE_CREATE_NEW);
+  EXPECT_FALSE(GetTrayBubbleView());
+  system_tray->CloseSystemBubble();
+}
+
 TEST_F(TraySessionLengthLimitTest, Notification) {
   // No notifications when no session limit.
   EXPECT_FALSE(GetNotification());
 
   // Limit is 15 min.
   UpdateSessionLengthLimitInMin(15);
   message_center::Notification* notification = GetNotification();
   EXPECT_TRUE(notification);
   EXPECT_EQ(message_center::SYSTEM_PRIORITY, notification->priority());
   base::string16 first_content = notification->message();
@@ skipping 62 matching lines @@
   // re-appear and be re-read.
   UpdateSessionLengthLimitInMin(15);
   notification = GetNotification();
   EXPECT_TRUE(notification);
   EXPECT_TRUE(notification->rich_notification_data()
                   .should_make_spoken_feedback_for_popup_updates);
 }
 
 }  // namespace test
 }  // namespace ash