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

#include "components/webui/flags/flags_state.h"

#include <stddef.h>

#include <array>
#include <map>
#include <memory>
#include <set>
#include <string>

#include "base/feature_list.h"
#include "base/format_macros.h"
#include "base/functional/bind.h"
#include "base/metrics/field_trial.h"
#include "base/metrics/field_trial_params.h"
#include "base/strings/stringprintf.h"
#include "base/strings/utf_string_conversions.h"
#include "base/test/scoped_feature_list.h"
#include "base/values.h"
#include "build/build_config.h"
#include "components/prefs/pref_registry_simple.h"
#include "components/prefs/testing_pref_service.h"
#include "components/strings/grit/components_strings.h"
#include "components/variations/variations_associated_data.h"
#include "components/variations/variations_switches.h"
#include "components/webui/flags/feature_entry.h"
#include "components/webui/flags/feature_entry_macros.h"
#include "components/webui/flags/flags_ui_pref_names.h"
#include "components/webui/flags/flags_ui_switches.h"
#include "components/webui/flags/pref_service_flags_storage.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace flags_ui {

namespace {

const char kFlags1[] = "flag1";
const char kFlags2[] = "flag2";
const char kFlags3[] = "flag3";
const char kFlags4[] = "flag4";
const char kFlags5[] = "flag5";
const char kFlags6[] = "flag6";
const char kFlags7[] = "flag7";
const char kFlags8[] = "flag8";
const char kFlags9[] = "flag9";
const char kFlags10[] = "flag10";
const char kFlags11[] = "flag11";
const char kFlags12[] = "flag12";
const char kFlags13[] = "flag13";
const char kFlags14[] = "flag14";

// Matches an entry of kRenamedFlags to test rename migration.
const char kOldFlagName[] = "prompt-api-for-gemini-nano";
const char kNewFlagName[] = "prompt-api";

const char kSwitch1[] = "switch";
const char kSwitch2[] = "switch2";
const char kSwitch3[] = "switch3";
const char kSwitch6[] = "switch6";
const char kValueForSwitch2[] = "value_for_switch2";

const char kStringSwitch[] = "string_switch";
const char kValueForStringSwitch[] = "value_for_string_switch";

const char kMultiSwitch1[] = "multi_switch1";
const char kMultiSwitch2[] = "multi_switch2";
const char kValueForMultiSwitch2[] = "value_for_multi_switch2";

const char kEnableDisableValue1[] = "value1";
const char kEnableDisableValue2[] = "value2";

const char kEnableFeatures[] = "dummy-enable-features";
const char kDisableFeatures[] = "dummy-disable-features";

const char kTestTrial[] = "TestTrial";
const char kTestParam1[] = "param1";
const char kTestParam2[] = "param2";
const char kTestParam3[] = "param:/3";
const char kTestParamValue[] = "value";

BASE_FEATURE(kTestFeature1, "FeatureName1", base::FEATURE_ENABLED_BY_DEFAULT);
BASE_FEATURE(kTestFeature2, "FeatureName2", base::FEATURE_ENABLED_BY_DEFAULT);
BASE_FEATURE(kTestFeature3, "FeatureName3", base::FEATURE_DISABLED_BY_DEFAULT);

const FeatureEntry::FeatureParam kTestVariationOther1[] = {
    {kTestParam1, kTestParamValue}};
const FeatureEntry::FeatureParam kTestVariationOther2[] = {
    {kTestParam2, kTestParamValue}};
const FeatureEntry::FeatureParam kTestVariationOther3[] = {
    {kTestParam1, kTestParamValue},
    {kTestParam3, kTestParamValue},
};

const FeatureEntry::FeatureVariation kTestVariations1[] = {
    {"dummy description 1", kTestVariationOther1, nullptr}};
const FeatureEntry::FeatureVariation kTestVariations2[] = {
    {"dummy description 2", kTestVariationOther2, nullptr}};
const FeatureEntry::FeatureVariation kTestVariations3[] = {
    {"dummy description 1", kTestVariationOther1, nullptr},
    {"dummy description 2", kTestVariationOther2, nullptr},
    {"dummy description 3", kTestVariationOther3, "t123456"}};

const char kTestVariation3Cmdline[] =
    "FeatureName3:param1/value/param%3A%2F3/value";

const char kDummyName[] = "";
const char kDummyDescription[] = "";

bool SkipFeatureEntry(const FeatureEntry& feature_entry) {
  return false;
}

#if BUILDFLAG(IS_ANDROID)
class MockJniDelegate : public cached_flags::JniDelegate {
 public:
  ~MockJniDelegate() override = default;

  MOCK_METHOD((void),
              CacheNativeFlagsImmediately,
              ((const std::map<std::string, std::string>&)),
              (override));

  MOCK_METHOD(
      (void),
      CacheFeatureParamsImmediately,
      ((const std::map<std::string, std::map<std::string, std::string>>&)),
      (override));

  MOCK_METHOD((void),
              EraseNativeFlagCachedValues,
              ((const std::vector<std::string>&)),
              (override));

  MOCK_METHOD((void),
              EraseFeatureParamCachedValues,
              ((const std::vector<std::string>&)),
              (override));
};
#endif

}  // namespace

const FeatureEntry::Choice kMultiChoices[] = {
    {kDummyDescription, "", ""},
    {kDummyDescription, kMultiSwitch1, ""},
    {kDummyDescription, kMultiSwitch2, kValueForMultiSwitch2},
};

const FeatureEntry::Choice kMultiChoicesWithEnableDisableFeatures1[] = {
    {kDummyDescription, "", ""},
    {kDummyDescription, kEnableFeatures,
     "MultiChoiceFeature1,MultiChoiceFeature2"},
    {kDummyDescription, kDisableFeatures, "MultiChoiceFeature1"},
};

const FeatureEntry::Choice kMultiChoicesWithEnableDisableFeatures2[] = {
    {kDummyDescription, "", ""},
    {kDummyDescription, kEnableFeatures, "MultiChoiceFeature3"},
    {kDummyDescription, kDisableFeatures, "MultiChoiceFeature3"},
};

