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
115
116
|
// Copyright 2021 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/functional/bind.h"
#include "base/test/scoped_feature_list.h"
#include "build/build_config.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/chrome_content_browser_client.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/common/pref_names.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/scoped_testing_local_state.h"
#include "chrome/test/base/testing_browser_process.h"
#include "chrome/test/base/ui_test_utils.h"
#include "components/embedder_support/user_agent_utils.h"
#include "components/prefs/pref_service.h"
#include "content/public/test/browser_test.h"
#include "content/public/test/browser_test_utils.h"
#include "net/http/http_request_headers.h"
#include "third_party/blink/public/common/features.h"
namespace policy {
using ReductionPolicyState =
embedder_support::UserAgentReductionEnterprisePolicyState;
class UserAgentBrowserTest : public InProcessBrowserTest {
public:
UserAgentBrowserTest() {
embedded_test_server()->RegisterRequestMonitor(base::BindRepeating(
&UserAgentBrowserTest::MonitorUserAgent, base::Unretained(this)));
}
void SetUp() override {
ASSERT_TRUE(embedded_test_server()->Start());
empty_url_ = embedded_test_server()->GetURL("/empty.html");
InProcessBrowserTest::SetUp();
}
void set_user_agent_reduction_policy(ReductionPolicyState policy) {
browser()->profile()->GetPrefs()->SetInteger(prefs::kUserAgentReduction,
static_cast<int>(policy));
}
ReductionPolicyState user_agent_reduction_policy() {
return static_cast<ReductionPolicyState>(
browser()->profile()->GetPrefs()->GetInteger(
prefs::kUserAgentReduction));
}
std::string observed_user_agent() { return observered_user_agent_; }
GURL empty_url() { return empty_url_; }
private:
void MonitorUserAgent(const net::test_server::HttpRequest& request) {
if (request.GetURL() == empty_url()) {
observered_user_agent_ =
request.headers.find(net::HttpRequestHeaders::kUserAgent)->second;
}
}
std::string observered_user_agent_;
base::test::ScopedFeatureList scoped_feature_list_;
GURL empty_url_;
};
IN_PROC_BROWSER_TEST_F(UserAgentBrowserTest, EnterprisePolicyInitialized) {
// Check that default is set correctly
EXPECT_EQ(ReductionPolicyState::kDefault, user_agent_reduction_policy());
}
IN_PROC_BROWSER_TEST_F(UserAgentBrowserTest, ReductionPolicyDefault) {
set_user_agent_reduction_policy(ReductionPolicyState::kDefault);
ASSERT_TRUE(ui_test_utils::NavigateToURL(browser(), empty_url()));
EXPECT_EQ(observed_user_agent(), embedder_support::GetUserAgent());
}
class ReduceUserAgentPlatformBrowserTest : public InProcessBrowserTest {
public:
ReduceUserAgentPlatformBrowserTest() = default;
void SetUp() override {
std::unique_ptr<base::FeatureList> feature_list(new base::FeatureList);
feature_list->InitFromCommandLine(
"ReduceUserAgentMinorVersion,ReduceUserAgentPlatformOsCpu", "");
scoped_feature_list_.InitWithFeatureList(std::move(feature_list));
InProcessBrowserTest::SetUp();
}
content::WebContents* web_contents() const {
return browser()->tab_strip_model()->GetActiveWebContents();
}
private:
base::test::ScopedFeatureList scoped_feature_list_;
};
IN_PROC_BROWSER_TEST_F(ReduceUserAgentPlatformBrowserTest, NavigatorPlatform) {
#if BUILDFLAG(IS_ANDROID)
EXPECT_EQ("Linux armv81",
content::EvalJs(web_contents(), "navigator.platform"));
#elif BUILDFLAG(IS_MAC)
EXPECT_EQ("MacIntel", content::EvalJs(web_contents(), "navigator.platform"));
#elif BUILDFLAG(IS_WIN)
EXPECT_EQ("Win32", content::EvalJs(web_contents(), "navigator.platform"));
#else
EXPECT_EQ("Linux x86_64",
content::EvalJs(web_contents(), "navigator.platform"));
#endif
}
} // namespace policy
|
