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

#include "ash/constants/ash_features.h"
#include "ash/constants/ash_pref_names.h"
#include "chrome/browser/ash/input_method/editor_geolocation_mock_provider.h"
#include "chrome/browser/ash/input_method/editor_geolocation_provider.h"
#include "chrome/browser/ash/input_method/editor_mediator.h"
#include "chrome/browser/ash/policy/handlers/configuration_policy_handler_ash.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/global_features.h"
#include "chrome/browser/policy/policy_test_utils.h"
#include "chrome/browser/policy/profile_policy_connector.h"
#include "chrome/browser/profiles/profile.h"
#include "chromeos/constants/chromeos_features.h"
#include "components/policy/policy_constants.h"
#include "components/prefs/pref_service.h"
#include "content/public/test/browser_test.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace policy {

class OrcaPolicyTest : public PolicyTest {
 public:
  OrcaPolicyTest() {
    feature_list_.InitWithFeatures(
        /*enabled_features=*/{chromeos::features::kOrca,
                              chromeos::features::kFeatureManagementOrca,
                              ash::features::kOrcaForManagedUsers},
        // TODO: b:329215512: Remove the OrcaUseAccountCapabilities from the
        // disable list.
        /*disabled_features=*/{chromeos::features::kOrcaDogfood,
                               ash::features::kOrcaUseAccountCapabilities});
  }

 public:
  Profile* profile() { return browser()->GetProfile(); }

 protected:
  base::test::ScopedFeatureList feature_list_;
};

IN_PROC_BROWSER_TEST_F(OrcaPolicyTest, EnablesChromeOSHMWIfOrcaPolicyUnset) {
  ash::input_method::EditorMediator editor_mediator(
      g_browser_process->GetFeatures()->application_locale_storage(), profile(),
      std::make_unique<ash::input_method::EditorGeolocationMockProvider>("au"));

  EXPECT_EQ(profile()->GetPrefs()->GetInteger(ash::prefs::kHmwManagedSettings),
            0);
  EXPECT_TRUE(editor_mediator.IsAllowedForUse());
  EXPECT_TRUE(editor_mediator.CanAccessFeedback());
}

IN_PROC_BROWSER_TEST_F(
    OrcaPolicyTest,
    EnablesChromeOSHMWAndFeedbackIfPolicySetToEnabledWithAIImprovement) {
  ash::input_method::EditorMediator editor_mediator(
      g_browser_process->GetFeatures()->application_locale_storage(), profile(),
      std::make_unique<ash::input_method::EditorGeolocationMockProvider>("au"));
  PolicyMap policies;

  policies.Set(key::kHelpMeWriteSettings, POLICY_LEVEL_MANDATORY,
               POLICY_SCOPE_USER, POLICY_SOURCE_CLOUD, base::Value(0), nullptr);
  UpdateProviderPolicy(policies);

  EXPECT_EQ(profile()->GetPrefs()->GetInteger(ash::prefs::kHmwManagedSettings),
            0);
  EXPECT_TRUE(editor_mediator.IsAllowedForUse());
  EXPECT_TRUE(editor_mediator.CanAccessFeedback());
}

IN_PROC_BROWSER_TEST_F(
    OrcaPolicyTest,
    EnablesChromeOSHMWButDisableFeedbackIfPolicySetToEnabledWithoutAIImprovement) {
  ash::input_method::EditorMediator editor_mediator(
      g_browser_process->GetFeatures()->application_locale_storage(), profile(),
      std::make_unique<ash::input_method::EditorGeolocationMockProvider>("au"));
  PolicyMap policies;

  policies.Set(key::kHelpMeWriteSettings, POLICY_LEVEL_MANDATORY,
               POLICY_SCOPE_USER, POLICY_SOURCE_CLOUD, base::Value(1), nullptr);
  UpdateProviderPolicy(policies);

  EXPECT_EQ(profile()->GetPrefs()->GetInteger(ash::prefs::kHmwManagedSettings),
            1);
  EXPECT_TRUE(editor_mediator.IsAllowedForUse());
  EXPECT_FALSE(editor_mediator.CanAccessFeedback());
}

IN_PROC_BROWSER_TEST_F(OrcaPolicyTest,
                       DisableChromeOSHMWFeedbackIfPolicySetToDisabled) {
  ash::input_method::EditorMediator editor_mediator(
      g_browser_process->GetFeatures()->application_locale_storage(), profile(),
      std::make_unique<ash::input_method::EditorGeolocationMockProvider>("au"));
  PolicyMap policies;

  policies.Set(key::kHelpMeWriteSettings, POLICY_LEVEL_MANDATORY,
               POLICY_SCOPE_USER, POLICY_SOURCE_CLOUD, base::Value(2), nullptr);
  UpdateProviderPolicy(policies);

  EXPECT_EQ(profile()->GetPrefs()->GetInteger(ash::prefs::kHmwManagedSettings),
            2);
  EXPECT_TRUE(editor_mediator.IsAllowedForUse());
  EXPECT_FALSE(editor_mediator.CanAccessFeedback());
}

}  // namespace policy