// The entries that are set for these tests. The 3rd entry is not supported on
// the current platform, all others are.
auto kEntries = std::to_array<FeatureEntry>({
    {kFlags1, kDummyName, kDummyDescription,
     0,  // Ends up being mapped to the current platform.
     SINGLE_VALUE_TYPE(kSwitch1)},
    {kFlags2, kDummyName, kDummyDescription,
     0,  // Ends up being mapped to the current platform.
     SINGLE_VALUE_TYPE_AND_VALUE(kSwitch2, kValueForSwitch2)},
    {kFlags3, kDummyName, kDummyDescription,
     0,  // This ends up enabling for an OS other than the current.
     SINGLE_VALUE_TYPE(kSwitch3)},
    {kFlags4, kDummyName, kDummyDescription,
     0,  // Ends up being mapped to the current platform.
     MULTI_VALUE_TYPE(kMultiChoices)},
    {kFlags5, kDummyName, kDummyDescription,
     0,  // Ends up being mapped to the current platform.
     ENABLE_DISABLE_VALUE_TYPE_AND_VALUE(kSwitch1,
                                         kEnableDisableValue1,
                                         kSwitch2,
                                         kEnableDisableValue2)},
    {kFlags6, kDummyName, kDummyDescription, 0,
     SINGLE_DISABLE_VALUE_TYPE(kSwitch6)},
    {kFlags7, kDummyName, kDummyDescription,
     0,  // Ends up being mapped to the current platform.
     FEATURE_VALUE_TYPE(kTestFeature1)},
    {kFlags8, kDummyName, kDummyDescription,
     0,  // Ends up being mapped to the current platform.
     FEATURE_WITH_PARAMS_VALUE_TYPE(kTestFeature1,
                                    kTestVariations1,
                                    kTestTrial)},
    {kFlags9, kDummyName, kDummyDescription,
     0,  // Ends up being mapped to the current platform.
     FEATURE_WITH_PARAMS_VALUE_TYPE(kTestFeature1,
                                    kTestVariations1,
                                    kTestTrial)},
    {kFlags10, kDummyName, kDummyDescription,
     0,  // Ends up being mapped to the current platform.
     FEATURE_WITH_PARAMS_VALUE_TYPE(kTestFeature2,
                                    kTestVariations2,
                                    kTestTrial)},
    {kFlags11, kDummyName, kDummyDescription,
     0,  // Ends up being mapped to the current platform.
     ORIGIN_LIST_VALUE_TYPE(kStringSwitch, kValueForStringSwitch)},
    {kFlags12, kDummyName, kDummyDescription,
     0,  // Ends up being mapped to the current platform.
     FEATURE_WITH_PARAMS_VALUE_TYPE(kTestFeature3,
                                    kTestVariations3,
                                    kTestTrial)},
    {kFlags13, kDummyName, kDummyDescription,
     0,  // Ends up being mapped to the current platform.
     MULTI_VALUE_TYPE(kMultiChoicesWithEnableDisableFeatures1)},
    {kFlags14, kDummyName, kDummyDescription,
     0,  // Ends up being mapped to the current platform.
     MULTI_VALUE_TYPE(kMultiChoicesWithEnableDisableFeatures2)},
    {kNewFlagName, kDummyName, kDummyDescription,
     0,  // Ends up being mapped to the current platform.
     SINGLE_VALUE_TYPE(kSwitch1)},
});

class FlagsStateTest : public ::testing::Test,
                       public flags_ui::FlagsState::Delegate {
 protected:
  FlagsStateTest() : flags_storage_(&prefs_) {
    prefs_.registry()->RegisterListPref(prefs::kAboutFlagsEntries);
    prefs_.registry()->RegisterDictionaryPref(prefs::kAboutFlagsOriginLists);

    for (size_t i = 0; i < std::size(kEntries); ++i) {
      kEntries[i].supported_platforms = FlagsState::GetCurrentPlatform();
    }

    int os_other_than_current = 1;
    while (os_other_than_current == FlagsState::GetCurrentPlatform()) {
      os_other_than_current <<= 1;
    }
    kEntries[2].supported_platforms = os_other_than_current;
    flags_state_ = std::make_unique<FlagsState>(kEntries, this);

#if BUILDFLAG(IS_ANDROID)
    auto jni_delegate = std::make_unique<MockJniDelegate>();
    mock_jni_delegate_ = jni_delegate.get();
    flags_state_->SetJniDelegateForTesting(std::move(jni_delegate));
#endif
  }

  ~FlagsStateTest() override { variations::test::ClearAllVariationParams(); }

  // FlagsState::Delegate:
  bool ShouldExcludeFlag(const FlagsStorage* storage,
                         const FeatureEntry& entry) override {
    return exclude_flags_.count(entry.internal_name) != 0;
  }

  TestingPrefServiceSimple prefs_;
  PrefServiceFlagsStorage flags_storage_;
  std::unique_ptr<FlagsState> flags_state_;
  std::set<std::string> exclude_flags_;

#if BUILDFLAG(IS_ANDROID)
  raw_ptr<MockJniDelegate> mock_jni_delegate_;
#endif
};

TEST_F(FlagsStateTest, NoChangeNoRestart) {
  EXPECT_FALSE(flags_state_->IsRestartNeededToCommitChanges());
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags1, false);
  EXPECT_FALSE(flags_state_->IsRestartNeededToCommitChanges());

  // kFlags6 is enabled by default, so enabling should not require a restart.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags6, true);
  EXPECT_FALSE(flags_state_->IsRestartNeededToCommitChanges());
}

TEST_F(FlagsStateTest, ChangeNeedsRestart) {
  EXPECT_FALSE(flags_state_->IsRestartNeededToCommitChanges());
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags1, true);
  EXPECT_TRUE(flags_state_->IsRestartNeededToCommitChanges());
}

