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
117
118
119
|
// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/bind.h"
#include "base/run_loop.h"
#include "components/web_restrictions/browser/mock_web_restrictions_client.h"
#include "components/web_restrictions/browser/web_restrictions_client.h"
#include "content/public/test/test_browser_thread_bundle.h"
#include "testing/gtest/include/gtest/gtest.h"
using web_restrictions::WebRestrictionsClient;
using web_restrictions::MockWebRestrictionsClient;
namespace {
bool g_returned_result;
void ResultCallback(const base::Closure& quit_closure, bool result) {
g_returned_result = result;
quit_closure.Run();
}
} // namespace
class WebRestrictionsClientTest : public testing::Test {
protected:
// Mock the Java WebRestrictionsClient. The real version
// would need a content provider to do anything.
MockWebRestrictionsClient mock_;
content::TestBrowserThreadBundle thread_bundle_;
WebRestrictionsClient client_;
};
TEST_F(WebRestrictionsClientTest, ShouldProceed) {
client_.SetAuthority("Good");
// First call should go to Web Restrictions Content Provider, and return a
// delayed result.
{
g_returned_result = false;
base::RunLoop run_loop;
EXPECT_EQ(web_restrictions::PENDING,
client_.ShouldProceed(
true, GURL("http://example.com"),
base::Bind(&ResultCallback, run_loop.QuitClosure())));
run_loop.Run();
EXPECT_TRUE(g_returned_result);
}
// A repeated call should go to the cache and return a result immediately.
{
base::RunLoop run_loop;
EXPECT_EQ(web_restrictions::ALLOW,
client_.ShouldProceed(
true, GURL("http://example.com"),
base::Bind(&ResultCallback, run_loop.QuitClosure())));
}
// However a different url should miss the cache
{
g_returned_result = false;
base::RunLoop run_loop;
EXPECT_EQ(web_restrictions::PENDING,
client_.ShouldProceed(
true, GURL("http://example.com/2"),
base::Bind(&ResultCallback, run_loop.QuitClosure())));
run_loop.Run();
EXPECT_TRUE(g_returned_result);
}
// Switching the authority should clear the cache.
{
client_.SetAuthority("Good2");
g_returned_result = false;
base::RunLoop run_loop;
EXPECT_EQ(web_restrictions::PENDING,
client_.ShouldProceed(
true, GURL("http://example.com/2"),
base::Bind(&ResultCallback, run_loop.QuitClosure())));
run_loop.Run();
EXPECT_TRUE(g_returned_result);
}
// Try getting a bad result
{
client_.SetAuthority("Bad");
g_returned_result = true;
base::RunLoop run_loop;
EXPECT_EQ(web_restrictions::PENDING,
client_.ShouldProceed(
true, GURL("http://example.com/2"),
base::Bind(&ResultCallback, run_loop.QuitClosure())));
run_loop.Run();
EXPECT_FALSE(g_returned_result);
std::string error_html;
EXPECT_TRUE(
client_.GetErrorHtml(GURL("http://example.com/2"), &error_html));
EXPECT_EQ("http://example.com/2", error_html);
}
}
TEST_F(WebRestrictionsClientTest, RequestPermission) {
client_.SetAuthority("Good");
{
base::RunLoop run_loop;
g_returned_result = false;
client_.RequestPermission(
GURL("http://example.com"),
base::Bind(&ResultCallback, run_loop.QuitClosure()));
run_loop.Run();
EXPECT_TRUE(g_returned_result);
client_.SetAuthority("Bad");
}
{
base::RunLoop run_loop;
g_returned_result = true;
client_.RequestPermission(
GURL("http://example.com"),
base::Bind(&ResultCallback, run_loop.QuitClosure()));
run_loop.Run();
EXPECT_FALSE(g_returned_result);
}
}
|
