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
|
// 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 "chrome/browser/policy/safe_search_policy_test.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/browser/ui/location_bar/location_bar.h"
#include "chrome/common/chrome_features.h"
#include "chrome/test/base/ui_test_utils.h"
#include "components/omnibox/browser/omnibox_edit_model.h"
#include "components/policy/policy_constants.h"
#include "components/safe_search_api/safe_search_util.h"
#include "content/public/test/test_navigation_observer.h"
namespace policy {
SafeSearchPolicyTest::SafeSearchPolicyTest() {
// TODO(crbug.com/40248833): Use HTTPS URLs in tests to avoid having to
// disable this feature.
feature_list_.InitAndDisableFeature(features::kHttpsUpgrades);
}
SafeSearchPolicyTest::~SafeSearchPolicyTest() = default;
void SafeSearchPolicyTest::ApplySafeSearchPolicy(
std::optional<base::Value> legacy_safe_search,
std::optional<base::Value> google_safe_search,
std::optional<base::Value> legacy_youtube,
std::optional<base::Value> youtube_restrict) {
PolicyMap policies;
SetPolicy(&policies, key::kForceSafeSearch, std::move(legacy_safe_search));
SetPolicy(&policies, key::kForceGoogleSafeSearch,
std::move(google_safe_search));
SetPolicy(&policies, key::kForceYouTubeSafetyMode, std::move(legacy_youtube));
SetPolicy(&policies, key::kForceYouTubeRestrict, std::move(youtube_restrict));
UpdateProviderPolicy(policies);
}
// static
GURL SafeSearchPolicyTest::GetExpectedSearchURL(bool expect_safe_search) {
std::string expected_url("http://google.com/");
if (expect_safe_search) {
expected_url += "?" +
std::string(safe_search_api::kSafeSearchSafeParameter) +
"&" + safe_search_api::kSafeSearchSsuiParameter;
}
return GURL(expected_url);
}
// static
void SafeSearchPolicyTest::CheckSafeSearch(Browser* browser,
bool expect_safe_search,
const std::string& url) {
content::WebContents* web_contents =
browser->tab_strip_model()->GetActiveWebContents();
content::TestNavigationObserver observer(web_contents);
ui_test_utils::SendToOmniboxAndSubmit(browser, url);
observer.Wait();
OmniboxEditModel* model =
browser->window()->GetLocationBar()->GetOmniboxView()->model();
EXPECT_TRUE(model->CurrentMatch(nullptr).destination_url.is_valid());
EXPECT_EQ(GetExpectedSearchURL(expect_safe_search),
web_contents->GetLastCommittedURL());
}
} // namespace policy
|
