chrome/browser/permissions/adaptive_quiet_notification_permission_ui_enabler_unittest.cc

// Copyright 2019 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 "chrome/browser/permissions/adaptive_quiet_notification_permission_ui_enabler.h"
#include <memory>

#include "base/test/scoped_feature_list.h"
#include "chrome/browser/permissions/quiet_notification_permission_ui_state.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/common/chrome_features.h"
#include "chrome/common/pref_names.h"
#include "chrome/test/base/testing_profile.h"
#include "components/content_settings/core/common/pref_names.h"
#include "components/prefs/testing_pref_service.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace {

using QuietUiEnablingMethod =
    QuietNotificationPermissionUiState::EnablingMethod;

}  // namespace

class AdaptiveQuietNotificationPermissionUiEnablerTest : public testing::Test {
 public:
  AdaptiveQuietNotificationPermissionUiEnablerTest()
      : testing_profile_(std::make_unique<TestingProfile>()) {}
  ~AdaptiveQuietNotificationPermissionUiEnablerTest() override = default;

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

  TestingProfile* profile() { return testing_profile_.get(); }

 private:
  content::BrowserTaskEnvironment task_environment_;
  std::unique_ptr<TestingProfile> testing_profile_;
};

// Check the |BackfillEnablingMethodIfMissing| method.
TEST_F(AdaptiveQuietNotificationPermissionUiEnablerTest,
       BackfillEnablingMethodIfMissing) {
  struct {
    bool enable_quiet_ui_pref;
    base::Optional<QuietUiEnablingMethod> quiet_ui_method_pref;
    bool should_show_promo_pref;
    QuietUiEnablingMethod expected_quiet_ui_method_pref;
  } kTests[] = {
      // If the quiet ui is not enabled, the method is always unspecified.
      {false, QuietUiEnablingMethod::kUnspecified, false,
       QuietUiEnablingMethod::kUnspecified},
      {false, QuietUiEnablingMethod::kAdaptive, false,
       QuietUiEnablingMethod::kUnspecified},
      {false, QuietUiEnablingMethod::kManual, true,
       QuietUiEnablingMethod::kUnspecified},
      // If the method is set already it will stay the same.
      {true, QuietUiEnablingMethod::kAdaptive, false,
       QuietUiEnablingMethod::kAdaptive},
      {true, QuietUiEnablingMethod::kManual, true,
       QuietUiEnablingMethod::kManual},
      // If the method is unspecified, use the promo perf to decide the method.
      {true, QuietUiEnablingMethod::kUnspecified, true,
       QuietUiEnablingMethod::kAdaptive},
      {true, QuietUiEnablingMethod::kUnspecified, false,
       QuietUiEnablingMethod::kManual},
      // If the method is unset, it should be treated as kUnspecified.
      {false, base::nullopt, false, QuietUiEnablingMethod::kUnspecified},
      {true, base::nullopt, true, QuietUiEnablingMethod::kAdaptive},
      {true, base::nullopt, false, QuietUiEnablingMethod::kManual},
  };

  base::test::ScopedFeatureList feature_list;
  feature_list.InitAndEnableFeature(features::kQuietNotificationPrompts);

  auto* permission_ui_enabler =
      AdaptiveQuietNotificationPermissionUiEnabler::GetForProfile(profile());

  int test_case = 0;

  for (const auto& kTest : kTests) {
    SCOPED_TRACE(test_case++);
    profile()->GetPrefs()->SetBoolean(
        prefs::kEnableQuietNotificationPermissionUi,
        kTest.enable_quiet_ui_pref);
    if (kTest.quiet_ui_method_pref.has_value()) {
      profile()->GetPrefs()->SetInteger(
          prefs::kQuietNotificationPermissionUiEnablingMethod,
          static_cast<int>(kTest.quiet_ui_method_pref.value()));
    } else {
      profile()->GetPrefs()->ClearPref(
          prefs::kQuietNotificationPermissionUiEnablingMethod);
    }
    profile()->GetPrefs()->SetBoolean(
        prefs::kQuietNotificationPermissionShouldShowPromo,
        kTest.should_show_promo_pref);

    permission_ui_enabler->BackfillEnablingMethodIfMissingForTesting();

    EXPECT_EQ(kTest.expected_quiet_ui_method_pref,
              QuietNotificationPermissionUiState::GetQuietUiEnablingMethod(
                  profile()));
  }
}