Index: chrome/renderer/content_settings_observer_browsertest.cc |
=================================================================== |
--- chrome/renderer/content_settings_observer_browsertest.cc (revision 241294) |
+++ chrome/renderer/content_settings_observer_browsertest.cc (working copy) |
@@ -19,7 +19,7 @@ |
class MockContentSettingsObserver : public ContentSettingsObserver { |
public: |
- explicit MockContentSettingsObserver(content::RenderView* render_view); |
+ explicit MockContentSettingsObserver(content::RenderFrame* render_frame); |
virtual bool Send(IPC::Message* message); |
@@ -33,8 +33,8 @@ |
}; |
MockContentSettingsObserver::MockContentSettingsObserver( |
- content::RenderView* render_view) |
- : ContentSettingsObserver(render_view, NULL), |
+ content::RenderFrame* render_frame) |
+ : ContentSettingsObserver(render_frame, NULL), |
image_url_("http://www.foo.com/image.jpg"), |
image_origin_("http://www.foo.com") { |
} |
@@ -48,13 +48,13 @@ |
IPC_END_MESSAGE_MAP() |
// Our super class deletes the message. |
- return RenderViewObserver::Send(message); |
+ return RenderFrameObserver::Send(message); |
} |
} // namespace |
TEST_F(ChromeRenderViewTest, DidBlockContentType) { |
- MockContentSettingsObserver observer(view_); |
+ MockContentSettingsObserver observer(view_->GetMainRenderFrame()); |
EXPECT_CALL(observer, |
OnContentBlocked(CONTENT_SETTINGS_TYPE_COOKIES)); |
observer.DidBlockContentType(CONTENT_SETTINGS_TYPE_COOKIES); |
@@ -69,7 +69,7 @@ |
TEST_F(ChromeRenderViewTest, DISABLED_AllowDOMStorage) { |
// Load some HTML, so we have a valid security origin. |
LoadHTML("<html></html>"); |
- MockContentSettingsObserver observer(view_); |
+ MockContentSettingsObserver observer(view_->GetMainRenderFrame()); |
ON_CALL(observer, |
OnAllowDOMStorage(_, _, _, _, _)).WillByDefault(DeleteArg<4>()); |
EXPECT_CALL(observer, |
@@ -105,7 +105,8 @@ |
CONTENT_SETTING_BLOCK, |
std::string(), |
false)); |
- ContentSettingsObserver* observer = ContentSettingsObserver::Get(view_); |
+ ContentSettingsObserver* observer = ContentSettingsObserver::Get( |
+ view_->GetMainRenderFrame()); |
observer->SetContentSettingRules(&content_setting_rules); |
// Make sure no pending messages are in the queue. |
@@ -142,12 +143,12 @@ |
std::string foo_plugin = "foo"; |
std::string bar_plugin = "bar"; |
- ContentSettingsObserver* observer = ContentSettingsObserver::Get(view_); |
+ ContentSettingsObserver* observer = |
+ ContentSettingsObserver::Get(view_->GetMainRenderFrame()); |
EXPECT_FALSE(observer->IsPluginTemporarilyAllowed(foo_plugin)); |
// Temporarily allow the "foo" plugin. |
- OnMessageReceived(ChromeViewMsg_LoadBlockedPlugins(MSG_ROUTING_NONE, |
- foo_plugin)); |
+ observer->OnLoadBlockedPlugins(foo_plugin); |
EXPECT_TRUE(observer->IsPluginTemporarilyAllowed(foo_plugin)); |
EXPECT_FALSE(observer->IsPluginTemporarilyAllowed(bar_plugin)); |
@@ -162,14 +163,13 @@ |
EXPECT_FALSE(observer->IsPluginTemporarilyAllowed(bar_plugin)); |
// Temporarily allow all plugins. |
- OnMessageReceived(ChromeViewMsg_LoadBlockedPlugins(MSG_ROUTING_NONE, |
- std::string())); |
+ observer->OnLoadBlockedPlugins(std::string()); |
EXPECT_TRUE(observer->IsPluginTemporarilyAllowed(foo_plugin)); |
EXPECT_TRUE(observer->IsPluginTemporarilyAllowed(bar_plugin)); |
} |
TEST_F(ChromeRenderViewTest, ImagesBlockedByDefault) { |
- MockContentSettingsObserver mock_observer(view_); |
+ MockContentSettingsObserver mock_observer(view_->GetMainRenderFrame()); |
// Load some HTML. |
LoadHTML("<html>Foo</html>"); |
@@ -185,7 +185,8 @@ |
std::string(), |
false)); |
- ContentSettingsObserver* observer = ContentSettingsObserver::Get(view_); |
+ ContentSettingsObserver* observer = ContentSettingsObserver::Get( |
+ view_->GetMainRenderFrame()); |
observer->SetContentSettingRules(&content_setting_rules); |
EXPECT_CALL(mock_observer, |
OnContentBlocked(CONTENT_SETTINGS_TYPE_IMAGES)); |
@@ -212,7 +213,7 @@ |
} |
TEST_F(ChromeRenderViewTest, ImagesAllowedByDefault) { |
- MockContentSettingsObserver mock_observer(view_); |
+ MockContentSettingsObserver mock_observer(view_->GetMainRenderFrame()); |
// Load some HTML. |
LoadHTML("<html>Foo</html>"); |
@@ -228,7 +229,8 @@ |
std::string(), |
false)); |
- ContentSettingsObserver* observer = ContentSettingsObserver::Get(view_); |
+ ContentSettingsObserver* observer = |
+ ContentSettingsObserver::Get(view_->GetMainRenderFrame()); |
observer->SetContentSettingRules(&content_setting_rules); |
EXPECT_CALL( |
mock_observer, |
@@ -265,7 +267,8 @@ |
std::string(), |
false)); |
- ContentSettingsObserver* observer = ContentSettingsObserver::Get(view_); |
+ ContentSettingsObserver* observer = |
+ ContentSettingsObserver::Get(view_->GetMainRenderFrame()); |
observer->SetContentSettingRules(&content_setting_rules); |
// Load a page which contains a script. |
@@ -300,7 +303,8 @@ |
std::string(), |
false)); |
- ContentSettingsObserver* observer = ContentSettingsObserver::Get(view_); |
+ ContentSettingsObserver* observer = |
+ ContentSettingsObserver::Get(view_->GetMainRenderFrame()); |
observer->SetContentSettingRules(&content_setting_rules); |
// Load a page which contains a script. |
@@ -324,7 +328,7 @@ |
} |
TEST_F(ChromeRenderViewTest, ContentSettingsInterstitialPages) { |
- MockContentSettingsObserver mock_observer(view_); |
+ MockContentSettingsObserver mock_observer(view_->GetMainRenderFrame()); |
// Block scripts. |
RendererContentSettingRules content_setting_rules; |
ContentSettingsForOneType& script_setting_rules = |
@@ -345,7 +349,8 @@ |
std::string(), |
false)); |
- ContentSettingsObserver* observer = ContentSettingsObserver::Get(view_); |
+ ContentSettingsObserver* observer = |
+ ContentSettingsObserver::Get(view_->GetMainRenderFrame()); |
observer->SetContentSettingRules(&content_setting_rules); |
observer->OnSetAsInterstitial(); |