1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50


base / test / test_suite_helper.cc [blame]

// 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 "base/test/test_suite_helper.h"

#include "base/base_switches.h"
#include "base/command_line.h"
#include "base/test/scoped_feature_list.h"

namespace base::test {

void InitScopedFeatureListForTesting(ScopedFeatureList& scoped_feature_list) {
  const CommandLine* command_line = CommandLine::ForCurrentProcess();

  // We set up a FeatureList via ScopedFeatureList::InitFromCommandLine().
  // This ensures that code using that API will not hit an error that it's
  // not set. It will be cleared by ~ScopedFeatureList().

  // TestFeatureForBrowserTest1 and TestFeatureForBrowserTest2 used in
  // ContentBrowserTestScopedFeatureListTest to ensure ScopedFeatureList keeps
  // features from command line.
  // TestBlinkFeatureDefault is used in RuntimeEnabledFeaturesTest to test a
  // behavior with OverrideState::OVERIDE_USE_DEFAULT.
  std::string enabled =
      command_line->GetSwitchValueASCII(switches::kEnableFeatures);
  std::string disabled =
      command_line->GetSwitchValueASCII(switches::kDisableFeatures);
  enabled += ",TestFeatureForBrowserTest1,*TestBlinkFeatureDefault";
  disabled += ",TestFeatureForBrowserTest2";
  scoped_feature_list.InitFromCommandLine(enabled, disabled);

  // The enable-features and disable-features flags were just slurped into a
  // FeatureList, so remove them from the command line. Tests should enable
  // and disable features via the ScopedFeatureList API rather than
  // command-line flags.
  CommandLine new_command_line(command_line->GetProgram());
  CommandLine::SwitchMap switches = command_line->GetSwitches();

  switches.erase(switches::kEnableFeatures);
  switches.erase(switches::kDisableFeatures);

  for (const auto& iter : switches) {
    new_command_line.AppendSwitchNative(iter.first, iter.second);
  }

  *CommandLine::ForCurrentProcess() = new_command_line;
}

}  // namespace base::test