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
|
// 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/scoped_feature_list.h"
#include "chrome/browser/enterprise/watermark/watermark_view.h"
#include "chrome/browser/ui/test/test_browser_ui.h"
#include "chrome/browser/ui/views/frame/browser_view.h"
#include "chrome/common/chrome_features.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/ui_test_utils.h"
#include "content/public/test/browser_test.h"
#include "net/dns/mock_host_resolver.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace enterprise_watermark {
namespace {
// This string checks that non-latin characters render correctly.
constexpr char kMultilingualWatermarkMessage[] = R"(
THIS IS CONFIDENTIAL!
😀😀😀 草草草 www
مضحك جداً
)";
// This string checks that long lines are properly handled by multiline logic.
constexpr char kLongLinesWatermarkMessage[] = R"(
This is a very long line that should be split up into multiple lines
This is a shorter line
It was not split
This is another very long line that should be split up into multiple lines
)";
class WatermarkBrowserTest : public UiBrowserTest,
public testing::WithParamInterface<const char*> {
public:
void SetUpOnMainThread() override {
host_resolver()->AddRule("*", "127.0.0.1");
ASSERT_TRUE(embedded_test_server()->Start());
}
void NavigateToTestPage() {
ASSERT_TRUE(ui_test_utils::NavigateToURL(
browser(), embedded_test_server()->GetURL(
"/enterprise/watermark/watermark_test_page.html")));
}
// Returns true if a watermark view object was available to set the watermark.
bool SetWatermark(const std::string& watermark_message) {
if (auto* watermark_view = BrowserView::GetBrowserViewForBrowser(browser())
->get_watermark_view_for_testing()) {
watermark_view->SetString(watermark_message);
return true;
}
return false;
}
void ShowUi(const std::string& name) override {
base::RunLoop().RunUntilIdle();
}
bool VerifyUi() override {
const auto* const test_info =
testing::UnitTest::GetInstance()->current_test_info();
return VerifyPixelUi(BrowserView::GetBrowserViewForBrowser(browser())
->contents_container(),
test_info->test_suite_name(),
test_info->name()) != ui::test::ActionResult::kFailed;
}
void WaitForUserDismissal() override {}
protected:
base::test::ScopedFeatureList scoped_features_;
};
} // namespace
// The test is enabled only for Windows since this is the standard for pixel
// golden tests in general (single platform to account for variability in
// rendering).
#if BUILDFLAG(IS_WIN)
#define MAYBE_WatermarkShownAfterNavigation WatermarkShownAfterNavigation
#else
#define MAYBE_WatermarkShownAfterNavigation WatermarkShownAfterNavigation
#endif
IN_PROC_BROWSER_TEST_P(WatermarkBrowserTest,
MAYBE_WatermarkShownAfterNavigation) {
NavigateToTestPage();
ASSERT_TRUE(SetWatermark(GetParam()));
ShowAndVerifyUi();
}
IN_PROC_BROWSER_TEST_P(WatermarkBrowserTest, WatermarkClearedAfterNavigation) {
ASSERT_TRUE(SetWatermark(GetParam()));
// Navigating away from a watermarked page should clear the watermark if no
// other verdict/policy is present to show a watermark.
NavigateToTestPage();
ShowAndVerifyUi();
}
INSTANTIATE_TEST_SUITE_P(All,
WatermarkBrowserTest,
testing::Values(kMultilingualWatermarkMessage,
kLongLinesWatermarkMessage));
} // namespace enterprise_watermark
|
