chrome/browser/hid/hid_pinned_notification_unittest.cc

// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/hid/hid_pinned_notification.h"

#include "chrome/browser/hid/hid_connection_tracker.h"
#include "chrome/browser/hid/hid_connection_tracker_factory.h"
#include "chrome/browser/hid/hid_system_tray_icon_unittest.h"
#include "chrome/browser/notifications/notification_display_service_tester.h"
#include "chrome/browser/notifications/system_notification_helper.h"
#include "chrome/test/base/testing_browser_process.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

class HidPinnedNotificationTest : public HidSystemTrayIconTestBase {
 public:
  void SetUp() override {
    HidSystemTrayIconTestBase::SetUp();
    TestingBrowserProcess::GetGlobal()->SetSystemNotificationHelper(
        std::make_unique<SystemNotificationHelper>());
    display_service_ =
        std::make_unique<NotificationDisplayServiceTester>(/*profile=*/nullptr);
  }

  void TearDown() override {
    TestingBrowserProcess::GetGlobal()->SetSystemNotificationHelper(nullptr);
    HidSystemTrayIconTestBase::TearDown();
  }

  void CheckIcon(const std::vector<std::pair<Profile*, size_t>>&
                     profile_connection_counts) override {
    EXPECT_FALSE(display_service_
                     ->GetDisplayedNotificationsForType(
                         NotificationHandler::Type::TRANSIENT)
                     .empty());

    // Check each button label and behavior of clicking the button.
    for (const auto& pair : profile_connection_counts) {
      auto* hid_connection_tracker = static_cast<MockHidConnectionTracker*>(
          HidConnectionTrackerFactory::GetForProfile(pair.first,
                                                     /*create=*/false));
      EXPECT_TRUE(hid_connection_tracker);
      auto maybe_notification = display_service_->GetNotification(
          HidPinnedNotification::GetNotificationId(pair.first));
      ASSERT_TRUE(maybe_notification);
      EXPECT_EQ(maybe_notification->title(), GetExpectedIconTooltip(
                                                 /*num_devices=*/pair.second));

      EXPECT_EQ(maybe_notification->rich_notification_data().buttons.size(),
                1u);
      EXPECT_EQ(maybe_notification->rich_notification_data().buttons[0].title,
                GetExpectedButtonTitleForProfile(pair.first));
      EXPECT_TRUE(maybe_notification->delegate());

      EXPECT_CALL(*hid_connection_tracker, ShowHidContentSettingsExceptions());
      SimulateButtonClick(pair.first);
    }
  }

  void CheckIconHidden() override {
    EXPECT_TRUE(display_service_
                    ->GetDisplayedNotificationsForType(
                        NotificationHandler::Type::TRANSIENT)
                    .empty());
  }

 private:
  void SimulateButtonClick(Profile* profile) {
    display_service_->SimulateClick(
        NotificationHandler::Type::TRANSIENT,
        HidPinnedNotification::GetNotificationId(profile),
        /*action_index=*/0, /*reply=*/absl::nullopt);
  }

  std::unique_ptr<NotificationDisplayServiceTester> display_service_;
};

TEST_F(HidPinnedNotificationTest, SingleProfileEmptyName) {
  // Current TestingProfileManager can't support empty profile name as it uses
  // profile name for profile path. Passing empty would result in a failure in
  // ProfileManager::IsAllowedProfilePath(). Changing the way
  // TestingProfileManager creating profile path like adding "profile" prefix
  // doesn't work either as some tests are written in a way that takes
  // assumption of testing profile path pattern. Hence it creates testing
  // profile with non-empty name and then change the profile name to empty which
  // can still achieve what this file wants to test.
  profile()->set_profile_name("");
  TestSingleProfile();
}

TEST_F(HidPinnedNotificationTest, SingleProfileNonEmptyName) {
  TestSingleProfile();
}

TEST_F(HidPinnedNotificationTest, MultipleProfiles) {
  TestMultipleProfiles();
}