Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "ash/common/system/chromeos/supervised/tray_supervised_user.h" | 5 #include "ash/common/system/chromeos/supervised/tray_supervised_user.h" |
| 6 | 6 |
| 7 #include "ash/common/login_status.h" | 7 #include "ash/common/login_status.h" |
| 8 #include "ash/common/test/ash_test.h" | |
| 8 #include "ash/common/test/test_system_tray_delegate.h" | 9 #include "ash/common/test/test_system_tray_delegate.h" |
| 9 #include "ash/test/ash_test_base.h" | |
| 10 #include "ui/message_center/message_center.h" | 10 #include "ui/message_center/message_center.h" |
| 11 #include "ui/message_center/notification.h" | 11 #include "ui/message_center/notification.h" |
| 12 #include "ui/message_center/notification_list.h" | 12 #include "ui/message_center/notification_list.h" |
| 13 #include "ui/message_center/notification_types.h" | 13 #include "ui/message_center/notification_types.h" |
| 14 | 14 |
| 15 using message_center::NotificationList; | 15 using message_center::NotificationList; |
| 16 | 16 |
| 17 namespace ash { | 17 namespace ash { |
| 18 | 18 |
| 19 class TraySupervisedUserTest : public test::AshTestBase { | 19 class TraySupervisedUserTest : public AshTest { |
| 20 public: | 20 public: |
| 21 TraySupervisedUserTest() {} | 21 TraySupervisedUserTest() {} |
| 22 ~TraySupervisedUserTest() override {} | 22 ~TraySupervisedUserTest() override {} |
| 23 | 23 |
| 24 protected: | 24 protected: |
| 25 message_center::Notification* GetPopup(); | 25 message_center::Notification* GetPopup(); |
| 26 | 26 |
| 27 private: | 27 private: |
| 28 DISALLOW_COPY_AND_ASSIGN(TraySupervisedUserTest); | 28 DISALLOW_COPY_AND_ASSIGN(TraySupervisedUserTest); |
| 29 }; | 29 }; |
| (...skipping 17 matching lines...) Expand all Loading... | |
| 47 | 47 |
| 48 void SetUp() override; | 48 void SetUp() override; |
| 49 void TearDown() override; | 49 void TearDown() override; |
| 50 | 50 |
| 51 private: | 51 private: |
| 52 DISALLOW_COPY_AND_ASSIGN(TraySupervisedUserInitialTest); | 52 DISALLOW_COPY_AND_ASSIGN(TraySupervisedUserInitialTest); |
| 53 }; | 53 }; |
| 54 | 54 |
| 55 void TraySupervisedUserInitialTest::SetUp() { | 55 void TraySupervisedUserInitialTest::SetUp() { |
| 56 test::TestSystemTrayDelegate::SetInitialLoginStatus(LoginStatus::SUPERVISED); | 56 test::TestSystemTrayDelegate::SetInitialLoginStatus(LoginStatus::SUPERVISED); |
| 57 test::AshTestBase::SetUp(); | 57 AshTest::SetUp(); |
| 58 } | 58 } |
| 59 | 59 |
| 60 void TraySupervisedUserInitialTest::TearDown() { | 60 void TraySupervisedUserInitialTest::TearDown() { |
| 61 test::AshTestBase::TearDown(); | 61 AshTest::TearDown(); |
| 62 // SetInitialLoginStatus() is reset in AshTestHelper::TearDown(). | 62 // Reset the initial login status to the default. |
| 63 test::TestSystemTrayDelegate::SetInitialLoginStatus(LoginStatus::USER); | |
|
James Cook
2016/10/13 23:03:21
See CL comment for why I did this. It's ugly, but
| |
| 63 } | 64 } |
| 64 | 65 |
| 65 TEST_F(TraySupervisedUserTest, SupervisedUserHasNotification) { | 66 TEST_F(TraySupervisedUserTest, SupervisedUserHasNotification) { |
| 66 test::TestSystemTrayDelegate* delegate = GetSystemTrayDelegate(); | 67 test::TestSystemTrayDelegate* delegate = GetSystemTrayDelegate(); |
| 67 delegate->SetLoginStatus(LoginStatus::SUPERVISED); | 68 delegate->SetLoginStatus(LoginStatus::SUPERVISED); |
| 68 | 69 |
| 69 message_center::Notification* notification = GetPopup(); | 70 message_center::Notification* notification = GetPopup(); |
| 70 ASSERT_NE(static_cast<message_center::Notification*>(NULL), notification); | 71 ASSERT_NE(static_cast<message_center::Notification*>(NULL), notification); |
| 71 EXPECT_EQ(static_cast<int>(message_center::SYSTEM_PRIORITY), | 72 EXPECT_EQ(static_cast<int>(message_center::SYSTEM_PRIORITY), |
| 72 notification->rich_notification_data().priority); | 73 notification->rich_notification_data().priority); |
| 73 } | 74 } |
| 74 | 75 |
| 75 TEST_F(TraySupervisedUserInitialTest, SupervisedUserNoCrash) { | 76 TEST_F(TraySupervisedUserInitialTest, SupervisedUserNoCrash) { |
| 76 // Initial login status is already SUPERVISED, which should create | 77 // Initial login status is already SUPERVISED, which should create |
| 77 // the notification and should not cause crashes. | 78 // the notification and should not cause crashes. |
| 78 message_center::Notification* notification = GetPopup(); | 79 message_center::Notification* notification = GetPopup(); |
| 79 ASSERT_NE(static_cast<message_center::Notification*>(NULL), notification); | 80 ASSERT_NE(static_cast<message_center::Notification*>(NULL), notification); |
| 80 EXPECT_EQ(static_cast<int>(message_center::SYSTEM_PRIORITY), | 81 EXPECT_EQ(static_cast<int>(message_center::SYSTEM_PRIORITY), |
| 81 notification->rich_notification_data().priority); | 82 notification->rich_notification_data().priority); |
| 82 } | 83 } |
| 83 | 84 |
| 84 } // namespace ash | 85 } // namespace ash |
| OLD | NEW |