src/ash/test/test_system_tray_delegate.h

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef ASH_TEST_TEST_SYSTEM_TRAY_DELEGATE_H_
   6 #define ASH_TEST_TEST_SYSTEM_TRAY_DELEGATE_H_
   7
   8 #include "ash/system/tray/default_system_tray_delegate.h"
   9 #include "base/time/time.h"
  10
  11 namespace ash {
  12 namespace test {
  13
  14 class TestSystemTrayDelegate : public DefaultSystemTrayDelegate {
  15  public:
  16   TestSystemTrayDelegate();
  17   ~TestSystemTrayDelegate() override;
  18
  19   // Changes the login status when initially the delegate is created. This will
  20   // be called before AshTestBase::SetUp() to test the case when chrome is
  21   // restarted right after the login (such like a flag is set).
  22   // This value will be reset in AshTestHelper::TearDown,  most test fixtures
  23   // don't need to care its lifecycle.
  24   static void SetInitialLoginStatus(user::LoginStatus login_status);
  25
  26   // Changes the current login status in the test. This also invokes
  27   // UpdateAfterLoginStatusChange(). Usually this is called in the test code to
  28   // set up a login status. This will fit to most of the test cases, but this
  29   // cannot be set during the initialization. To test the initialization,
  30   // consider using SetInitialLoginStatus() instead.
  31   void SetLoginStatus(user::LoginStatus login_status);
  32
  33   void set_should_show_display_notification(bool should_show) {
  34     should_show_display_notification_ = should_show;
  35   }
  36
  37   // Updates the session length limit so that the limit will come from now in
  38   // |new_limit|.
  39   void SetSessionLengthLimitForTest(const base::TimeDelta& new_limit);
  40
  41   // Clears the session length limit.
  42   void ClearSessionLengthLimit();
  43
  44   // Overridden from SystemTrayDelegate:
  45   user::LoginStatus GetUserLoginStatus() const override;
  46   bool IsUserSupervised() const override;
  47   bool ShouldShowDisplayNotification() override;
  48   bool GetSessionStartTime(base::TimeTicks* session_start_time) override;
  49   bool GetSessionLengthLimit(base::TimeDelta* session_length_limit) override;
  50   void ShutDown() override;
  51   void SignOut() override;
  52
  53  private:
  54   bool should_show_display_notification_;
  55   user::LoginStatus login_status_;
  56   base::TimeDelta session_length_limit_;
  57   bool session_length_limit_set_;
  58
  59   DISALLOW_COPY_AND_ASSIGN(TestSystemTrayDelegate);
  60 };
  61
  62 }  // namespace test
  63 }  // namespace ash
  64
  65 #endif  // ASH_TEST_TEST_SYSTEM_TRAY_DELEGATE_H_