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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// Copyright 2019 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/optimization_guide/core/optimization_guide_switches.h"
#include <optional>
#include "base/base64.h"
#include "base/command_line.h"
#include "build/build_config.h"
#include "components/optimization_guide/proto/hints.pb.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace optimization_guide {
namespace switches {
#if !BUILDFLAG(IS_WIN)
TEST(OptimizationGuideSwitchesTest, ParseHintsFetchOverrideFromCommandLine) {
base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(kFetchHintsOverride,
"whatever.com");
std::optional<std::vector<std::string>> parsed_hosts =
ParseHintsFetchOverrideFromCommandLine();
EXPECT_TRUE(parsed_hosts.has_value());
EXPECT_EQ(1ul, parsed_hosts.value().size());
EXPECT_EQ("whatever.com", parsed_hosts.value()[0]);
}
TEST(OptimizationGuideSwitchesTest,
ParseHintsFetchOverrideFromCommandLineMultipleHosts) {
base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(
kFetchHintsOverride, "whatever.com, whatever-2.com, ,");
std::optional<std::vector<std::string>> parsed_hosts =
ParseHintsFetchOverrideFromCommandLine();
EXPECT_TRUE(parsed_hosts.has_value());
EXPECT_EQ(2ul, parsed_hosts.value().size());
EXPECT_EQ("whatever.com", parsed_hosts.value()[0]);
EXPECT_EQ("whatever-2.com", parsed_hosts.value()[1]);
}
TEST(OptimizationGuideSwitchesTest,
ParseHintsFetchOverrideFromCommandLineNoSwitch) {
std::optional<std::vector<std::string>> parsed_hosts =
ParseHintsFetchOverrideFromCommandLine();
EXPECT_FALSE(parsed_hosts.has_value());
}
TEST(OptimizationGuideSwitchesTest, ParseComponentConfigFromCommandLine) {
optimization_guide::proto::Configuration config;
optimization_guide::proto::Hint* hint = config.add_hints();
hint->set_key("somedomain.org");
hint->set_key_representation(optimization_guide::proto::HOST);
std::string encoded_config;
config.SerializeToString(&encoded_config);
encoded_config = base::Base64Encode(encoded_config);
base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(kHintsProtoOverride,
encoded_config);
std::unique_ptr<optimization_guide::proto::Configuration> parsed_config =
ParseComponentConfigFromCommandLine();
EXPECT_EQ(1, parsed_config->hints_size());
EXPECT_EQ("somedomain.org", parsed_config->hints(0).key());
}
TEST(OptimizationGuideSwitchesTest,
ParseComponentConfigFromCommandLineNotAProto) {
base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(kHintsProtoOverride,
"not-a-proto");
std::unique_ptr<optimization_guide::proto::Configuration> parsed_config =
ParseComponentConfigFromCommandLine();
EXPECT_EQ(nullptr, parsed_config);
}
TEST(OptimizationGuideSwitchesTest,
ParseComponentConfigFromCommandLineSwitchNotSet) {
std::unique_ptr<optimization_guide::proto::Configuration> parsed_config =
ParseComponentConfigFromCommandLine();
EXPECT_EQ(nullptr, parsed_config);
}
TEST(OptimizationGuideSwitchesTest,
ParseComponentConfigFromCommandLineNotAConfiguration) {
optimization_guide::proto::HostInfo host_info;
host_info.set_host("whatever.com");
std::string encoded_proto;
host_info.SerializeToString(&encoded_proto);
encoded_proto = base::Base64Encode(encoded_proto);
base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(kHintsProtoOverride,
encoded_proto);
std::unique_ptr<optimization_guide::proto::Configuration> parsed_config =
ParseComponentConfigFromCommandLine();
EXPECT_EQ(nullptr, parsed_config);
}
#endif
} // namespace switches
} // namespace optimization_guide
|