TEST_F(FlagsStateTest, RenamedFlagMigration) {
  flags_storage_.SetFlags({kOldFlagName});
  std::set<std::string> enabled_flags;
  flags_state_->GetSanitizedEnabledFlags(&flags_storage_, &enabled_flags);
  EXPECT_THAT(enabled_flags, ::testing::ElementsAre(kNewFlagName));
  EXPECT_THAT(flags_storage_.GetFlags(), ::testing::ElementsAre(kNewFlagName));
}

// Tests that disabling a default enabled entry requires a restart.
TEST_F(FlagsStateTest, DisableChangeNeedsRestart) {
  EXPECT_FALSE(flags_state_->IsRestartNeededToCommitChanges());
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags6, false);
  EXPECT_TRUE(flags_state_->IsRestartNeededToCommitChanges());
}

TEST_F(FlagsStateTest, MultiFlagChangeNeedsRestart) {
  const FeatureEntry& entry = kEntries[3];
  ASSERT_EQ(kFlags4, entry.internal_name);
  EXPECT_FALSE(flags_state_->IsRestartNeededToCommitChanges());
  // Enable the 2nd choice of the multi-value.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry.NameForOption(2),
                                       true);
  EXPECT_TRUE(flags_state_->IsRestartNeededToCommitChanges());
  flags_state_->Reset();
  EXPECT_FALSE(flags_state_->IsRestartNeededToCommitChanges());
  // Enable the default choice now.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry.NameForOption(0),
                                       true);
  EXPECT_TRUE(flags_state_->IsRestartNeededToCommitChanges());
}

TEST_F(FlagsStateTest, AddTwoFlagsRemoveOne) {
  // Add two entries, check they're there.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags1, true);
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags2, true);

  {
    const base::ListValue& entries_list =
        prefs_.GetList(prefs::kAboutFlagsEntries);
    ASSERT_EQ(2u, entries_list.size());

    std::string s0 = entries_list[0].GetString();
    std::string s1 = entries_list[1].GetString();

    EXPECT_TRUE(s0 == kFlags1 || s1 == kFlags1);
    EXPECT_TRUE(s0 == kFlags2 || s1 == kFlags2);
  }

  // Remove one entry, check the other's still around.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags2, false);

  {
    const base::ListValue& entries_list =
        prefs_.GetList(prefs::kAboutFlagsEntries);
    ASSERT_EQ(1u, entries_list.size());
    std::string s0 = entries_list[0].GetString();
    EXPECT_TRUE(s0 == kFlags1);
  }
}

TEST_F(FlagsStateTest, AddTwoFlagsRemoveBoth) {
  // Add two entries, check the pref exists.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags1, true);
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags2, true);
  {
    const base::ListValue& entries_list =
        prefs_.GetList(prefs::kAboutFlagsEntries);
    ASSERT_EQ(2u, entries_list.size());
  }

  // Remove both, the pref should have been removed completely.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags1, false);
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags2, false);
  {
    const base::ListValue& entries_list =
        prefs_.GetList(prefs::kAboutFlagsEntries);
    EXPECT_TRUE(entries_list.empty());
  }
}

TEST_F(FlagsStateTest, CombineOriginListValues) {
  // Add a value in prefs, and on command line.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags11, true);
  const std::string prefs_value =
      "http://a.test,http://c.test,http://dupe.test";
  flags_state_->SetOriginListFlag(kFlags11, prefs_value, &flags_storage_);
  ASSERT_EQ(flags_storage_.GetOriginListFlag(kFlags11), prefs_value);

  base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
  const std::string cli_value = "http://dupe.test,http://b.test";
  command_line.AppendSwitchASCII(kStringSwitch, cli_value);
  ASSERT_EQ(command_line.GetSwitchValueASCII(kStringSwitch), cli_value);

  flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                       kNoSentinels, kEnableFeatures,
                                       kDisableFeatures);

  // Lists are concatenated together with duplicates removed, but are not
  // sorted.
  EXPECT_EQ(command_line.GetSwitchValueASCII(kStringSwitch),
            "http://dupe.test,http://b.test,http://a.test,http://c.test");
}

TEST_F(FlagsStateTest, ConvertFlagsToSwitches) {
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags1, true);

  base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
  command_line.AppendSwitch("foo");

  EXPECT_TRUE(command_line.HasSwitch("foo"));
  EXPECT_FALSE(command_line.HasSwitch(kSwitch1));

  flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                       kAddSentinels, kEnableFeatures,
                                       kDisableFeatures);

  EXPECT_TRUE(command_line.HasSwitch("foo"));
  EXPECT_TRUE(command_line.HasSwitch(kSwitch1));
  EXPECT_TRUE(command_line.HasSwitch(switches::kFlagSwitchesBegin));
  EXPECT_TRUE(command_line.HasSwitch(switches::kFlagSwitchesEnd));

  base::CommandLine command_line2(base::CommandLine::NO_PROGRAM);

  flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line2,
                                       kNoSentinels, kEnableFeatures,
                                       kDisableFeatures);

  EXPECT_TRUE(command_line2.HasSwitch(kSwitch1));
  EXPECT_FALSE(command_line2.HasSwitch(switches::kFlagSwitchesBegin));
  EXPECT_FALSE(command_line2.HasSwitch(switches::kFlagSwitchesEnd));

  base::CommandLine command_line3(base::CommandLine::NO_PROGRAM);
  // Enable 3rd variation (@4 since 0 is enable).
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, std::string(kFlags12).append("@4"), true);

  flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line3,
                                       kNoSentinels, kEnableFeatures,
                                       kDisableFeatures);

  EXPECT_TRUE(command_line3.HasSwitch(kEnableFeatures));
  EXPECT_EQ(command_line3.GetSwitchValueASCII(kEnableFeatures),
            kTestVariation3Cmdline);
}

