OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/auto_reset.h" | 5 #include "base/auto_reset.h" |
6 #include "base/command_line.h" | 6 #include "base/command_line.h" |
7 #include "chrome/browser/content_settings/host_content_settings_map.h" | 7 #include "chrome/browser/content_settings/host_content_settings_map.h" |
8 #include "chrome/browser/content_settings/tab_specific_content_settings.h" | 8 #include "chrome/browser/content_settings/tab_specific_content_settings.h" |
9 #include "chrome/browser/profiles/profile.h" | 9 #include "chrome/browser/profiles/profile.h" |
10 #include "chrome/browser/ui/content_settings/content_setting_bubble_model.h" | 10 #include "chrome/browser/ui/content_settings/content_setting_bubble_model.h" |
11 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h" | 11 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h" |
12 #include "chrome/browser/ui/tab_contents/test_tab_contents_wrapper.h" | 12 #include "chrome/browser/ui/tab_contents/test_tab_contents_wrapper.h" |
13 #include "chrome/common/chrome_switches.h" | 13 #include "chrome/common/chrome_switches.h" |
14 #include "chrome/test/base/testing_profile.h" | 14 #include "chrome/test/base/testing_profile.h" |
15 #include "content/browser/browser_thread.h" | 15 #include "content/browser/browser_thread.h" |
16 #include "content/browser/tab_contents/test_tab_contents.h" | 16 #include "content/browser/tab_contents/test_tab_contents.h" |
17 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
18 | 18 |
19 class ContentSettingBubbleModelTest : public TabContentsWrapperTestHarness { | 19 class ContentSettingBubbleModelTest : public TabContentsWrapperTestHarness { |
20 protected: | 20 protected: |
21 ContentSettingBubbleModelTest() | 21 ContentSettingBubbleModelTest() |
22 : ui_thread_(BrowserThread::UI, MessageLoop::current()) { | 22 : ui_thread_(BrowserThread::UI, MessageLoop::current()) { |
23 } | 23 } |
24 | 24 |
25 void CheckGeolocationBubble(size_t expected_domains, | 25 void CheckGeolocationBubble(size_t expected_domains, |
26 bool expect_clear_link, | 26 bool expect_clear_link, |
27 bool expect_reload_hint) { | 27 bool expect_reload_hint) { |
28 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( | 28 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( |
29 ContentSettingBubbleModel::CreateContentSettingBubbleModel( | 29 ContentSettingBubbleModel::CreateContentSettingBubbleModel( |
30 NULL, contents_wrapper(), profile_.get(), | 30 NULL, contents_wrapper(), profile(), |
31 CONTENT_SETTINGS_TYPE_GEOLOCATION)); | 31 CONTENT_SETTINGS_TYPE_GEOLOCATION)); |
32 const ContentSettingBubbleModel::BubbleContent& bubble_content = | 32 const ContentSettingBubbleModel::BubbleContent& bubble_content = |
33 content_setting_bubble_model->bubble_content(); | 33 content_setting_bubble_model->bubble_content(); |
34 EXPECT_TRUE(bubble_content.title.empty()); | 34 EXPECT_TRUE(bubble_content.title.empty()); |
35 EXPECT_TRUE(bubble_content.radio_group.radio_items.empty()); | 35 EXPECT_TRUE(bubble_content.radio_group.radio_items.empty()); |
36 EXPECT_TRUE(bubble_content.popup_items.empty()); | 36 EXPECT_TRUE(bubble_content.popup_items.empty()); |
37 EXPECT_EQ(expected_domains, bubble_content.domain_lists.size()); | 37 EXPECT_EQ(expected_domains, bubble_content.domain_lists.size()); |
38 EXPECT_NE(expect_clear_link || expect_reload_hint, | 38 EXPECT_NE(expect_clear_link || expect_reload_hint, |
39 bubble_content.custom_link.empty()); | 39 bubble_content.custom_link.empty()); |
40 EXPECT_EQ(expect_clear_link, bubble_content.custom_link_enabled); | 40 EXPECT_EQ(expect_clear_link, bubble_content.custom_link_enabled); |
41 EXPECT_FALSE(bubble_content.manage_link.empty()); | 41 EXPECT_FALSE(bubble_content.manage_link.empty()); |
42 } | 42 } |
43 | 43 |
44 BrowserThread ui_thread_; | 44 BrowserThread ui_thread_; |
45 }; | 45 }; |
46 | 46 |
47 TEST_F(ContentSettingBubbleModelTest, ImageRadios) { | 47 TEST_F(ContentSettingBubbleModelTest, ImageRadios) { |
48 TabSpecificContentSettings* content_settings = | 48 TabSpecificContentSettings* content_settings = |
49 contents_wrapper()->content_settings(); | 49 contents_wrapper()->content_settings(); |
50 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_IMAGES, | 50 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_IMAGES, |
51 std::string()); | 51 std::string()); |
52 | 52 |
53 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( | 53 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( |
54 ContentSettingBubbleModel::CreateContentSettingBubbleModel( | 54 ContentSettingBubbleModel::CreateContentSettingBubbleModel( |
55 NULL, contents_wrapper(), profile_.get(), | 55 NULL, contents_wrapper(), profile(), |
56 CONTENT_SETTINGS_TYPE_IMAGES)); | 56 CONTENT_SETTINGS_TYPE_IMAGES)); |
57 const ContentSettingBubbleModel::BubbleContent& bubble_content = | 57 const ContentSettingBubbleModel::BubbleContent& bubble_content = |
58 content_setting_bubble_model->bubble_content(); | 58 content_setting_bubble_model->bubble_content(); |
59 EXPECT_FALSE(bubble_content.title.empty()); | 59 EXPECT_FALSE(bubble_content.title.empty()); |
60 EXPECT_EQ(2U, bubble_content.radio_group.radio_items.size()); | 60 EXPECT_EQ(2U, bubble_content.radio_group.radio_items.size()); |
61 EXPECT_EQ(0, bubble_content.radio_group.default_item); | 61 EXPECT_EQ(0, bubble_content.radio_group.default_item); |
62 EXPECT_TRUE(bubble_content.custom_link.empty()); | 62 EXPECT_TRUE(bubble_content.custom_link.empty()); |
63 EXPECT_FALSE(bubble_content.manage_link.empty()); | 63 EXPECT_FALSE(bubble_content.manage_link.empty()); |
64 } | 64 } |
65 | 65 |
66 TEST_F(ContentSettingBubbleModelTest, Cookies) { | 66 TEST_F(ContentSettingBubbleModelTest, Cookies) { |
67 TabSpecificContentSettings* content_settings = | 67 TabSpecificContentSettings* content_settings = |
68 contents_wrapper()->content_settings(); | 68 contents_wrapper()->content_settings(); |
69 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_COOKIES, | 69 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_COOKIES, |
70 std::string()); | 70 std::string()); |
71 | 71 |
72 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( | 72 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( |
73 ContentSettingBubbleModel::CreateContentSettingBubbleModel( | 73 ContentSettingBubbleModel::CreateContentSettingBubbleModel( |
74 NULL, contents_wrapper(), profile_.get(), | 74 NULL, contents_wrapper(), profile(), |
75 CONTENT_SETTINGS_TYPE_COOKIES)); | 75 CONTENT_SETTINGS_TYPE_COOKIES)); |
76 const ContentSettingBubbleModel::BubbleContent& bubble_content = | 76 const ContentSettingBubbleModel::BubbleContent& bubble_content = |
77 content_setting_bubble_model->bubble_content(); | 77 content_setting_bubble_model->bubble_content(); |
78 EXPECT_FALSE(bubble_content.title.empty()); | 78 EXPECT_FALSE(bubble_content.title.empty()); |
79 EXPECT_FALSE(bubble_content.radio_group.radio_items.empty()); | 79 EXPECT_FALSE(bubble_content.radio_group.radio_items.empty()); |
80 EXPECT_FALSE(bubble_content.custom_link.empty()); | 80 EXPECT_FALSE(bubble_content.custom_link.empty()); |
81 EXPECT_TRUE(bubble_content.custom_link_enabled); | 81 EXPECT_TRUE(bubble_content.custom_link_enabled); |
82 EXPECT_FALSE(bubble_content.manage_link.empty()); | 82 EXPECT_FALSE(bubble_content.manage_link.empty()); |
83 } | 83 } |
84 | 84 |
85 TEST_F(ContentSettingBubbleModelTest, Plugins) { | 85 TEST_F(ContentSettingBubbleModelTest, Plugins) { |
86 TabSpecificContentSettings* content_settings = | 86 TabSpecificContentSettings* content_settings = |
87 contents_wrapper()->content_settings(); | 87 contents_wrapper()->content_settings(); |
88 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_PLUGINS, | 88 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_PLUGINS, |
89 std::string()); | 89 std::string()); |
90 | 90 |
91 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( | 91 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( |
92 ContentSettingBubbleModel::CreateContentSettingBubbleModel( | 92 ContentSettingBubbleModel::CreateContentSettingBubbleModel( |
93 NULL, contents_wrapper(), profile_.get(), | 93 NULL, contents_wrapper(), profile(), |
94 CONTENT_SETTINGS_TYPE_PLUGINS)); | 94 CONTENT_SETTINGS_TYPE_PLUGINS)); |
95 const ContentSettingBubbleModel::BubbleContent& bubble_content = | 95 const ContentSettingBubbleModel::BubbleContent& bubble_content = |
96 content_setting_bubble_model->bubble_content(); | 96 content_setting_bubble_model->bubble_content(); |
97 EXPECT_FALSE(bubble_content.title.empty()); | 97 EXPECT_FALSE(bubble_content.title.empty()); |
98 EXPECT_EQ(2U, bubble_content.radio_group.radio_items.size()); | 98 EXPECT_EQ(2U, bubble_content.radio_group.radio_items.size()); |
99 EXPECT_FALSE(bubble_content.custom_link.empty()); | 99 EXPECT_FALSE(bubble_content.custom_link.empty()); |
100 EXPECT_TRUE(bubble_content.custom_link_enabled); | 100 EXPECT_TRUE(bubble_content.custom_link_enabled); |
101 EXPECT_FALSE(bubble_content.manage_link.empty()); | 101 EXPECT_FALSE(bubble_content.manage_link.empty()); |
102 } | 102 } |
103 | 103 |
104 TEST_F(ContentSettingBubbleModelTest, MultiplePlugins) { | 104 TEST_F(ContentSettingBubbleModelTest, MultiplePlugins) { |
105 CommandLine* cmd = CommandLine::ForCurrentProcess(); | 105 CommandLine* cmd = CommandLine::ForCurrentProcess(); |
106 AutoReset<CommandLine> auto_reset(cmd, *cmd); | 106 AutoReset<CommandLine> auto_reset(cmd, *cmd); |
107 cmd->AppendSwitch(switches::kEnableResourceContentSettings); | 107 cmd->AppendSwitch(switches::kEnableResourceContentSettings); |
108 cmd->AppendSwitch(switches::kEnableClickToPlay); | 108 cmd->AppendSwitch(switches::kEnableClickToPlay); |
109 | 109 |
110 HostContentSettingsMap* map = profile_->GetHostContentSettingsMap(); | 110 HostContentSettingsMap* map = profile()->GetHostContentSettingsMap(); |
111 std::string fooPlugin = "foo"; | 111 std::string fooPlugin = "foo"; |
112 std::string barPlugin = "bar"; | 112 std::string barPlugin = "bar"; |
113 | 113 |
114 // Navigating to some sample url prevents the GetURL method from returning an | 114 // Navigating to some sample url prevents the GetURL method from returning an |
115 // invalid empty URL. | 115 // invalid empty URL. |
116 contents()->NavigateAndCommit(GURL("http://www.example.com")); | 116 contents()->NavigateAndCommit(GURL("http://www.example.com")); |
117 GURL url = contents()->GetURL(); | 117 GURL url = contents()->GetURL(); |
118 map->AddExceptionForURL(url, | 118 map->AddExceptionForURL(url, |
119 url, | 119 url, |
120 CONTENT_SETTINGS_TYPE_PLUGINS, | 120 CONTENT_SETTINGS_TYPE_PLUGINS, |
121 fooPlugin, | 121 fooPlugin, |
122 CONTENT_SETTING_ALLOW); | 122 CONTENT_SETTING_ALLOW); |
123 map->AddExceptionForURL(url, | 123 map->AddExceptionForURL(url, |
124 url, | 124 url, |
125 CONTENT_SETTINGS_TYPE_PLUGINS, | 125 CONTENT_SETTINGS_TYPE_PLUGINS, |
126 barPlugin, | 126 barPlugin, |
127 CONTENT_SETTING_ASK); | 127 CONTENT_SETTING_ASK); |
128 | 128 |
129 TabSpecificContentSettings* content_settings = | 129 TabSpecificContentSettings* content_settings = |
130 contents_wrapper()->content_settings(); | 130 contents_wrapper()->content_settings(); |
131 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_PLUGINS, | 131 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_PLUGINS, |
132 fooPlugin); | 132 fooPlugin); |
133 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_PLUGINS, | 133 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_PLUGINS, |
134 barPlugin); | 134 barPlugin); |
135 | 135 |
136 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( | 136 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( |
137 ContentSettingBubbleModel::CreateContentSettingBubbleModel( | 137 ContentSettingBubbleModel::CreateContentSettingBubbleModel( |
138 NULL, contents_wrapper(), profile_.get(), | 138 NULL, contents_wrapper(), profile(), |
139 CONTENT_SETTINGS_TYPE_PLUGINS)); | 139 CONTENT_SETTINGS_TYPE_PLUGINS)); |
140 const ContentSettingBubbleModel::BubbleContent& bubble_content = | 140 const ContentSettingBubbleModel::BubbleContent& bubble_content = |
141 content_setting_bubble_model->bubble_content(); | 141 content_setting_bubble_model->bubble_content(); |
142 EXPECT_EQ(2U, bubble_content.radio_group.radio_items.size()); | 142 EXPECT_EQ(2U, bubble_content.radio_group.radio_items.size()); |
143 EXPECT_EQ(1, bubble_content.radio_group.default_item); | 143 EXPECT_EQ(1, bubble_content.radio_group.default_item); |
144 | 144 |
145 content_setting_bubble_model->OnRadioClicked(0); | 145 content_setting_bubble_model->OnRadioClicked(0); |
146 // Nothing should have changed. | 146 // Nothing should have changed. |
147 EXPECT_EQ(CONTENT_SETTING_ALLOW, | 147 EXPECT_EQ(CONTENT_SETTING_ALLOW, |
148 map->GetContentSetting(url, | 148 map->GetContentSetting(url, |
(...skipping 28 matching lines...) Expand all Loading... |
177 NavigateAndCommit(page_url); | 177 NavigateAndCommit(page_url); |
178 TabSpecificContentSettings* content_settings = | 178 TabSpecificContentSettings* content_settings = |
179 contents_wrapper()->content_settings(); | 179 contents_wrapper()->content_settings(); |
180 | 180 |
181 // One permitted frame, but not in the content map: requires reload. | 181 // One permitted frame, but not in the content map: requires reload. |
182 content_settings->OnGeolocationPermissionSet(frame1_url, true); | 182 content_settings->OnGeolocationPermissionSet(frame1_url, true); |
183 CheckGeolocationBubble(1, false, true); | 183 CheckGeolocationBubble(1, false, true); |
184 | 184 |
185 // Add it to the content map, should now have a clear link. | 185 // Add it to the content map, should now have a clear link. |
186 HostContentSettingsMap* setting_map = | 186 HostContentSettingsMap* setting_map = |
187 profile_->GetHostContentSettingsMap(); | 187 profile()->GetHostContentSettingsMap(); |
188 setting_map->SetContentSetting( | 188 setting_map->SetContentSetting( |
189 ContentSettingsPattern::FromURLNoWildcard(frame1_url), | 189 ContentSettingsPattern::FromURLNoWildcard(frame1_url), |
190 ContentSettingsPattern::FromURLNoWildcard(page_url), | 190 ContentSettingsPattern::FromURLNoWildcard(page_url), |
191 CONTENT_SETTINGS_TYPE_GEOLOCATION, | 191 CONTENT_SETTINGS_TYPE_GEOLOCATION, |
192 std::string(), | 192 std::string(), |
193 CONTENT_SETTING_ALLOW); | 193 CONTENT_SETTING_ALLOW); |
194 CheckGeolocationBubble(1, true, false); | 194 CheckGeolocationBubble(1, true, false); |
195 | 195 |
196 // Change the default to allow: no message needed. | 196 // Change the default to allow: no message needed. |
197 profile_->GetHostContentSettingsMap()->SetDefaultContentSetting( | 197 profile()->GetHostContentSettingsMap()->SetDefaultContentSetting( |
198 CONTENT_SETTINGS_TYPE_GEOLOCATION, CONTENT_SETTING_ALLOW); | 198 CONTENT_SETTINGS_TYPE_GEOLOCATION, CONTENT_SETTING_ALLOW); |
199 CheckGeolocationBubble(1, false, false); | 199 CheckGeolocationBubble(1, false, false); |
200 | 200 |
201 // Second frame denied, but not stored in the content map: requires reload. | 201 // Second frame denied, but not stored in the content map: requires reload. |
202 content_settings->OnGeolocationPermissionSet(frame2_url, false); | 202 content_settings->OnGeolocationPermissionSet(frame2_url, false); |
203 CheckGeolocationBubble(2, false, true); | 203 CheckGeolocationBubble(2, false, true); |
204 | 204 |
205 // Change the default to block: offer a clear link for the persisted frame 1. | 205 // Change the default to block: offer a clear link for the persisted frame 1. |
206 profile_->GetHostContentSettingsMap()->SetDefaultContentSetting( | 206 profile()->GetHostContentSettingsMap()->SetDefaultContentSetting( |
207 CONTENT_SETTINGS_TYPE_GEOLOCATION, CONTENT_SETTING_BLOCK); | 207 CONTENT_SETTINGS_TYPE_GEOLOCATION, CONTENT_SETTING_BLOCK); |
208 CheckGeolocationBubble(2, true, false); | 208 CheckGeolocationBubble(2, true, false); |
209 } | 209 } |
210 | 210 |
211 TEST_F(ContentSettingBubbleModelTest, FileURL) { | 211 TEST_F(ContentSettingBubbleModelTest, FileURL) { |
212 std::string file_url("file:///tmp/test.html"); | 212 std::string file_url("file:///tmp/test.html"); |
213 NavigateAndCommit(GURL(file_url)); | 213 NavigateAndCommit(GURL(file_url)); |
214 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( | 214 scoped_ptr<ContentSettingBubbleModel> content_setting_bubble_model( |
215 ContentSettingBubbleModel::CreateContentSettingBubbleModel( | 215 ContentSettingBubbleModel::CreateContentSettingBubbleModel( |
216 NULL, contents_wrapper(), profile_.get(), | 216 NULL, contents_wrapper(), profile(), |
217 CONTENT_SETTINGS_TYPE_IMAGES)); | 217 CONTENT_SETTINGS_TYPE_IMAGES)); |
218 std::string title = | 218 std::string title = |
219 content_setting_bubble_model->bubble_content().radio_group.radio_items[0]; | 219 content_setting_bubble_model->bubble_content().radio_group.radio_items[0]; |
220 ASSERT_NE(std::string::npos, title.find(file_url)); | 220 ASSERT_NE(std::string::npos, title.find(file_url)); |
221 } | 221 } |
OLD | NEW |