| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "chrome/browser/permissions/permission_context_base.h" | 5 #include "chrome/browser/permissions/permission_context_base.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/command_line.h" | 8 #include "base/command_line.h" |
| 9 #include "chrome/browser/content_settings/host_content_settings_map_factory.h" | 9 #include "chrome/browser/content_settings/host_content_settings_map_factory.h" |
| 10 #include "chrome/browser/infobars/infobar_service.h" | 10 #include "chrome/browser/infobars/infobar_service.h" |
| 11 #include "chrome/browser/permissions/permission_queue_controller.h" | 11 #include "chrome/browser/permissions/permission_infobar_manager.h" |
| 12 #include "chrome/browser/permissions/permission_request_id.h" | 12 #include "chrome/browser/permissions/permission_request_id.h" |
| 13 #include "chrome/browser/ui/website_settings/permission_bubble_manager.h" | 13 #include "chrome/browser/ui/website_settings/permission_bubble_manager.h" |
| 14 #include "chrome/common/chrome_switches.h" | 14 #include "chrome/common/chrome_switches.h" |
| 15 #include "chrome/test/base/chrome_render_view_host_test_harness.h" | 15 #include "chrome/test/base/chrome_render_view_host_test_harness.h" |
| 16 #include "chrome/test/base/testing_profile.h" | 16 #include "chrome/test/base/testing_profile.h" |
| 17 #include "components/content_settings/core/browser/host_content_settings_map.h" | 17 #include "components/content_settings/core/browser/host_content_settings_map.h" |
| 18 #include "components/content_settings/core/common/content_settings.h" | 18 #include "components/content_settings/core/common/content_settings.h" |
| 19 #include "components/content_settings/core/common/content_settings_types.h" | 19 #include "components/content_settings/core/common/content_settings_types.h" |
| 20 #include "content/public/browser/render_frame_host.h" | 20 #include "content/public/browser/render_frame_host.h" |
| 21 #include "content/public/browser/web_contents.h" | 21 #include "content/public/browser/web_contents.h" |
| 22 #include "content/public/test/mock_render_process_host.h" | 22 #include "content/public/test/mock_render_process_host.h" |
| 23 #include "content/public/test/web_contents_tester.h" | 23 #include "content/public/test/web_contents_tester.h" |
| 24 #include "testing/gtest/include/gtest/gtest.h" | 24 #include "testing/gtest/include/gtest/gtest.h" |
| 25 | 25 |
| 26 class TestPermissionContext : public PermissionContextBase { | 26 class TestPermissionContext : public PermissionContextBase { |
| 27 public: | 27 public: |
| 28 TestPermissionContext(Profile* profile, | 28 TestPermissionContext(Profile* profile, |
| 29 const ContentSettingsType permission_type) | 29 const ContentSettingsType permission_type) |
| 30 : PermissionContextBase(profile, permission_type), | 30 : PermissionContextBase(profile, permission_type), |
| 31 permission_set_(false), | 31 permission_set_(false), |
| 32 permission_granted_(false), | 32 permission_granted_(false), |
| 33 tab_context_updated_(false) {} | 33 tab_context_updated_(false) {} |
| 34 | 34 |
| 35 ~TestPermissionContext() override {} | 35 ~TestPermissionContext() override {} |
| 36 | 36 |
| 37 PermissionQueueController* GetInfoBarController() { | |
| 38 return GetQueueController(); | |
| 39 } | |
| 40 | |
| 41 bool permission_granted() { | 37 bool permission_granted() { |
| 42 return permission_granted_; | 38 return permission_granted_; |
| 43 } | 39 } |
| 44 | 40 |
| 45 bool permission_set() { | 41 bool permission_set() { |
| 46 return permission_set_; | 42 return permission_set_; |
| 47 } | 43 } |
| 48 | 44 |
| 49 bool tab_context_updated() { | 45 bool tab_context_updated() { |
| 50 return tab_context_updated_; | 46 return tab_context_updated_; |
| (...skipping 19 matching lines...) Expand all Loading... |
| 70 bool permission_set_; | 66 bool permission_set_; |
| 71 bool permission_granted_; | 67 bool permission_granted_; |
| 72 bool tab_context_updated_; | 68 bool tab_context_updated_; |
| 73 }; | 69 }; |
| 74 | 70 |
| 75 class PermissionContextBaseTests : public ChromeRenderViewHostTestHarness { | 71 class PermissionContextBaseTests : public ChromeRenderViewHostTestHarness { |
| 76 protected: | 72 protected: |
| 77 PermissionContextBaseTests() {} | 73 PermissionContextBaseTests() {} |
| 78 | 74 |
| 79 // Accept or dismiss the permission bubble or infobar. | 75 // Accept or dismiss the permission bubble or infobar. |
| 80 void RespondToPermission(TestPermissionContext* context, | 76 void RespondToPermission(bool accept) { |
| 81 const PermissionRequestID& id, | 77 if (PermissionBubbleManager::Enabled()) { |
| 82 const GURL& url, | 78 PermissionBubbleManager* manager = |
| 83 bool accept) { | 79 PermissionBubbleManager::FromWebContents(web_contents()); |
| 84 if (!PermissionBubbleManager::Enabled()) { | 80 if (accept) |
| 85 context->GetInfoBarController()->OnPermissionSet( | 81 manager->Accept(); |
| 86 id, url, url, accept, accept); | 82 else |
| 87 return; | 83 manager->Closing(); |
| 84 } else { |
| 85 PermissionInfoBarManager* manager = |
| 86 PermissionInfoBarManager::FromWebContents(web_contents()); |
| 87 if (accept) |
| 88 manager->Accept(); |
| 89 else |
| 90 manager->Closing(); |
| 88 } | 91 } |
| 89 | |
| 90 PermissionBubbleManager* manager = | |
| 91 PermissionBubbleManager::FromWebContents(web_contents()); | |
| 92 if (accept) | |
| 93 manager->Accept(); | |
| 94 else | |
| 95 manager->Closing(); | |
| 96 } | 92 } |
| 97 | 93 |
| 98 void TestAskAndGrant_TestContent() { | 94 void TestAskAndGrant_TestContent() { |
| 99 TestPermissionContext permission_context( | 95 TestPermissionContext permission_context( |
| 100 profile(), CONTENT_SETTINGS_TYPE_NOTIFICATIONS); | 96 profile(), CONTENT_SETTINGS_TYPE_NOTIFICATIONS); |
| 101 GURL url("http://www.google.com"); | 97 GURL url("http://www.google.com"); |
| 102 content::WebContentsTester::For(web_contents())->NavigateAndCommit(url); | 98 content::WebContentsTester::For(web_contents())->NavigateAndCommit(url); |
| 103 | 99 |
| 104 const PermissionRequestID id( | 100 const PermissionRequestID id( |
| 105 web_contents()->GetRenderProcessHost()->GetID(), | 101 web_contents()->GetRenderProcessHost()->GetID(), |
| 106 web_contents()->GetMainFrame()->GetRoutingID(), | 102 web_contents()->GetMainFrame()->GetRoutingID(), |
| 107 -1); | 103 0); |
| 108 permission_context.RequestPermission( | 104 permission_context.RequestPermission( |
| 109 web_contents(), | 105 web_contents(), |
| 110 id, url, true, | 106 id, url, true, |
| 111 base::Bind(&TestPermissionContext::TrackPermissionDecision, | 107 base::Bind(&TestPermissionContext::TrackPermissionDecision, |
| 112 base::Unretained(&permission_context))); | 108 base::Unretained(&permission_context))); |
| 113 | 109 |
| 114 RespondToPermission(&permission_context, id, url, true); | 110 RespondToPermission(true); |
| 115 EXPECT_TRUE(permission_context.permission_set()); | 111 EXPECT_TRUE(permission_context.permission_set()); |
| 116 EXPECT_TRUE(permission_context.permission_granted()); | 112 EXPECT_TRUE(permission_context.permission_granted()); |
| 117 EXPECT_TRUE(permission_context.tab_context_updated()); | 113 EXPECT_TRUE(permission_context.tab_context_updated()); |
| 118 | 114 |
| 119 ContentSetting setting = | 115 ContentSetting setting = |
| 120 HostContentSettingsMapFactory::GetForProfile(profile()) | 116 HostContentSettingsMapFactory::GetForProfile(profile()) |
| 121 ->GetContentSetting(url.GetOrigin(), | 117 ->GetContentSetting(url.GetOrigin(), |
| 122 url.GetOrigin(), | 118 url.GetOrigin(), |
| 123 CONTENT_SETTINGS_TYPE_NOTIFICATIONS, | 119 CONTENT_SETTINGS_TYPE_NOTIFICATIONS, |
| 124 std::string()); | 120 std::string()); |
| 125 EXPECT_EQ(CONTENT_SETTING_ALLOW, setting); | 121 EXPECT_EQ(CONTENT_SETTING_ALLOW, setting); |
| 126 } | 122 } |
| 127 | 123 |
| 128 void TestAskAndDismiss_TestContent() { | 124 void TestAskAndDismiss_TestContent() { |
| 129 TestPermissionContext permission_context( | 125 TestPermissionContext permission_context( |
| 130 profile(), CONTENT_SETTINGS_TYPE_MIDI_SYSEX); | 126 profile(), CONTENT_SETTINGS_TYPE_MIDI_SYSEX); |
| 131 GURL url("http://www.google.es"); | 127 GURL url("http://www.google.es"); |
| 132 content::WebContentsTester::For(web_contents())->NavigateAndCommit(url); | 128 content::WebContentsTester::For(web_contents())->NavigateAndCommit(url); |
| 133 | 129 |
| 134 const PermissionRequestID id( | 130 const PermissionRequestID id( |
| 135 web_contents()->GetRenderProcessHost()->GetID(), | 131 web_contents()->GetRenderProcessHost()->GetID(), |
| 136 web_contents()->GetMainFrame()->GetRoutingID(), | 132 web_contents()->GetMainFrame()->GetRoutingID(), |
| 137 -1); | 133 0); |
| 138 permission_context.RequestPermission( | 134 permission_context.RequestPermission( |
| 139 web_contents(), | 135 web_contents(), |
| 140 id, url, true, | 136 id, url, true, |
| 141 base::Bind(&TestPermissionContext::TrackPermissionDecision, | 137 base::Bind(&TestPermissionContext::TrackPermissionDecision, |
| 142 base::Unretained(&permission_context))); | 138 base::Unretained(&permission_context))); |
| 143 | 139 |
| 144 RespondToPermission(&permission_context, id, url, false); | 140 RespondToPermission(false); |
| 145 EXPECT_TRUE(permission_context.permission_set()); | 141 EXPECT_TRUE(permission_context.permission_set()); |
| 146 EXPECT_FALSE(permission_context.permission_granted()); | 142 EXPECT_FALSE(permission_context.permission_granted()); |
| 147 EXPECT_TRUE(permission_context.tab_context_updated()); | 143 EXPECT_TRUE(permission_context.tab_context_updated()); |
| 148 | 144 |
| 149 ContentSetting setting = | 145 ContentSetting setting = |
| 150 HostContentSettingsMapFactory::GetForProfile(profile()) | 146 HostContentSettingsMapFactory::GetForProfile(profile()) |
| 151 ->GetContentSetting(url.GetOrigin(), | 147 ->GetContentSetting(url.GetOrigin(), |
| 152 url.GetOrigin(), | 148 url.GetOrigin(), |
| 153 CONTENT_SETTINGS_TYPE_MIDI_SYSEX, | 149 CONTENT_SETTINGS_TYPE_MIDI_SYSEX, |
| 154 std::string()); | 150 std::string()); |
| 155 EXPECT_EQ(CONTENT_SETTING_ASK, setting); | 151 EXPECT_EQ(CONTENT_SETTING_ASK, setting); |
| 156 } | 152 } |
| 157 | 153 |
| 158 void TestRequestPermissionInvalidUrl(ContentSettingsType type) { | 154 void TestRequestPermissionInvalidUrl(ContentSettingsType type) { |
| 159 TestPermissionContext permission_context(profile(), type); | 155 TestPermissionContext permission_context(profile(), type); |
| 160 GURL url; | 156 GURL url; |
| 161 ASSERT_FALSE(url.is_valid()); | 157 ASSERT_FALSE(url.is_valid()); |
| 162 content::WebContentsTester::For(web_contents())->NavigateAndCommit(url); | 158 content::WebContentsTester::For(web_contents())->NavigateAndCommit(url); |
| 163 | 159 |
| 164 const PermissionRequestID id( | 160 const PermissionRequestID id( |
| 165 web_contents()->GetRenderProcessHost()->GetID(), | 161 web_contents()->GetRenderProcessHost()->GetID(), |
| 166 web_contents()->GetMainFrame()->GetRoutingID(), | 162 web_contents()->GetMainFrame()->GetRoutingID(), |
| 167 -1); | 163 0); |
| 168 permission_context.RequestPermission( | 164 permission_context.RequestPermission( |
| 169 web_contents(), | 165 web_contents(), |
| 170 id, url, true, | 166 id, url, true, |
| 171 base::Bind(&TestPermissionContext::TrackPermissionDecision, | 167 base::Bind(&TestPermissionContext::TrackPermissionDecision, |
| 172 base::Unretained(&permission_context))); | 168 base::Unretained(&permission_context))); |
| 173 | 169 |
| 174 EXPECT_TRUE(permission_context.permission_set()); | 170 EXPECT_TRUE(permission_context.permission_set()); |
| 175 EXPECT_FALSE(permission_context.permission_granted()); | 171 EXPECT_FALSE(permission_context.permission_granted()); |
| 176 EXPECT_TRUE(permission_context.tab_context_updated()); | 172 EXPECT_TRUE(permission_context.tab_context_updated()); |
| 177 | 173 |
| 178 ContentSetting setting = | 174 ContentSetting setting = |
| 179 HostContentSettingsMapFactory::GetForProfile(profile()) | 175 HostContentSettingsMapFactory::GetForProfile(profile()) |
| 180 ->GetContentSetting(url.GetOrigin(), | 176 ->GetContentSetting(url.GetOrigin(), |
| 181 url.GetOrigin(), | 177 url.GetOrigin(), |
| 182 type, | 178 type, |
| 183 std::string()); | 179 std::string()); |
| 184 EXPECT_EQ(CONTENT_SETTING_ASK, setting); | 180 EXPECT_EQ(CONTENT_SETTING_ASK, setting); |
| 185 } | 181 } |
| 186 | 182 |
| 187 void TestGrantAndRevoke_TestContent(ContentSettingsType type, | 183 void TestGrantAndRevoke_TestContent(ContentSettingsType type, |
| 188 ContentSetting expected_default) { | 184 ContentSetting expected_default) { |
| 189 TestPermissionContext permission_context(profile(), type); | 185 TestPermissionContext permission_context(profile(), type); |
| 190 GURL url("https://www.google.com"); | 186 GURL url("https://www.google.com"); |
| 191 content::WebContentsTester::For(web_contents())->NavigateAndCommit(url); | 187 content::WebContentsTester::For(web_contents())->NavigateAndCommit(url); |
| 192 | 188 |
| 193 const PermissionRequestID id( | 189 const PermissionRequestID id( |
| 194 web_contents()->GetRenderProcessHost()->GetID(), | 190 web_contents()->GetRenderProcessHost()->GetID(), |
| 195 web_contents()->GetMainFrame()->GetRoutingID(), | 191 web_contents()->GetMainFrame()->GetRoutingID(), |
| 196 -1); | 192 0); |
| 197 permission_context.RequestPermission( | 193 permission_context.RequestPermission( |
| 198 web_contents(), | 194 web_contents(), |
| 199 id, url, true, | 195 id, url, true, |
| 200 base::Bind(&TestPermissionContext::TrackPermissionDecision, | 196 base::Bind(&TestPermissionContext::TrackPermissionDecision, |
| 201 base::Unretained(&permission_context))); | 197 base::Unretained(&permission_context))); |
| 202 | 198 |
| 203 RespondToPermission(&permission_context, id, url, true); | 199 RespondToPermission(true); |
| 204 EXPECT_TRUE(permission_context.permission_set()); | 200 EXPECT_TRUE(permission_context.permission_set()); |
| 205 EXPECT_TRUE(permission_context.permission_granted()); | 201 EXPECT_TRUE(permission_context.permission_granted()); |
| 206 EXPECT_TRUE(permission_context.tab_context_updated()); | 202 EXPECT_TRUE(permission_context.tab_context_updated()); |
| 207 | 203 |
| 208 ContentSetting setting = | 204 ContentSetting setting = |
| 209 HostContentSettingsMapFactory::GetForProfile(profile()) | 205 HostContentSettingsMapFactory::GetForProfile(profile()) |
| 210 ->GetContentSetting(url.GetOrigin(), | 206 ->GetContentSetting(url.GetOrigin(), |
| 211 url.GetOrigin(), | 207 url.GetOrigin(), |
| 212 type, | 208 type, |
| 213 std::string()); | 209 std::string()); |
| (...skipping 11 matching lines...) Expand all Loading... |
| 225 HostContentSettingsMapFactory::GetForProfile(profile()) | 221 HostContentSettingsMapFactory::GetForProfile(profile()) |
| 226 ->GetDefaultContentSetting(type, nullptr); | 222 ->GetDefaultContentSetting(type, nullptr); |
| 227 EXPECT_EQ(default_setting, setting_after_reset); | 223 EXPECT_EQ(default_setting, setting_after_reset); |
| 228 } | 224 } |
| 229 | 225 |
| 230 private: | 226 private: |
| 231 // ChromeRenderViewHostTestHarness: | 227 // ChromeRenderViewHostTestHarness: |
| 232 void SetUp() override { | 228 void SetUp() override { |
| 233 ChromeRenderViewHostTestHarness::SetUp(); | 229 ChromeRenderViewHostTestHarness::SetUp(); |
| 234 InfoBarService::CreateForWebContents(web_contents()); | 230 InfoBarService::CreateForWebContents(web_contents()); |
| 231 PermissionInfoBarManager::CreateForWebContents(web_contents()); |
| 235 PermissionBubbleManager::CreateForWebContents(web_contents()); | 232 PermissionBubbleManager::CreateForWebContents(web_contents()); |
| 236 } | 233 } |
| 237 | 234 |
| 238 DISALLOW_COPY_AND_ASSIGN(PermissionContextBaseTests); | 235 DISALLOW_COPY_AND_ASSIGN(PermissionContextBaseTests); |
| 239 }; | 236 }; |
| 240 | 237 |
| 241 // Simulates clicking Accept. The permission should be granted and | 238 // Simulates clicking Accept. The permission should be granted and |
| 242 // saved for future use. | 239 // saved for future use. |
| 243 TEST_F(PermissionContextBaseTests, TestAskAndGrant) { | 240 TEST_F(PermissionContextBaseTests, TestAskAndGrant) { |
| 244 TestAskAndGrant_TestContent(); | 241 TestAskAndGrant_TestContent(); |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 290 #if defined(ENABLE_NOTIFICATIONS) | 287 #if defined(ENABLE_NOTIFICATIONS) |
| 291 TestGrantAndRevoke_TestContent(CONTENT_SETTINGS_TYPE_NOTIFICATIONS, | 288 TestGrantAndRevoke_TestContent(CONTENT_SETTINGS_TYPE_NOTIFICATIONS, |
| 292 CONTENT_SETTING_ASK); | 289 CONTENT_SETTING_ASK); |
| 293 #endif | 290 #endif |
| 294 TestGrantAndRevoke_TestContent(CONTENT_SETTINGS_TYPE_MIDI_SYSEX, | 291 TestGrantAndRevoke_TestContent(CONTENT_SETTINGS_TYPE_MIDI_SYSEX, |
| 295 CONTENT_SETTING_ASK); | 292 CONTENT_SETTING_ASK); |
| 296 TestGrantAndRevoke_TestContent(CONTENT_SETTINGS_TYPE_PUSH_MESSAGING, | 293 TestGrantAndRevoke_TestContent(CONTENT_SETTINGS_TYPE_PUSH_MESSAGING, |
| 297 CONTENT_SETTING_ASK); | 294 CONTENT_SETTING_ASK); |
| 298 } | 295 } |
| 299 #endif | 296 #endif |
| OLD | NEW |