TEST_F(FlagsStateTest, RegisterAllFeatureVariationParameters) {
  const FeatureEntry& entry = kEntries[7];
  std::unique_ptr<base::FeatureList> feature_list(new base::FeatureList);

  // Select the "Default" variation.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry.NameForOption(0),
                                       true);
  flags_state_->RegisterAllFeatureVariationParameters(&flags_storage_,
                                                      feature_list.get());
  // No value should be associated.
  EXPECT_EQ("", base::GetFieldTrialParamValue(kTestTrial, kTestParam1));
  // The trial should not be created.
  base::FieldTrial* trial = base::FieldTrialList::Find(kTestTrial);
  EXPECT_EQ(nullptr, trial);

  // Select the default "Enabled" variation.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry.NameForOption(1),
                                       true);

  flags_state_->RegisterAllFeatureVariationParameters(&flags_storage_,
                                                      feature_list.get());
  // No value should be associated as this is the default option.
  EXPECT_EQ("", base::GetFieldTrialParamValue(kTestTrial, kTestParam1));

  // The trial should be created.
  trial = base::FieldTrialList::Find(kTestTrial);
  EXPECT_NE(nullptr, trial);
  // The about:flags group should be selected for the trial.
  EXPECT_EQ(internal::kTrialGroupAboutFlags, trial->group_name());

  // Select the only one variation.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry.NameForOption(2),
                                       true);
  flags_state_->RegisterAllFeatureVariationParameters(&flags_storage_,
                                                      feature_list.get());
  // Associating for the second time should not change the value.
  EXPECT_EQ("", base::GetFieldTrialParamValue(kTestTrial, kTestParam1));
}

TEST_F(FlagsStateTest, RegisterAllFeatureVariationParametersNonDefault) {
  const FeatureEntry& entry = kEntries[7];
  std::unique_ptr<base::FeatureList> feature_list(new base::FeatureList);

  // Select the only one variation.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry.NameForOption(2),
                                       true);
  flags_state_->RegisterAllFeatureVariationParameters(&flags_storage_,
                                                      feature_list.get());

  // Set the feature_list as the main instance so that
  // base::GetFieldTrialParamValueByFeature below works.
  base::test::ScopedFeatureList scoped_feature_list;
  scoped_feature_list.InitWithFeatureList(std::move(feature_list));

  // The param should have the value predefined in this variation.
  EXPECT_EQ(kTestParamValue,
            base::GetFieldTrialParamValue(kTestTrial, kTestParam1));

  // The value should be associated also via the name of the feature.
  EXPECT_EQ(kTestParamValue,
            base::GetFieldTrialParamValueByFeature(kTestFeature1, kTestParam1));
}

// Verifies that variation IDs are still correctly collected and returned by
// RegisterAllFeatureVariationParameters().
TEST_F(FlagsStateTest, RegisterAllFeatureVariationParametersVariationIds) {
  const FeatureEntry& entry = kEntries[11];
  ASSERT_EQ(kFlags12, entry.internal_name);
  std::unique_ptr<base::FeatureList> feature_list =
      std::make_unique<base::FeatureList>();

  // Select the 3rd variation (@4).
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, std::string(kFlags12).append("@4"), true);

  std::vector<std::string> variation_ids =
      flags_state_->RegisterAllFeatureVariationParameters(&flags_storage_,
                                                          feature_list.get());

  ASSERT_EQ(1u, variation_ids.size());
  EXPECT_EQ("t123456", variation_ids[0]);
}

TEST_F(FlagsStateTest, RegisterAllFeatureVariationParametersWithDefaultTrials) {
  const FeatureEntry& entry1 = kEntries[8];
  const FeatureEntry& entry2 = kEntries[9];
  std::unique_ptr<base::FeatureList> feature_list(new base::FeatureList);

  // Select the only one variation for each FeatureEntry.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry1.NameForOption(2),
                                       true);
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry2.NameForOption(2),
                                       true);
  flags_state_->RegisterAllFeatureVariationParameters(&flags_storage_,
                                                      feature_list.get());

  // Set the feature_list as the main instance so that
  // base::GetFieldTrialParamValueByFeature below works.
  base::test::ScopedFeatureList scoped_feature_list;
  scoped_feature_list.InitWithFeatureList(std::move(feature_list));

  // The params should have the values predefined in these variations
  // (accessible via the names of the features).
  EXPECT_EQ(kTestParamValue,
            base::GetFieldTrialParamValueByFeature(kTestFeature1, kTestParam1));
  EXPECT_EQ(kTestParamValue,
            base::GetFieldTrialParamValueByFeature(kTestFeature2, kTestParam2));
  // The params are registered in the same trial.
  EXPECT_EQ(kTestParamValue,
            base::GetFieldTrialParamValue(kTestTrial, kTestParam1));
  EXPECT_EQ(kTestParamValue,
            base::GetFieldTrialParamValue(kTestTrial, kTestParam2));
}

base::CommandLine::StringType CreateSwitch(const std::string& value) {
#if BUILDFLAG(IS_WIN)
  return base::ASCIIToWide(value);
#else
  return value;
#endif
}

TEST_F(FlagsStateTest, RemoveFlagSwitches) {
  base::CommandLine::SwitchMap switch_list;
  switch_list[kSwitch1] = base::CommandLine::StringType();
  switch_list[switches::kFlagSwitchesBegin] = base::CommandLine::StringType();
  switch_list[switches::kFlagSwitchesEnd] = base::CommandLine::StringType();
  switch_list["foo"] = base::CommandLine::StringType();

  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags1, true);

  // This shouldn't do anything before ConvertFlagsToSwitches() wasn't called.
  flags_state_->RemoveFlagsSwitches(&switch_list);
  ASSERT_EQ(4u, switch_list.size());
  EXPECT_TRUE(switch_list.contains(kSwitch1));
  EXPECT_TRUE(switch_list.contains(switches::kFlagSwitchesBegin));
  EXPECT_TRUE(switch_list.contains(switches::kFlagSwitchesEnd));
  EXPECT_TRUE(switch_list.contains("foo"));

  // Call ConvertFlagsToSwitches(), then RemoveFlagsSwitches() again.
  base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
  command_line.AppendSwitch("foo");
  flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                       kAddSentinels, kEnableFeatures,
                                       kDisableFeatures);
  flags_state_->RemoveFlagsSwitches(&switch_list);

  // Now the about:flags-related switch should have been removed.
  ASSERT_EQ(1u, switch_list.size());
  EXPECT_TRUE(switch_list.contains("foo"));
}

