| // 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/session/session_state_delegate.h" |
| #include "ash/shell.h" |
| #include "ash/system/system_notifier.h" |
| #include "ash/test/ash_test_base.h" |
| #include "ash/test/test_session_state_delegate.h" |
| #include "ash/test/test_shell_delegate.h" |
| #include "chrome/browser/ui/ash/multi_user/multi_user_notification_blocker_chromeos.h" |
| #include "chrome/browser/ui/ash/multi_user/multi_user_window_manager_chromeos.h" |
| #include "chrome/test/base/testing_browser_process.h" |
| #include "chrome/test/base/testing_profile_manager.h" |
| #include "components/user_manager/user_info.h" |
| #include "ui/message_center/message_center.h" |
| #include "ui/message_center/notification.h" |
| |
| class MultiUserNotificationBlockerChromeOSTest |
| : public ash::test::AshTestBase, |
| public message_center::NotificationBlocker::Observer { |
| public: |
| MultiUserNotificationBlockerChromeOSTest() |
| : state_changed_count_(0), |
| testing_profile_manager_(TestingBrowserProcess::GetGlobal()), |
| window_id_(0) {} |
| ~MultiUserNotificationBlockerChromeOSTest() override {} |
| |
| // ash::test::AshTestBase overrides: |
| void SetUp() override { |
| ash::test::AshTestBase::SetUp(); |
| ASSERT_TRUE(testing_profile_manager_.SetUp()); |
| |
| // MultiUserWindowManager is initialized after the log in. |
| testing_profile_manager_.CreateTestingProfile(GetDefaultUserId()); |
| |
| ash::test::TestShellDelegate* shell_delegate = |
| static_cast<ash::test::TestShellDelegate*>( |
| ash::Shell::GetInstance()->delegate()); |
| shell_delegate->set_multi_profiles_enabled(true); |
| chrome::MultiUserWindowManager::CreateInstance(); |
| |
| ash::test::TestSessionStateDelegate* session_state_delegate = |
| static_cast<ash::test::TestSessionStateDelegate*>( |
| ash::Shell::GetInstance()->session_state_delegate()); |
| session_state_delegate->AddUser("test2@example.com"); |
| |
| // Disable any animations for the test. |
| GetMultiUserWindowManager()->SetAnimationSpeedForTest( |
| chrome::MultiUserWindowManagerChromeOS::ANIMATION_SPEED_DISABLED); |
| GetMultiUserWindowManager()->notification_blocker_->AddObserver(this); |
| } |
| |
| void TearDown() override { |
| GetMultiUserWindowManager()->notification_blocker_->RemoveObserver(this); |
| if (chrome::MultiUserWindowManager::GetInstance()) |
| chrome::MultiUserWindowManager::DeleteInstance(); |
| ash::test::AshTestBase::TearDown(); |
| } |
| |
| // message_center::NotificationBlocker::Observer ovverrides: |
| void OnBlockingStateChanged( |
| message_center::NotificationBlocker* blocker) override { |
| state_changed_count_++; |
| } |
| |
| protected: |
| chrome::MultiUserWindowManagerChromeOS* GetMultiUserWindowManager() { |
| return static_cast<chrome::MultiUserWindowManagerChromeOS*>( |
| chrome::MultiUserWindowManager::GetInstance()); |
| } |
| |
| const std::string GetDefaultUserId() { |
| return ash::Shell::GetInstance() |
| ->session_state_delegate() |
| ->GetUserInfo(0) |
| ->GetUserID(); |
| } |
| |
| const message_center::NotificationBlocker* blocker() { |
| return GetMultiUserWindowManager()->notification_blocker_.get(); |
| } |
| |
| void CreateProfile(const std::string& name) { |
| testing_profile_manager_.CreateTestingProfile(name); |
| } |
| |
| void SwitchActiveUser(const std::string& name) { |
| ash::Shell::GetInstance()->session_state_delegate()->SwitchActiveUser(name); |
| if (chrome::MultiUserWindowManager::GetMultiProfileMode() == |
| chrome::MultiUserWindowManager::MULTI_PROFILE_MODE_SEPARATED) { |
| static_cast<chrome::MultiUserWindowManagerChromeOS*>( |
| chrome::MultiUserWindowManager::GetInstance())->ActiveUserChanged( |
| name); |
| } |
| } |
| |
| int GetStateChangedCountAndReset() { |
| int result = state_changed_count_; |
| state_changed_count_ = 0; |
| return result; |
| } |
| |
| bool ShouldShowNotificationAsPopup( |
| const message_center::NotifierId& notifier_id, |
| const std::string profile_id) { |
| message_center::NotifierId id_with_profile = notifier_id; |
| id_with_profile.profile_id = profile_id; |
| return blocker()->ShouldShowNotificationAsPopup(id_with_profile); |
| } |
| |
| bool ShouldShowNotification( |
| const message_center::NotifierId& notifier_id, |
| const std::string profile_id) { |
| message_center::NotifierId id_with_profile = notifier_id; |
| id_with_profile.profile_id = profile_id; |
| return blocker()->ShouldShowNotification(id_with_profile); |
| } |
| |
| aura::Window* CreateWindowForProfile(const std::string& name) { |
| aura::Window* window = CreateTestWindowInShellWithId(window_id_++); |
| chrome::MultiUserWindowManager::GetInstance()->SetWindowOwner(window, name); |
| return window; |
| } |
| |
| private: |
| int state_changed_count_; |
| TestingProfileManager testing_profile_manager_; |
| int window_id_; |
| |
| DISALLOW_COPY_AND_ASSIGN(MultiUserNotificationBlockerChromeOSTest); |
| }; |
| |
| TEST_F(MultiUserNotificationBlockerChromeOSTest, Basic) { |
| ASSERT_EQ(chrome::MultiUserWindowManager::MULTI_PROFILE_MODE_SEPARATED, |
| chrome::MultiUserWindowManager::GetMultiProfileMode()); |
| |
| message_center::NotifierId notifier_id( |
| message_center::NotifierId::APPLICATION, "test-app"); |
| // Only allowed the system notifier. |
| message_center::NotifierId ash_system_notifier( |
| message_center::NotifierId::SYSTEM_COMPONENT, |
| ash::system_notifier::kNotifierDisplay); |
| // Other system notifiers should be treated as same as a normal notifier. |
| message_center::NotifierId random_system_notifier( |
| message_center::NotifierId::SYSTEM_COMPONENT, "random_system_component"); |
| |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(notifier_id, "")); |
| EXPECT_TRUE(ShouldShowNotificationAsPopup(ash_system_notifier, "")); |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(random_system_notifier, "")); |
| EXPECT_TRUE(ShouldShowNotificationAsPopup(notifier_id, GetDefaultUserId())); |
| EXPECT_FALSE(ShouldShowNotification(notifier_id, "")); |
| EXPECT_TRUE(ShouldShowNotification(ash_system_notifier, "")); |
| EXPECT_FALSE(ShouldShowNotification(random_system_notifier, "")); |
| EXPECT_TRUE(ShouldShowNotification(notifier_id, GetDefaultUserId())); |
| EXPECT_TRUE(ShouldShowNotification(random_system_notifier, |
| GetDefaultUserId())); |
| |
| CreateProfile("test2@example.com"); |
| EXPECT_EQ(0, GetStateChangedCountAndReset()); |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(notifier_id, "")); |
| EXPECT_TRUE(ShouldShowNotificationAsPopup(ash_system_notifier, "")); |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(random_system_notifier, "")); |
| EXPECT_TRUE(ShouldShowNotificationAsPopup(notifier_id, GetDefaultUserId())); |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(notifier_id, "test2@example.com")); |
| EXPECT_TRUE(ShouldShowNotificationAsPopup(random_system_notifier, |
| GetDefaultUserId())); |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(random_system_notifier, |
| "test2@example.com")); |
| EXPECT_FALSE(ShouldShowNotification(notifier_id, "")); |
| EXPECT_TRUE(ShouldShowNotification(ash_system_notifier, "")); |
| EXPECT_FALSE(ShouldShowNotification(random_system_notifier, "")); |
| EXPECT_TRUE(ShouldShowNotification(notifier_id, GetDefaultUserId())); |
| EXPECT_FALSE(ShouldShowNotification(notifier_id, "test2@example.com")); |
| EXPECT_TRUE(ShouldShowNotification(random_system_notifier, |
| GetDefaultUserId())); |
| EXPECT_FALSE(ShouldShowNotification(random_system_notifier, |
| "test2@example.com")); |
| |
| SwitchActiveUser("test2@example.com"); |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(notifier_id, "")); |
| EXPECT_TRUE(ShouldShowNotificationAsPopup(ash_system_notifier, "")); |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(random_system_notifier, "")); |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(notifier_id, GetDefaultUserId())); |
| EXPECT_TRUE(ShouldShowNotificationAsPopup(notifier_id, "test2@example.com")); |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(random_system_notifier, |
| GetDefaultUserId())); |
| EXPECT_TRUE(ShouldShowNotificationAsPopup(random_system_notifier, |
| "test2@example.com")); |
| EXPECT_FALSE(ShouldShowNotification(notifier_id, "")); |
| EXPECT_TRUE(ShouldShowNotification(ash_system_notifier, "")); |
| EXPECT_FALSE(ShouldShowNotification(random_system_notifier, "")); |
| EXPECT_FALSE(ShouldShowNotification(notifier_id, GetDefaultUserId())); |
| EXPECT_TRUE(ShouldShowNotification(notifier_id, "test2@example.com")); |
| EXPECT_FALSE(ShouldShowNotification(random_system_notifier, |
| GetDefaultUserId())); |
| EXPECT_TRUE(ShouldShowNotification(random_system_notifier, |
| "test2@example.com")); |
| |
| SwitchActiveUser(GetDefaultUserId()); |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(notifier_id, "")); |
| EXPECT_TRUE(ShouldShowNotificationAsPopup(ash_system_notifier, "")); |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(random_system_notifier, "")); |
| EXPECT_TRUE(ShouldShowNotificationAsPopup(notifier_id, GetDefaultUserId())); |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(notifier_id, "test2@example.com")); |
| EXPECT_TRUE(ShouldShowNotificationAsPopup(random_system_notifier, |
| GetDefaultUserId())); |
| EXPECT_FALSE(ShouldShowNotificationAsPopup(random_system_notifier, |
| "test2@example.com")); |
| EXPECT_FALSE(ShouldShowNotification(notifier_id, "")); |
| EXPECT_TRUE(ShouldShowNotification(ash_system_notifier, "")); |
| EXPECT_FALSE(ShouldShowNotification(random_system_notifier, "")); |
| EXPECT_TRUE(ShouldShowNotification(notifier_id, GetDefaultUserId())); |
| EXPECT_FALSE(ShouldShowNotification(notifier_id, "test2@example.com")); |
| EXPECT_TRUE(ShouldShowNotification(random_system_notifier, |
| GetDefaultUserId())); |
| EXPECT_FALSE(ShouldShowNotification(random_system_notifier, |
| "test2@example.com")); |
| } |
