| Index: chrome/renderer/content_settings_observer_browsertest.cc
|
| diff --git a/chrome/renderer/content_settings_observer_browsertest.cc b/chrome/renderer/content_settings_observer_browsertest.cc
|
| index 735501dec3d077306a40539a44da75c82a17a06f..0daa65f68fd79edeecd79f783e398ac241de5aa3 100644
|
| --- a/chrome/renderer/content_settings_observer_browsertest.cc
|
| +++ b/chrome/renderer/content_settings_observer_browsertest.cc
|
| @@ -2,6 +2,8 @@
|
| // Use of this source code is governed by a BSD-style license that can be
|
| // found in the LICENSE file.
|
|
|
| +#include <vector>
|
| +
|
| #include "chrome/common/content_settings.h"
|
| #include "chrome/common/render_messages.h"
|
| #include "chrome/renderer/content_settings_observer.h"
|
| @@ -29,11 +31,16 @@ class MockContentSettingsObserver : public ContentSettingsObserver {
|
|
|
| MOCK_METHOD5(OnAllowDOMStorage,
|
| void(int, const GURL&, const GURL&, bool, IPC::Message*));
|
| + std::vector<ContentSettingPatternSourceTuple> image_setting_rules_;
|
| + GURL image_url_;
|
| + std::string image_origin_;
|
| };
|
|
|
| MockContentSettingsObserver::MockContentSettingsObserver(
|
| content::RenderView* render_view)
|
| - : ContentSettingsObserver(render_view) {
|
| + : ContentSettingsObserver(render_view, &image_setting_rules_),
|
| + image_url_("http://www.foo.com/image.jpg"),
|
| + image_origin_("http://www.foo.com") {
|
| }
|
|
|
| bool MockContentSettingsObserver::Send(IPC::Message* message) {
|
| @@ -170,3 +177,80 @@ TEST_F(ChromeRenderViewTest, PluginsTemporarilyAllowed) {
|
| EXPECT_EQ(CONTENT_SETTING_BLOCK,
|
| observer->GetContentSetting(CONTENT_SETTINGS_TYPE_PLUGINS));
|
| }
|
| +
|
| +TEST_F(ChromeRenderViewTest, ImagesBlockedByDefault) {
|
| + MockContentSettingsObserver mock_observer(view_);
|
| +
|
| + // Load some HTML.
|
| + LoadHTML("<html>Foo</html>");
|
| +
|
| + // Set the default image blocking setting.
|
| + mock_observer.image_setting_rules_.push_back(
|
| + ContentSettingPatternSourceTuple(ContentSettingsPattern::Wildcard(),
|
| + ContentSettingsPattern::Wildcard(),
|
| + CONTENT_SETTING_BLOCK,
|
| + "",
|
| + false));
|
| +
|
| + ContentSettingsObserver* observer = ContentSettingsObserver::Get(view_);
|
| + EXPECT_CALL(mock_observer,
|
| + OnContentBlocked(CONTENT_SETTINGS_TYPE_IMAGES, std::string()));
|
| + EXPECT_FALSE(observer->AllowImage(GetMainFrame(),
|
| + true, mock_observer.image_url_));
|
| + ::testing::Mock::VerifyAndClearExpectations(&observer);
|
| +
|
| + // Create an exception which allows the image.
|
| + mock_observer.image_setting_rules_.insert(
|
| + mock_observer.image_setting_rules_.begin(),
|
| + ContentSettingPatternSourceTuple(
|
| + ContentSettingsPattern::Wildcard(),
|
| + ContentSettingsPattern::FromString(mock_observer.image_origin_),
|
| + CONTENT_SETTING_ALLOW,
|
| + "",
|
| + false));
|
| +
|
| + EXPECT_CALL(
|
| + mock_observer,
|
| + OnContentBlocked(CONTENT_SETTINGS_TYPE_IMAGES, std::string())).Times(0);
|
| + EXPECT_TRUE(observer->AllowImage(GetMainFrame(), true,
|
| + mock_observer.image_url_));
|
| + ::testing::Mock::VerifyAndClearExpectations(&observer);
|
| +}
|
| +
|
| +TEST_F(ChromeRenderViewTest, ImagesAllowedByDefault) {
|
| + MockContentSettingsObserver mock_observer(view_);
|
| +
|
| + // Load some HTML.
|
| + LoadHTML("<html>Foo</html>");
|
| +
|
| + // Set the default image blocking setting.
|
| + mock_observer.image_setting_rules_.push_back(
|
| + ContentSettingPatternSourceTuple(ContentSettingsPattern::Wildcard(),
|
| + ContentSettingsPattern::Wildcard(),
|
| + CONTENT_SETTING_ALLOW,
|
| + "",
|
| + false));
|
| +
|
| + ContentSettingsObserver* observer = ContentSettingsObserver::Get(view_);
|
| + EXPECT_CALL(
|
| + mock_observer,
|
| + OnContentBlocked(CONTENT_SETTINGS_TYPE_IMAGES, std::string())).Times(0);
|
| + EXPECT_TRUE(observer->AllowImage(GetMainFrame(), true,
|
| + mock_observer.image_url_));
|
| + ::testing::Mock::VerifyAndClearExpectations(&observer);
|
| +
|
| + // Create an exception which blocks the image.
|
| + mock_observer.image_setting_rules_.insert(
|
| + mock_observer.image_setting_rules_.begin(),
|
| + ContentSettingPatternSourceTuple(
|
| + ContentSettingsPattern::Wildcard(),
|
| + ContentSettingsPattern::FromString(mock_observer.image_origin_),
|
| + CONTENT_SETTING_BLOCK,
|
| + "",
|
| + false));
|
| + EXPECT_CALL(mock_observer,
|
| + OnContentBlocked(CONTENT_SETTINGS_TYPE_IMAGES, std::string()));
|
| + EXPECT_FALSE(observer->AllowImage(GetMainFrame(),
|
| + true, mock_observer.image_url_));
|
| + ::testing::Mock::VerifyAndClearExpectations(&observer);
|
| +}
|
|
|