TEST_F(FlagsStateTest, RemoveFlagSwitches_Features) {
  struct Cases {
    int enabled_choice;  // 0: default, 1: enabled, 2: disabled.
    const char* existing_enable_features;
    const char* existing_disable_features;
    const char* expected_enable_features;
    const char* expected_disable_features;
  };
  auto cases = std::to_array<Cases>({
      // Default value: Should not affect existing flags.
      {0, nullptr, nullptr, nullptr, nullptr},
      {0, "A,B", "C", "A,B", "C"},
      // "Enable" option: should only affect enabled list.
      {1, nullptr, nullptr, "FeatureName1", nullptr},
      {1, "A,B", "C", "A,B,FeatureName1", "C"},
      // "Disable" option: should only affect disabled list.
      {2, nullptr, nullptr, nullptr, "FeatureName1"},
      {2, "A,B", "C", "A,B", "C,FeatureName1"},
  });

  for (size_t i = 0; i < std::size(cases); ++i) {
    SCOPED_TRACE(base::StringPrintf(
        "Test[%" PRIuS "]: %d [%s] [%s]", i, cases[i].enabled_choice,
        cases[i].existing_enable_features ? cases[i].existing_enable_features
                                          : "null",
        cases[i].existing_disable_features ? cases[i].existing_disable_features
                                           : "null"));

    base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
    if (cases[i].existing_enable_features) {
      command_line.AppendSwitchASCII(kEnableFeatures,
                                     cases[i].existing_enable_features);
    }
    if (cases[i].existing_disable_features) {
      command_line.AppendSwitchASCII(kDisableFeatures,
                                     cases[i].existing_disable_features);
    }

    flags_state_->Reset();

    const std::string entry_name = base::StringPrintf(
        "%s%s%d", kFlags7, testing::kMultiSeparator, cases[i].enabled_choice);
    flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry_name, true);

    flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                         kAddSentinels, kEnableFeatures,
                                         kDisableFeatures);
    auto switch_list = command_line.GetSwitches();
    EXPECT_EQ(cases[i].expected_enable_features != nullptr,
              switch_list.contains(kEnableFeatures));
    if (cases[i].expected_enable_features) {
      EXPECT_EQ(CreateSwitch(cases[i].expected_enable_features),
                switch_list[kEnableFeatures]);
    }

    EXPECT_EQ(cases[i].expected_disable_features != nullptr,
              switch_list.contains(kDisableFeatures));
    if (cases[i].expected_disable_features) {
      EXPECT_EQ(CreateSwitch(cases[i].expected_disable_features),
                switch_list[kDisableFeatures]);
    }

    // RemoveFlagsSwitches() should result in the original values for these
    // switches.
    switch_list = command_line.GetSwitches();
    flags_state_->RemoveFlagsSwitches(&switch_list);
    EXPECT_EQ(cases[i].existing_enable_features != nullptr,
              switch_list.contains(kEnableFeatures));
    if (cases[i].existing_enable_features) {
      EXPECT_EQ(CreateSwitch(cases[i].existing_enable_features),
                switch_list[kEnableFeatures]);
    }
    EXPECT_EQ(cases[i].existing_disable_features != nullptr,
              switch_list.contains(kEnableFeatures));
    if (cases[i].existing_disable_features) {
      EXPECT_EQ(CreateSwitch(cases[i].existing_disable_features),
                switch_list[kDisableFeatures]);
    }
  }
}

// Tests enabling entries that aren't supported on the current platform.
TEST_F(FlagsStateTest, PersistAndPrune) {
  // Enable entries 1 and 3.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags1, true);
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags3, true);
  base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
  EXPECT_FALSE(command_line.HasSwitch(kSwitch1));
  EXPECT_FALSE(command_line.HasSwitch(kSwitch3));

  // Convert the flags to switches. Entry 3 shouldn't be among the switches
  // as it is not applicable to the current platform.
  flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                       kAddSentinels, kEnableFeatures,
                                       kDisableFeatures);
  EXPECT_TRUE(command_line.HasSwitch(kSwitch1));
  EXPECT_FALSE(command_line.HasSwitch(kSwitch3));

  // FeatureEntry 3 should show still be persisted in preferences though.
  const base::ListValue& entries_list =
      prefs_.GetList(prefs::kAboutFlagsEntries);
  EXPECT_EQ(2U, entries_list.size());
  std::string s0 = entries_list[0].GetString();
  EXPECT_EQ(kFlags1, s0);
  std::string s1 = entries_list[1].GetString();
  EXPECT_EQ(kFlags3, s1);
}

// Tests that switches which should have values get them in the command
// line.
TEST_F(FlagsStateTest, CheckValues) {
  // Enable entries 1 and 2.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags1, true);
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags2, true);
  base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
  EXPECT_FALSE(command_line.HasSwitch(kSwitch1));
  EXPECT_FALSE(command_line.HasSwitch(kSwitch2));

  // Convert the flags to switches.
  flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                       kAddSentinels, kEnableFeatures,
                                       kDisableFeatures);
  EXPECT_TRUE(command_line.HasSwitch(kSwitch1));
  EXPECT_EQ(std::string(), command_line.GetSwitchValueASCII(kSwitch1));
  EXPECT_TRUE(command_line.HasSwitch(kSwitch2));
  EXPECT_EQ(std::string(kValueForSwitch2),
            command_line.GetSwitchValueASCII(kSwitch2));

  // Confirm that there is no '=' in the command line for simple switches.
  std::string switch1_with_equals =
      std::string("--") + std::string(kSwitch1) + std::string("=");
#if BUILDFLAG(IS_WIN)
  EXPECT_EQ(std::wstring::npos, command_line.GetCommandLineString().find(
                                    base::ASCIIToWide(switch1_with_equals)));
#else
  EXPECT_EQ(std::string::npos,
            command_line.GetCommandLineString().find(switch1_with_equals));
#endif

  // And confirm there is a '=' for switches with values.
  std::string switch2_with_equals =
      std::string("--") + std::string(kSwitch2) + std::string("=");
