// Copyright 2017 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/ash/login/smart_lock/smart_lock_notification_controller.h"

#include "ash/constants/ash_features.h"
#include "ash/public/cpp/notification_utils.h"
#include "base/check_deref.h"
#include "base/strings/utf_string_conversions.h"
#include "chrome/test/base/browser_with_test_window_test.h"
#include "chromeos/ash/components/browser_context_helper/browser_context_helper.h"
#include "chromeos/ash/components/proximity_auth/proximity_auth_pref_names.h"
#include "components/sync_preferences/testing_pref_service_syncable.h"
#include "components/user_manager/user.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "ui/message_center/message_center.h"
#include "ui/message_center/public/cpp/notification.h"
#include "ui/message_center/public/cpp/notification_types.h"

namespace ash {
namespace {

const char kPhoneName[] = "Nexus 6";
const char16_t kPhoneName16[] = u"Nexus 6";

class TestableNotificationController : public SmartLockNotificationController {
 public:
  explicit TestableNotificationController(Profile* profile)
      : SmartLockNotificationController(profile) {}

  TestableNotificationController(const TestableNotificationController&) =
      delete;
  TestableNotificationController& operator=(
      const TestableNotificationController&) = delete;

  ~TestableNotificationController() override = default;

  // SmartLockNotificationController:
  MOCK_METHOD0(LaunchMultiDeviceSettings, void());
  MOCK_METHOD0(LockScreen, void());
};

class SmartLockNotificationControllerTest : public BrowserWithTestWindowTest {
 protected:
  SmartLockNotificationControllerTest() = default;

  SmartLockNotificationControllerTest(
      const SmartLockNotificationControllerTest&) = delete;
  SmartLockNotificationControllerTest& operator=(
      const SmartLockNotificationControllerTest&) = delete;

  ~SmartLockNotificationControllerTest() override = default;

  void SetUp() override {
    BrowserWithTestWindowTest::SetUp();

    notification_controller_ =
        std::make_unique<testing::StrictMock<TestableNotificationController>>(
            profile());
  }

  const message_center::Notification* GetNotification(
      const std::string& notification_id) {
    const user_manager::User& user = CHECK_DEREF(
        BrowserContextHelper::Get()->GetUserByBrowserContext(profile()));
    return message_center::MessageCenter::Get()->FindVisibleNotificationById(
        CreateUserScopedNotificationId(notification_id, user.username_hash()));
  }

  std::unique_ptr<testing::StrictMock<TestableNotificationController>>
      notification_controller_;
};

TEST_F(SmartLockNotificationControllerTest,
       TestShowChromebookAddedNotification) {
  const char kNotificationId[] = "easyunlock_notification_ids.chromebook_added";

  notification_controller_->ShowChromebookAddedNotification();
  const message_center::Notification* notification =
      GetNotification(kNotificationId);
  ASSERT_TRUE(notification);
  ASSERT_EQ(1u, notification->buttons().size());
  EXPECT_EQ(message_center::SYSTEM_PRIORITY, notification->priority());

  // Clicking notification button should launch settings.
  EXPECT_CALL(*notification_controller_, LaunchMultiDeviceSettings());
  notification->delegate()->Click(0, std::nullopt);

  // Clicking the notification itself should also launch settings.
  EXPECT_CALL(*notification_controller_, LaunchMultiDeviceSettings());
  notification->delegate()->Click(std::nullopt, std::nullopt);
}

TEST_F(SmartLockNotificationControllerTest, TestShowPairingChangeNotification) {
  const char kNotificationId[] = "easyunlock_notification_ids.pairing_change";

  notification_controller_->ShowPairingChangeNotification();
  const message_center::Notification* notification =
      GetNotification(kNotificationId);
  ASSERT_TRUE(notification);
  ASSERT_EQ(2u, notification->buttons().size());
  EXPECT_EQ(message_center::SYSTEM_PRIORITY, notification->priority());

  // Clicking the notification itself should do nothing.
  notification->delegate()->Click(std::nullopt, std::nullopt);

  // Clicking 1st notification button should lock screen settings.
  EXPECT_CALL(*notification_controller_, LockScreen());
  notification->delegate()->Click(0, std::nullopt);

  // Clicking 2nd notification button should launch settings.
  EXPECT_CALL(*notification_controller_, LaunchMultiDeviceSettings());
  notification->delegate()->Click(1, std::nullopt);
}

TEST_F(SmartLockNotificationControllerTest,
       TestShowPairingChangeAppliedNotification) {
  const char kNotificationId[] =
      "easyunlock_notification_ids.pairing_change_applied";

  notification_controller_->ShowPairingChangeAppliedNotification(kPhoneName);
  const message_center::Notification* notification =
      GetNotification(kNotificationId);
  ASSERT_TRUE(notification);
  ASSERT_EQ(1u, notification->buttons().size());
  EXPECT_EQ(message_center::SYSTEM_PRIORITY, notification->priority());

  // Check that the phone name is in the notification message.
  EXPECT_NE(std::string::npos, notification->message().find(kPhoneName16));

  // Clicking notification button should launch settings.
  EXPECT_CALL(*notification_controller_, LaunchMultiDeviceSettings());
  notification->delegate()->Click(0, std::nullopt);

  // Clicking the notification itself should also launch settings.
  EXPECT_CALL(*notification_controller_, LaunchMultiDeviceSettings());
  notification->delegate()->Click(std::nullopt, std::nullopt);
}

TEST_F(SmartLockNotificationControllerTest,
       PairingAppliedRemovesPairingChange) {
  const char kPairingChangeId[] = "easyunlock_notification_ids.pairing_change";
  const char kPairingAppliedId[] =
      "easyunlock_notification_ids.pairing_change_applied";

  notification_controller_->ShowPairingChangeNotification();
  EXPECT_TRUE(GetNotification(kPairingChangeId));

  notification_controller_->ShowPairingChangeAppliedNotification(kPhoneName);
  EXPECT_FALSE(GetNotification(kPairingChangeId));
  EXPECT_TRUE(GetNotification(kPairingAppliedId));
}

}  // namespace
}  // namespace ash