| OLD | NEW |
|---|
| (Empty) |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "chrome/browser/content_setting_bubble_model.h" | |
| 6 | |
| 7 #include "base/auto_reset.h" | |
| 8 #include "base/command_line.h" | |
| 9 #include "chrome/browser/content_settings/host_content_settings_map.h" | |
| 10 #include "chrome/browser/profiles/profile.h" | |
| 11 #include "chrome/browser/geolocation/geolocation_content_settings_map.h" | |
| 12 #include "chrome/common/chrome_switches.h" | |
| 13 #include "chrome/test/testing_profile.h" | |
| 14 #include "content/browser/browser_thread.h" | |
| 15 #include "content/browser/renderer_host/test_render_view_host.h" | |
| 16 #include "content/browser/tab_contents/test_tab_contents.h" | |
| 17 #include "testing/gtest/include/gtest/gtest.h" | |
| 18 | |
| 19 class ContentSettingBubbleModelTest : public RenderViewHostTestHarness { | |
| 20 protected: | |
| 21 ContentSettingBubbleModelTest() | |
| 22 : ui_thread_(BrowserThread::UI, MessageLoop::current()) { | |
| 23 } | |
| 24 | |
| 25 void CheckGeolocationBubble(size_t expected_domains, | |
| 26 bool expect_clear_link, | |
| 27 bool expect_reload_hint) { | |
| 28 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( | |
| 29 ContentSettingBubbleModel::CreateContentSettingBubbleModel( | |
| 30 contents(), profile_.get(), CONTENT_SETTINGS_TYPE_GEOLOCATION)); | |
| 31 const ContentSettingBubbleModel::BubbleContent& bubble_content = | |
| 32 content_setting_bubble_model->bubble_content(); | |
| 33 EXPECT_TRUE(bubble_content.title.empty()); | |
| 34 EXPECT_TRUE(bubble_content.radio_group.radio_items.empty()); | |
| 35 EXPECT_TRUE(bubble_content.popup_items.empty()); | |
| 36 EXPECT_EQ(expected_domains, bubble_content.domain_lists.size()); | |
| 37 EXPECT_NE(expect_clear_link || expect_reload_hint, | |
| 38 bubble_content.custom_link.empty()); | |
| 39 EXPECT_EQ(expect_clear_link, bubble_content.custom_link_enabled); | |
| 40 EXPECT_FALSE(bubble_content.manage_link.empty()); | |
| 41 } | |
| 42 | |
| 43 BrowserThread ui_thread_; | |
| 44 }; | |
| 45 | |
| 46 TEST_F(ContentSettingBubbleModelTest, ImageRadios) { | |
| 47 TabSpecificContentSettings* content_settings = | |
| 48 contents()->GetTabSpecificContentSettings(); | |
| 49 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_IMAGES, | |
| 50 std::string()); | |
| 51 | |
| 52 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( | |
| 53 ContentSettingBubbleModel::CreateContentSettingBubbleModel( | |
| 54 contents(), profile_.get(), CONTENT_SETTINGS_TYPE_IMAGES)); | |
| 55 const ContentSettingBubbleModel::BubbleContent& bubble_content = | |
| 56 content_setting_bubble_model->bubble_content(); | |
| 57 EXPECT_FALSE(bubble_content.title.empty()); | |
| 58 EXPECT_EQ(2U, bubble_content.radio_group.radio_items.size()); | |
| 59 EXPECT_EQ(0, bubble_content.radio_group.default_item); | |
| 60 EXPECT_TRUE(bubble_content.custom_link.empty()); | |
| 61 EXPECT_FALSE(bubble_content.manage_link.empty()); | |
| 62 } | |
| 63 | |
| 64 TEST_F(ContentSettingBubbleModelTest, Cookies) { | |
| 65 TabSpecificContentSettings* content_settings = | |
| 66 contents()->GetTabSpecificContentSettings(); | |
| 67 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_COOKIES, | |
| 68 std::string()); | |
| 69 | |
| 70 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( | |
| 71 ContentSettingBubbleModel::CreateContentSettingBubbleModel( | |
| 72 contents(), profile_.get(), CONTENT_SETTINGS_TYPE_COOKIES)); | |
| 73 const ContentSettingBubbleModel::BubbleContent& bubble_content = | |
| 74 content_setting_bubble_model->bubble_content(); | |
| 75 EXPECT_FALSE(bubble_content.title.empty()); | |
| 76 EXPECT_FALSE(bubble_content.radio_group.radio_items.empty()); | |
| 77 EXPECT_FALSE(bubble_content.custom_link.empty()); | |
| 78 EXPECT_TRUE(bubble_content.custom_link_enabled); | |
| 79 EXPECT_FALSE(bubble_content.manage_link.empty()); | |
| 80 } | |
| 81 | |
| 82 TEST_F(ContentSettingBubbleModelTest, Plugins) { | |
| 83 TabSpecificContentSettings* content_settings = | |
| 84 contents()->GetTabSpecificContentSettings(); | |
| 85 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_PLUGINS, | |
| 86 std::string()); | |
| 87 | |
| 88 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( | |
| 89 ContentSettingBubbleModel::CreateContentSettingBubbleModel( | |
| 90 contents(), profile_.get(), CONTENT_SETTINGS_TYPE_PLUGINS)); | |
| 91 const ContentSettingBubbleModel::BubbleContent& bubble_content = | |
| 92 content_setting_bubble_model->bubble_content(); | |
| 93 EXPECT_FALSE(bubble_content.title.empty()); | |
| 94 EXPECT_EQ(2U, bubble_content.radio_group.radio_items.size()); | |
| 95 EXPECT_FALSE(bubble_content.custom_link.empty()); | |
| 96 EXPECT_TRUE(bubble_content.custom_link_enabled); | |
| 97 EXPECT_FALSE(bubble_content.manage_link.empty()); | |
| 98 } | |
| 99 | |
| 100 TEST_F(ContentSettingBubbleModelTest, MultiplePlugins) { | |
| 101 CommandLine* cmd = CommandLine::ForCurrentProcess(); | |
| 102 AutoReset<CommandLine> auto_reset(cmd, *cmd); | |
| 103 cmd->AppendSwitch(switches::kEnableResourceContentSettings); | |
| 104 cmd->AppendSwitch(switches::kEnableClickToPlay); | |
| 105 | |
| 106 HostContentSettingsMap* map = profile_->GetHostContentSettingsMap(); | |
| 107 std::string fooPlugin = "foo"; | |
| 108 std::string barPlugin = "bar"; | |
| 109 GURL url = contents()->GetURL(); | |
| 110 map->AddExceptionForURL(url, | |
| 111 CONTENT_SETTINGS_TYPE_PLUGINS, | |
| 112 fooPlugin, | |
| 113 CONTENT_SETTING_ALLOW); | |
| 114 map->AddExceptionForURL(url, | |
| 115 CONTENT_SETTINGS_TYPE_PLUGINS, | |
| 116 barPlugin, | |
| 117 CONTENT_SETTING_ASK); | |
| 118 | |
| 119 TabSpecificContentSettings* content_settings = | |
| 120 contents()->GetTabSpecificContentSettings(); | |
| 121 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_PLUGINS, | |
| 122 fooPlugin); | |
| 123 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_PLUGINS, | |
| 124 barPlugin); | |
| 125 | |
| 126 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( | |
| 127 ContentSettingBubbleModel::CreateContentSettingBubbleModel( | |
| 128 contents(), profile_.get(), CONTENT_SETTINGS_TYPE_PLUGINS)); | |
| 129 const ContentSettingBubbleModel::BubbleContent& bubble_content = | |
| 130 content_setting_bubble_model->bubble_content(); | |
| 131 EXPECT_EQ(2U, bubble_content.radio_group.radio_items.size()); | |
| 132 EXPECT_EQ(1, bubble_content.radio_group.default_item); | |
| 133 | |
| 134 content_setting_bubble_model->OnRadioClicked(0); | |
| 135 // Nothing should have changed. | |
| 136 EXPECT_EQ(CONTENT_SETTING_ALLOW, | |
| 137 map->GetContentSetting(url, | |
| 138 CONTENT_SETTINGS_TYPE_PLUGINS, | |
| 139 fooPlugin)); | |
| 140 EXPECT_EQ(CONTENT_SETTING_ASK, | |
| 141 map->GetContentSetting(url, | |
| 142 CONTENT_SETTINGS_TYPE_PLUGINS, | |
| 143 barPlugin)); | |
| 144 | |
| 145 content_setting_bubble_model.reset(); | |
| 146 // Both plug-ins should be click-to-play now. | |
| 147 EXPECT_EQ(CONTENT_SETTING_ALLOW, | |
| 148 map->GetContentSetting(url, | |
| 149 CONTENT_SETTINGS_TYPE_PLUGINS, | |
| 150 fooPlugin)); | |
| 151 EXPECT_EQ(CONTENT_SETTING_ALLOW, | |
| 152 map->GetContentSetting(url, | |
| 153 CONTENT_SETTINGS_TYPE_PLUGINS, | |
| 154 barPlugin)); | |
| 155 } | |
| 156 | |
| 157 TEST_F(ContentSettingBubbleModelTest, Geolocation) { | |
| 158 const GURL page_url("http://toplevel.example/"); | |
| 159 const GURL frame1_url("http://host1.example/"); | |
| 160 const GURL frame2_url("http://host2.example:999/"); | |
| 161 | |
| 162 NavigateAndCommit(page_url); | |
| 163 TabSpecificContentSettings* content_settings = | |
| 164 contents()->GetTabSpecificContentSettings(); | |
| 165 | |
| 166 // One permitted frame, but not in the content map: requires reload. | |
| 167 content_settings->OnGeolocationPermissionSet(frame1_url, true); | |
| 168 CheckGeolocationBubble(1, false, true); | |
| 169 | |
| 170 // Add it to the content map, should now have a clear link. | |
| 171 GeolocationContentSettingsMap* setting_map = | |
| 172 profile_->GetGeolocationContentSettingsMap(); | |
| 173 setting_map->SetContentSetting(frame1_url, page_url, CONTENT_SETTING_ALLOW); | |
| 174 CheckGeolocationBubble(1, true, false); | |
| 175 | |
| 176 // Change the default to allow: no message needed. | |
| 177 setting_map->SetDefaultContentSetting(CONTENT_SETTING_ALLOW); | |
| 178 CheckGeolocationBubble(1, false, false); | |
| 179 | |
| 180 // Second frame denied, but not stored in the content map: requires reload. | |
| 181 content_settings->OnGeolocationPermissionSet(frame2_url, false); | |
| 182 CheckGeolocationBubble(2, false, true); | |
| 183 | |
| 184 // Change the default to block: offer a clear link for the persisted frame 1. | |
| 185 setting_map->SetDefaultContentSetting(CONTENT_SETTING_BLOCK); | |
| 186 CheckGeolocationBubble(2, true, false); | |
| 187 } | |
| 188 | |
| 189 TEST_F(ContentSettingBubbleModelTest, FileURL) { | |
| 190 std::string file_url("file:///tmp/test.html"); | |
| 191 NavigateAndCommit(GURL(file_url)); | |
| 192 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( | |
| 193 ContentSettingBubbleModel::CreateContentSettingBubbleModel( | |
| 194 contents(), profile_.get(), CONTENT_SETTINGS_TYPE_IMAGES)); | |
| 195 std::string title = | |
| 196 content_setting_bubble_model->bubble_content().radio_group.radio_items[0]; | |
| 197 ASSERT_NE(std::string::npos, title.find(file_url)); | |
| 198 } | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 6854035:
Move blocked content from TabContents to TabContentsWrapper. (Closed)
Base URL: svn://svn.chromium.org/chrome/trunk/src