#if BUILDFLAG(IS_WIN)
  EXPECT_NE(std::wstring::npos, command_line.GetCommandLineString().find(
                                    base::ASCIIToWide(switch2_with_equals)));
#else
  EXPECT_NE(std::string::npos,
            command_line.GetCommandLineString().find(switch2_with_equals));
#endif

  // And it should persist.
  const base::ListValue& entries_list =
      prefs_.GetList(prefs::kAboutFlagsEntries);
  EXPECT_EQ(2U, entries_list.size());
  std::string s0 = entries_list[0].GetString();
  EXPECT_EQ(kFlags1, s0);
  std::string s1 = entries_list[1].GetString();
  EXPECT_EQ(kFlags2, s1);
}

// Tests multi-value type entries.
TEST_F(FlagsStateTest, MultiValues) {
  const FeatureEntry& entry = kEntries[3];
  ASSERT_EQ(kFlags4, entry.internal_name);

  // Initially, the first "deactivated" option of the multi entry should
  // be set.
  {
    base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
    flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                         kAddSentinels, kEnableFeatures,
                                         kDisableFeatures);
    EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch1));
    EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch2));
  }

  // Enable the 2nd choice of the multi-value.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry.NameForOption(2),
                                       true);
  {
    base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
    flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                         kAddSentinels, kEnableFeatures,
                                         kDisableFeatures);
    EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch1));
    EXPECT_TRUE(command_line.HasSwitch(kMultiSwitch2));
    EXPECT_EQ(std::string(kValueForMultiSwitch2),
              command_line.GetSwitchValueASCII(kMultiSwitch2));
  }

  // Disable the multi-value entry.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry.NameForOption(0),
                                       true);
  {
    base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
    flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                         kAddSentinels, kEnableFeatures,
                                         kDisableFeatures);
    EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch1));
    EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch2));
  }
}

TEST_F(FlagsStateTest, MultiValuesWithEnableFeatures) {
  const FeatureEntry& feature_value_entry = kEntries[6];
  ASSERT_EQ(feature_value_entry.internal_name, kFlags7);
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, feature_value_entry.NameForOption(1), true);

  const FeatureEntry& multi_values_entry_1 = kEntries[12];
  ASSERT_EQ(multi_values_entry_1.internal_name, kFlags13);
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, multi_values_entry_1.NameForOption(1), true);

  const FeatureEntry& multi_values_entry_2 = kEntries[13];
  ASSERT_EQ(multi_values_entry_2.internal_name, kFlags14);
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, multi_values_entry_2.NameForOption(1), true);

  base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
  flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                       kAddSentinels, kEnableFeatures,
                                       kDisableFeatures);

  EXPECT_EQ(command_line.GetSwitchValueASCII(kEnableFeatures),
            "MultiChoiceFeature1,MultiChoiceFeature2,MultiChoiceFeature3,"
            "FeatureName1");
  EXPECT_EQ(command_line.GetSwitchValueASCII(kDisableFeatures), "");
}

TEST_F(FlagsStateTest, MultiValuesWithDisableFeatures) {
  const FeatureEntry& feature_value_entry = kEntries[6];
  ASSERT_EQ(feature_value_entry.internal_name, kFlags7);
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, feature_value_entry.NameForOption(2), true);

  const FeatureEntry& multi_values_entry_1 = kEntries[12];
  ASSERT_EQ(multi_values_entry_1.internal_name, kFlags13);
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, multi_values_entry_1.NameForOption(2), true);

  const FeatureEntry& multi_values_entry_2 = kEntries[13];
  ASSERT_EQ(multi_values_entry_2.internal_name, kFlags14);
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, multi_values_entry_2.NameForOption(2), true);

  base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
  flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                       kAddSentinels, kEnableFeatures,
                                       kDisableFeatures);

  EXPECT_EQ(command_line.GetSwitchValueASCII(kEnableFeatures), "");
  EXPECT_EQ(command_line.GetSwitchValueASCII(kDisableFeatures),
            "MultiChoiceFeature1,MultiChoiceFeature3,FeatureName1");
}

// Tests that disable flags are added when an entry is disabled.
TEST_F(FlagsStateTest, DisableFlagCommandLine) {
  // Nothing selected.
  {
    base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
    flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                         kAddSentinels, kEnableFeatures,
                                         kDisableFeatures);
    EXPECT_FALSE(command_line.HasSwitch(kSwitch6));
  }

  // Disable the entry 6.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags6, false);
  {
    base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
    flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                         kAddSentinels, kEnableFeatures,
                                         kDisableFeatures);
    EXPECT_TRUE(command_line.HasSwitch(kSwitch6));
  }

  // Enable entry 6.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, kFlags6, true);
  {
    base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
    flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                         kAddSentinels, kEnableFeatures,
                                         kDisableFeatures);
    EXPECT_FALSE(command_line.HasSwitch(kSwitch6));
  }
}

TEST_F(FlagsStateTest, EnableDisableValues) {
  const FeatureEntry& entry = kEntries[4];
  ASSERT_EQ(kFlags5, entry.internal_name);

  // Nothing selected.
  {
    base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
    flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                         kAddSentinels, kEnableFeatures,
                                         kDisableFeatures);
    EXPECT_FALSE(command_line.HasSwitch(kSwitch1));
    EXPECT_FALSE(command_line.HasSwitch(kSwitch2));
  }

  // "Enable" option selected.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry.NameForOption(1),
                                       true);
  {
    base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
    flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                         kAddSentinels, kEnableFeatures,
                                         kDisableFeatures);
    EXPECT_TRUE(command_line.HasSwitch(kSwitch1));
    EXPECT_FALSE(command_line.HasSwitch(kSwitch2));
    EXPECT_EQ(kEnableDisableValue1, command_line.GetSwitchValueASCII(kSwitch1));
  }

  // "Disable" option selected.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry.NameForOption(2),
                                       true);
  {
    base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
    flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                         kAddSentinels, kEnableFeatures,
                                         kDisableFeatures);
    EXPECT_FALSE(command_line.HasSwitch(kSwitch1));
    EXPECT_TRUE(command_line.HasSwitch(kSwitch2));
    EXPECT_EQ(kEnableDisableValue2, command_line.GetSwitchValueASCII(kSwitch2));
  }

  // "Default" option selected, same as nothing selected.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry.NameForOption(0),
                                       true);
  {
    base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
    flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                         kAddSentinels, kEnableFeatures,
                                         kDisableFeatures);
    EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch1));
    EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch2));
  }

  // "Disable" option selected, but flag filtered out by exclude predicate.
  flags_state_->SetFeatureEntryEnabled(&flags_storage_, entry.NameForOption(2),
                                       true);
  exclude_flags_.insert(entry.internal_name);
  {
    base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
    flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                         kAddSentinels, kEnableFeatures,
                                         kDisableFeatures);
    EXPECT_FALSE(command_line.HasSwitch(kSwitch1));
    EXPECT_FALSE(command_line.HasSwitch(kSwitch2));
  }
  exclude_flags_.clear();
}

