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
|
// Copyright (c) 2011 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 "chrome/common/content_settings.h"
#include "chrome/common/render_messages.h"
#include "chrome/renderer/content_settings_observer.h"
#include "chrome/test/render_view_test.h"
#include "content/common/view_messages.h"
#include "ipc/ipc_message_macros.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
class MockContentSettingsObserver : public ContentSettingsObserver {
public:
explicit MockContentSettingsObserver(RenderView* render_view);
virtual bool Send(IPC::Message* message);
MOCK_METHOD2(OnContentBlocked,
void(ContentSettingsType, const std::string&));
};
MockContentSettingsObserver::MockContentSettingsObserver(
RenderView* render_view)
: ContentSettingsObserver(render_view) {
}
bool MockContentSettingsObserver::Send(IPC::Message* message) {
IPC_BEGIN_MESSAGE_MAP(MockContentSettingsObserver, *message)
IPC_MESSAGE_HANDLER(ViewHostMsg_ContentBlocked, OnContentBlocked)
IPC_MESSAGE_UNHANDLED(ADD_FAILURE())
IPC_END_MESSAGE_MAP()
// Our super class deletes the message.
return RenderViewObserver::Send(message);
}
} // namespace
TEST_F(RenderViewTest, DidBlockContentType) {
MockContentSettingsObserver observer(view_);
EXPECT_CALL(observer,
OnContentBlocked(CONTENT_SETTINGS_TYPE_COOKIES, std::string()));
observer.DidBlockContentType(CONTENT_SETTINGS_TYPE_COOKIES, std::string());
// Blocking the same content type a second time shouldn't send a notification.
observer.DidBlockContentType(CONTENT_SETTINGS_TYPE_COOKIES, std::string());
::testing::Mock::VerifyAndClearExpectations(&observer);
// Blocking two different plugins should send two notifications.
std::string kFooPlugin = "foo";
std::string kBarPlugin = "bar";
EXPECT_CALL(observer,
OnContentBlocked(CONTENT_SETTINGS_TYPE_PLUGINS, kFooPlugin));
EXPECT_CALL(observer,
OnContentBlocked(CONTENT_SETTINGS_TYPE_PLUGINS, kBarPlugin));
observer.DidBlockContentType(CONTENT_SETTINGS_TYPE_PLUGINS, kFooPlugin);
observer.DidBlockContentType(CONTENT_SETTINGS_TYPE_PLUGINS, kBarPlugin);
}
// Regression test for http://crbug.com/35011
TEST_F(RenderViewTest, JSBlockSentAfterPageLoad) {
// 1. Load page with JS.
std::string html = "<html>"
"<head>"
"<script>document.createElement('div');</script>"
"</head>"
"<body>"
"</body>"
"</html>";
render_thread_.sink().ClearMessages();
LoadHTML(html.c_str());
// 2. Block JavaScript.
ContentSettings settings;
for (int i = 0; i < CONTENT_SETTINGS_NUM_TYPES; ++i)
settings.settings[i] = CONTENT_SETTING_ALLOW;
settings.settings[CONTENT_SETTINGS_TYPE_JAVASCRIPT] = CONTENT_SETTING_BLOCK;
ContentSettingsObserver* observer = ContentSettingsObserver::Get(view_);
observer->SetContentSettings(settings);
// Make sure no pending messages are in the queue.
ProcessPendingMessages();
render_thread_.sink().ClearMessages();
// 3. Reload page.
ViewMsg_Navigate_Params params;
std::string url_str = "data:text/html;charset=utf-8,";
url_str.append(html);
GURL url(url_str);
params.url = url;
params.navigation_type = ViewMsg_Navigate_Type::RELOAD;
view_->OnNavigate(params);
ProcessPendingMessages();
// 4. Verify that the notification that javascript was blocked is sent after
// the navigation notifiction is sent.
int navigation_index = -1;
int block_index = -1;
for (size_t i = 0; i < render_thread_.sink().message_count(); ++i) {
const IPC::Message* msg = render_thread_.sink().GetMessageAt(i);
if (msg->type() == ViewHostMsg_FrameNavigate::ID)
navigation_index = i;
if (msg->type() == ViewHostMsg_ContentBlocked::ID)
block_index = i;
}
EXPECT_NE(-1, navigation_index);
EXPECT_NE(-1, block_index);
EXPECT_LT(navigation_index, block_index);
}
|
