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
|
// 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
|