TEST_F(FlagsStateTest, FeatureValues) {
  const FeatureEntry& entry = kEntries[6];
  ASSERT_EQ(kFlags7, entry.internal_name);

  struct Cases {
    int enabled_choice;
    const char* existing_enable_features;
    const char* existing_disable_features;
    const char* expected_enable_features;
    const char* expected_disable_features;
  };
  auto cases = std::to_array<Cases>({
      // Nothing selected.
      {-1, nullptr, nullptr, "", ""},
      // "Default" option selected, same as nothing selected.
      {0, nullptr, nullptr, "", ""},
      // "Enable" option selected.
      {1, nullptr, nullptr, "FeatureName1", ""},
      // "Disable" option selected.
      {2, nullptr, nullptr, "", "FeatureName1"},
      // "Enable" option should get added to the existing list.
      {1, "Foo,Bar", nullptr, "Foo,Bar,FeatureName1", ""},
      // "Disable" option should get added to the existing list.
      {2, nullptr, "Foo,Bar", "", "Foo,Bar,FeatureName1"},
  });

  for (size_t i = 0; i < std::size(cases); ++i) {
    SCOPED_TRACE(base::StringPrintf(
        "Test[%" PRIuS "]: %d [%s] [%s]", i, cases[i].enabled_choice,
        cases[i].existing_enable_features ? cases[i].existing_enable_features
                                          : "null",
        cases[i].existing_disable_features ? cases[i].existing_disable_features
                                           : "null"));

    if (cases[i].enabled_choice != -1) {
      flags_state_->SetFeatureEntryEnabled(
          &flags_storage_, entry.NameForOption(cases[i].enabled_choice), true);
    }

    base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
    if (cases[i].existing_enable_features) {
      command_line.AppendSwitchASCII(kEnableFeatures,
                                     cases[i].existing_enable_features);
    }
    if (cases[i].existing_disable_features) {
      command_line.AppendSwitchASCII(kDisableFeatures,
                                     cases[i].existing_disable_features);
    }

    flags_state_->ConvertFlagsToSwitches(&flags_storage_, &command_line,
                                         kAddSentinels, kEnableFeatures,
                                         kDisableFeatures);
    EXPECT_EQ(cases[i].expected_enable_features,
              command_line.GetSwitchValueASCII(kEnableFeatures));
    EXPECT_EQ(cases[i].expected_disable_features,
              command_line.GetSwitchValueASCII(kDisableFeatures));
  }
}

TEST_F(FlagsStateTest, GetFlagFeatureEntries) {
  base::ListValue supported_entries;
  base::ListValue unsupported_entries;
  flags_state_->GetFlagFeatureEntries(&flags_storage_, kGeneralAccessFlagsOnly,
                                      supported_entries, unsupported_entries,
                                      base::BindRepeating(&SkipFeatureEntry));
  // All |kEntries| except for |kFlags3| should be supported.
  auto supported_count = supported_entries.size();
  auto unsupported_count = unsupported_entries.size();
  EXPECT_EQ(14u, supported_count);
  EXPECT_EQ(1u, unsupported_count);
  EXPECT_EQ(std::size(kEntries), supported_count + unsupported_count);
}

#if BUILDFLAG(IS_ANDROID)
// Verify that appropriate JNI calls are made when SetFeatureEntryEnabled() is
// called. Note that when a feature is set to something other than "Default",
// SetFeatureEntryEnabled() will first make a call to itself to set the feature
// to "Default" and then continue its execution to set the feature to the
// selected value, which means that each call to SetFeatureEntryEnabled() will
// trigger two sets of JNI calls.

// Test that a FEATURE_VALUE can be correctly set to "Enabled" and "Disabled"
TEST_F(FlagsStateTest, VerifyJniCalls_1) {
  const FeatureEntry& feature1 = kEntries[6];
  ASSERT_EQ(kFlags7, feature1.internal_name);

  // Set feature1 to "Enabled"
  std::map<std::string, std::string> empty_flags;
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(empty_flags));
  std::map<std::string, std::string> flags1 = {{"FeatureName1", "true"}};
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(flags1));
  flags_state_->SetFeatureEntryEnabled(&flags_storage_,
                                       feature1.NameForOption(1), true);

  // Set feature1 to "Disabled"
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(empty_flags));
  std::map<std::string, std::string> flags2 = {{"FeatureName1", "false"}};
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(flags2));
  flags_state_->SetFeatureEntryEnabled(&flags_storage_,
                                       feature1.NameForOption(2), true);
}

// Test that a FEATURE_WITH_PARAMS_VALUE can be correctly set to
// "Enabled" and "Disabled" part 1
TEST_F(FlagsStateTest, VerifyJniCalls_2) {
  const FeatureEntry& feature_with_param1 = kEntries[9];
  ASSERT_EQ(kFlags10, feature_with_param1.internal_name);

  // Set feature_with_param1 to "Disabled"
  std::map<std::string, std::string> empty_flags;
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(empty_flags));
  std::map<std::string, std::map<std::string, std::string>> empty_params;
  EXPECT_CALL(*mock_jni_delegate_, CacheFeatureParamsImmediately(empty_params));
  std::map<std::string, std::string> flags1 = {{"FeatureName2", "false"}};
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(flags1));
  std::map<std::string, std::map<std::string, std::string>> params1 = {
      {"FeatureName2", {}}};
  EXPECT_CALL(*mock_jni_delegate_, CacheFeatureParamsImmediately(params1));
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, feature_with_param1.NameForOption(3), true);

  // Set feature_with_param1 to "Enabled dummy description 2"
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(empty_flags));
  EXPECT_CALL(*mock_jni_delegate_, CacheFeatureParamsImmediately(empty_params));
  std::map<std::string, std::string> flags2 = {{"FeatureName2", "true"}};
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(flags2));
  std::map<std::string, std::map<std::string, std::string>> params2 = {
      {"FeatureName2", {{"param2", "value"}}}};
  EXPECT_CALL(*mock_jni_delegate_, CacheFeatureParamsImmediately(params2));
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, feature_with_param1.NameForOption(2), true);
}

// Test that a FEATURE_WITH_PARAMS_VALUE can be correctly set to
// "Enabled" and "Disabled" part 2
TEST_F(FlagsStateTest, VerifyJniCalls_3) {
  const FeatureEntry& feature_with_param2 = kEntries[11];
  ASSERT_EQ(kFlags12, feature_with_param2.internal_name);

  // Set feature_with_param2 to "Enabled"
  std::map<std::string, std::string> empty_flags;
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(empty_flags));
  std::map<std::string, std::map<std::string, std::string>> empty_params;
  EXPECT_CALL(*mock_jni_delegate_, CacheFeatureParamsImmediately(empty_params));
  std::map<std::string, std::string> flags1 = {{"FeatureName3", "true"}};
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(flags1));
  std::map<std::string, std::map<std::string, std::string>> params1 = {
      {"FeatureName3", {}}};
  EXPECT_CALL(*mock_jni_delegate_, CacheFeatureParamsImmediately(params1));
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, feature_with_param2.NameForOption(1), true);

  // Set feature_with_param2 to "Enabled dummy description 1"
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(empty_flags));
  EXPECT_CALL(*mock_jni_delegate_, CacheFeatureParamsImmediately(empty_params));
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(flags1));
  std::map<std::string, std::map<std::string, std::string>> params2 = {
      {"FeatureName3", {{"param1", "value"}}}};
  EXPECT_CALL(*mock_jni_delegate_, CacheFeatureParamsImmediately(params2));
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, feature_with_param2.NameForOption(2), true);

  // Set feature_with_param2 to "Enabled dummy description 3"
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(empty_flags));
  EXPECT_CALL(*mock_jni_delegate_, CacheFeatureParamsImmediately(empty_params));
  EXPECT_CALL(*mock_jni_delegate_, CacheNativeFlagsImmediately(flags1));
  std::map<std::string, std::map<std::string, std::string>> params3 = {
      {"FeatureName3", {{"param1", "value"}, {"param:/3", "value"}}}};
  EXPECT_CALL(*mock_jni_delegate_, CacheFeatureParamsImmediately(params3));
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, feature_with_param2.NameForOption(4), true);
}

// Test that a FEATURE_VALUE can be correctly set to "Default"
TEST_F(FlagsStateTest, VerifyJniCalls_4) {
  const FeatureEntry& feature1 = kEntries[6];
  ASSERT_EQ(kFlags7, feature1.internal_name);

  // Set feature1 to "Disabled"
  flags_state_->SetFeatureEntryEnabled(&flags_storage_,
                                       feature1.NameForOption(2), true);

  // Set feature1 to "Default"
  std::vector<std::string> flags_to_erase = {"FeatureName1"};
  EXPECT_CALL(*mock_jni_delegate_, EraseNativeFlagCachedValues(flags_to_erase));
  flags_state_->SetFeatureEntryEnabled(&flags_storage_,
                                       feature1.NameForOption(0), true);
}

// Test that a FEATURE_WITH_PARAMS_VALUE can be correctly set to "Default"
TEST_F(FlagsStateTest, VerifyJniCalls_5) {
  const FeatureEntry& feature_with_param1 = kEntries[9];
  ASSERT_EQ(kFlags10, feature_with_param1.internal_name);

  // Set feature_with_param1 to "Enabled dummy description 2"
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, feature_with_param1.NameForOption(2), true);

  // Set feature_with_param1 to "Default"
  std::vector<std::string> flags_to_erase = {"FeatureName2"};
  EXPECT_CALL(*mock_jni_delegate_, EraseNativeFlagCachedValues(flags_to_erase));
  EXPECT_CALL(*mock_jni_delegate_,
              EraseFeatureParamCachedValues(flags_to_erase));
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, feature_with_param1.NameForOption(0), true);
}

// Test that ResetAllFlags() correctly resets all features to "Default"
TEST_F(FlagsStateTest, VerifyJniCalls_6) {
  const FeatureEntry& feature_with_param1 = kEntries[9];
  ASSERT_EQ(kFlags10, feature_with_param1.internal_name);

  // Set feature_with_param1 to "Enabled"
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, feature_with_param1.NameForOption(1), true);

  const FeatureEntry& feature_with_param2 = kEntries[11];
  ASSERT_EQ(kFlags12, feature_with_param2.internal_name);

  // Set feature_with_param2 to "Enabled dummy description 3"
  flags_state_->SetFeatureEntryEnabled(
      &flags_storage_, feature_with_param2.NameForOption(4), true);

  // Reset all features to "Default"
  std::vector<std::string> flags_to_erase = {"FeatureName2", "FeatureName3"};
  EXPECT_CALL(*mock_jni_delegate_, EraseNativeFlagCachedValues(flags_to_erase));
  EXPECT_CALL(*mock_jni_delegate_,
              EraseFeatureParamCachedValues(flags_to_erase));
  flags_state_->ResetAllFlags(&flags_storage_);
}
#endif

}  // namespace flags_ui