OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 <set> | 5 #include <set> |
6 #include <string> | 6 #include <string> |
7 #include <utility> | 7 #include <utility> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/hash_tables.h" | 10 #include "base/hash_tables.h" |
11 #include "base/memory/scoped_vector.h" | 11 #include "base/memory/scoped_vector.h" |
12 #include "base/synchronization/waitable_event.h" | 12 #include "base/synchronization/waitable_event.h" |
13 #include "chrome/browser/content_settings/host_content_settings_map.h" | 13 #include "chrome/browser/content_settings/host_content_settings_map.h" |
14 #include "chrome/browser/content_settings/tab_specific_content_settings.h" | 14 #include "chrome/browser/content_settings/tab_specific_content_settings.h" |
15 #include "chrome/browser/geolocation/chrome_geolocation_permission_context.h" | 15 #include "chrome/browser/geolocation/chrome_geolocation_permission_context.h" |
16 #include "chrome/browser/geolocation/chrome_geolocation_permission_context_facto
ry.h" | 16 #include "chrome/browser/geolocation/chrome_geolocation_permission_context_facto
ry.h" |
17 #include "chrome/browser/geolocation/geolocation_permission_request_id.h" | 17 #include "chrome/browser/geolocation/geolocation_permission_request_id.h" |
18 #include "chrome/browser/infobars/confirm_infobar_delegate.h" | 18 #include "chrome/browser/infobars/confirm_infobar_delegate.h" |
19 #include "chrome/browser/infobars/infobar.h" | 19 #include "chrome/browser/infobars/infobar.h" |
20 #include "chrome/browser/infobars/infobar_service.h" | 20 #include "chrome/browser/infobars/infobar_service.h" |
21 #include "chrome/common/chrome_notification_types.h" | 21 #include "chrome/common/chrome_notification_types.h" |
22 #include "chrome/test/base/chrome_render_view_host_test_harness.h" | 22 #include "chrome/test/base/chrome_render_view_host_test_harness.h" |
23 #include "chrome/test/base/testing_profile.h" | 23 #include "chrome/test/base/testing_profile.h" |
24 #include "content/public/browser/navigation_details.h" | 24 #include "content/public/browser/navigation_details.h" |
25 #include "content/public/browser/notification_registrar.h" | 25 #include "content/public/browser/notification_registrar.h" |
26 #include "content/public/browser/notification_service.h" | 26 #include "content/public/browser/notification_service.h" |
27 #include "content/public/browser/web_contents.h" | 27 #include "content/public/browser/web_contents.h" |
28 #include "content/public/test/mock_render_process_host.h" | 28 #include "content/public/test/mock_render_process_host.h" |
29 #include "content/public/test/test_browser_thread.h" | |
30 #include "content/public/test/test_renderer_host.h" | 29 #include "content/public/test/test_renderer_host.h" |
31 #include "content/public/test/web_contents_tester.h" | 30 #include "content/public/test/web_contents_tester.h" |
32 #include "extensions/browser/view_type_utils.h" | 31 #include "extensions/browser/view_type_utils.h" |
33 #include "testing/gtest/include/gtest/gtest.h" | 32 #include "testing/gtest/include/gtest/gtest.h" |
34 | 33 |
35 #if defined(OS_ANDROID) | 34 #if defined(OS_ANDROID) |
36 #include "base/prefs/pref_service.h" | 35 #include "base/prefs/pref_service.h" |
37 #include "chrome/browser/android/mock_google_location_settings_helper.h" | 36 #include "chrome/browser/android/mock_google_location_settings_helper.h" |
38 #include "chrome/common/pref_names.h" | 37 #include "chrome/common/pref_names.h" |
39 #endif | 38 #endif |
(...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
91 void ClosedDelegateTracker::Clear() { | 90 void ClosedDelegateTracker::Clear() { |
92 removed_infobar_delegates_.clear(); | 91 removed_infobar_delegates_.clear(); |
93 } | 92 } |
94 | 93 |
95 | 94 |
96 // GeolocationPermissionContextTests ------------------------------------------ | 95 // GeolocationPermissionContextTests ------------------------------------------ |
97 | 96 |
98 // This class sets up GeolocationArbitrator. | 97 // This class sets up GeolocationArbitrator. |
99 class GeolocationPermissionContextTests | 98 class GeolocationPermissionContextTests |
100 : public ChromeRenderViewHostTestHarness { | 99 : public ChromeRenderViewHostTestHarness { |
101 public: | |
102 GeolocationPermissionContextTests(); | |
103 | |
104 protected: | 100 protected: |
105 virtual ~GeolocationPermissionContextTests(); | 101 // ChromeRenderViewHostTestHarness: |
| 102 virtual void SetUp() OVERRIDE; |
| 103 virtual void TearDown() OVERRIDE; |
106 | 104 |
107 GeolocationPermissionRequestID RequestID(int bridge_id); | 105 GeolocationPermissionRequestID RequestID(int bridge_id); |
108 GeolocationPermissionRequestID RequestIDForTab(int tab, int bridge_id); | 106 GeolocationPermissionRequestID RequestIDForTab(int tab, int bridge_id); |
109 InfoBarService* infobar_service() { | 107 InfoBarService* infobar_service() { |
110 return InfoBarService::FromWebContents(web_contents()); | 108 return InfoBarService::FromWebContents(web_contents()); |
111 } | 109 } |
112 InfoBarService* infobar_service_for_tab(int tab) { | 110 InfoBarService* infobar_service_for_tab(int tab) { |
113 return InfoBarService::FromWebContents(extra_tabs_[tab]); | 111 return InfoBarService::FromWebContents(extra_tabs_[tab]); |
114 } | 112 } |
115 | 113 |
(...skipping 11 matching lines...) Expand all Loading... |
127 bool allowed); | 125 bool allowed); |
128 void AddNewTab(const GURL& url); | 126 void AddNewTab(const GURL& url); |
129 void CheckTabContentsState(const GURL& requesting_frame, | 127 void CheckTabContentsState(const GURL& requesting_frame, |
130 ContentSetting expected_content_setting); | 128 ContentSetting expected_content_setting); |
131 | 129 |
132 scoped_refptr<ChromeGeolocationPermissionContext> | 130 scoped_refptr<ChromeGeolocationPermissionContext> |
133 geolocation_permission_context_; | 131 geolocation_permission_context_; |
134 ClosedDelegateTracker closed_delegate_tracker_; | 132 ClosedDelegateTracker closed_delegate_tracker_; |
135 ScopedVector<content::WebContents> extra_tabs_; | 133 ScopedVector<content::WebContents> extra_tabs_; |
136 | 134 |
137 private: | |
138 // ChromeRenderViewHostTestHarness: | |
139 virtual void SetUp() OVERRIDE; | |
140 virtual void TearDown() OVERRIDE; | |
141 | |
142 content::TestBrowserThread ui_thread_; | |
143 content::TestBrowserThread db_thread_; | |
144 | |
145 // A map between renderer child id and a pair represending the bridge id and | 135 // A map between renderer child id and a pair represending the bridge id and |
146 // whether the requested permission was allowed. | 136 // whether the requested permission was allowed. |
147 base::hash_map<int, std::pair<int, bool> > responses_; | 137 base::hash_map<int, std::pair<int, bool> > responses_; |
148 }; | 138 }; |
149 | 139 |
150 GeolocationPermissionContextTests::GeolocationPermissionContextTests() | |
151 : ChromeRenderViewHostTestHarness(), | |
152 ui_thread_(content::BrowserThread::UI, MessageLoop::current()), | |
153 db_thread_(content::BrowserThread::DB) { | |
154 } | |
155 | |
156 GeolocationPermissionContextTests::~GeolocationPermissionContextTests() { | |
157 } | |
158 | |
159 GeolocationPermissionRequestID GeolocationPermissionContextTests::RequestID( | 140 GeolocationPermissionRequestID GeolocationPermissionContextTests::RequestID( |
160 int bridge_id) { | 141 int bridge_id) { |
161 return GeolocationPermissionRequestID( | 142 return GeolocationPermissionRequestID( |
162 web_contents()->GetRenderProcessHost()->GetID(), | 143 web_contents()->GetRenderProcessHost()->GetID(), |
163 web_contents()->GetRenderViewHost()->GetRoutingID(), | 144 web_contents()->GetRenderViewHost()->GetRoutingID(), |
164 bridge_id); | 145 bridge_id); |
165 } | 146 } |
166 | 147 |
167 GeolocationPermissionRequestID | 148 GeolocationPermissionRequestID |
168 GeolocationPermissionContextTests::RequestIDForTab(int tab, int bridge_id) { | 149 GeolocationPermissionContextTests::RequestIDForTab(int tab, int bridge_id) { |
(...skipping 77 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
246 EXPECT_EQ(1U, state_map.count(requesting_frame.GetOrigin())); | 227 EXPECT_EQ(1U, state_map.count(requesting_frame.GetOrigin())); |
247 EXPECT_EQ(0U, state_map.count(requesting_frame)); | 228 EXPECT_EQ(0U, state_map.count(requesting_frame)); |
248 GeolocationSettingsState::StateMap::const_iterator settings = | 229 GeolocationSettingsState::StateMap::const_iterator settings = |
249 state_map.find(requesting_frame.GetOrigin()); | 230 state_map.find(requesting_frame.GetOrigin()); |
250 ASSERT_FALSE(settings == state_map.end()) | 231 ASSERT_FALSE(settings == state_map.end()) |
251 << "geolocation state not found " << requesting_frame; | 232 << "geolocation state not found " << requesting_frame; |
252 EXPECT_EQ(expected_content_setting, settings->second); | 233 EXPECT_EQ(expected_content_setting, settings->second); |
253 } | 234 } |
254 | 235 |
255 void GeolocationPermissionContextTests::SetUp() { | 236 void GeolocationPermissionContextTests::SetUp() { |
256 db_thread_.Start(); | |
257 ChromeRenderViewHostTestHarness::SetUp(); | 237 ChromeRenderViewHostTestHarness::SetUp(); |
258 | 238 |
259 // Set up required helpers, and make this be as "tabby" as the code requires. | 239 // Set up required helpers, and make this be as "tabby" as the code requires. |
260 extensions::SetViewType(web_contents(), extensions::VIEW_TYPE_TAB_CONTENTS); | 240 extensions::SetViewType(web_contents(), extensions::VIEW_TYPE_TAB_CONTENTS); |
261 InfoBarService::CreateForWebContents(web_contents()); | 241 InfoBarService::CreateForWebContents(web_contents()); |
262 TabSpecificContentSettings::CreateForWebContents(web_contents()); | 242 TabSpecificContentSettings::CreateForWebContents(web_contents()); |
263 #if defined(OS_ANDROID) | 243 #if defined(OS_ANDROID) |
264 MockGoogleLocationSettingsHelper::SetLocationStatus(true, true); | 244 MockGoogleLocationSettingsHelper::SetLocationStatus(true, true); |
265 #endif | 245 #endif |
266 geolocation_permission_context_ = | 246 geolocation_permission_context_ = |
267 ChromeGeolocationPermissionContextFactory::GetForProfile(profile()); | 247 ChromeGeolocationPermissionContextFactory::GetForProfile(profile()); |
268 } | 248 } |
269 | 249 |
270 void GeolocationPermissionContextTests::TearDown() { | 250 void GeolocationPermissionContextTests::TearDown() { |
271 extra_tabs_.clear(); | 251 extra_tabs_.clear(); |
272 ChromeRenderViewHostTestHarness::TearDown(); | 252 ChromeRenderViewHostTestHarness::TearDown(); |
273 // Schedule another task on the DB thread to notify us that it's safe to | |
274 // carry on with the test. | |
275 base::WaitableEvent done(false, false); | |
276 content::BrowserThread::PostTask( | |
277 content::BrowserThread::DB, FROM_HERE, | |
278 base::Bind(&base::WaitableEvent::Signal, base::Unretained(&done))); | |
279 done.Wait(); | |
280 db_thread_.Stop(); | |
281 } | 253 } |
282 | 254 |
283 // Tests ---------------------------------------------------------------------- | 255 // Tests ---------------------------------------------------------------------- |
284 | 256 |
285 TEST_F(GeolocationPermissionContextTests, SinglePermission) { | 257 TEST_F(GeolocationPermissionContextTests, SinglePermission) { |
286 GURL requesting_frame("http://www.example.com/geolocation"); | 258 GURL requesting_frame("http://www.example.com/geolocation"); |
287 NavigateAndCommit(requesting_frame); | 259 NavigateAndCommit(requesting_frame); |
288 EXPECT_EQ(0U, infobar_service()->infobar_count()); | 260 EXPECT_EQ(0U, infobar_service()->infobar_count()); |
289 RequestGeolocationPermission(RequestID(0), requesting_frame); | 261 RequestGeolocationPermission(RequestID(0), requesting_frame); |
290 ASSERT_EQ(1U, infobar_service()->infobar_count()); | 262 ASSERT_EQ(1U, infobar_service()->infobar_count()); |
(...skipping 409 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
700 details.entry = web_contents()->GetController().GetLastCommittedEntry(); | 672 details.entry = web_contents()->GetController().GetLastCommittedEntry(); |
701 EXPECT_FALSE(infobar_0->ShouldExpire(details)); | 673 EXPECT_FALSE(infobar_0->ShouldExpire(details)); |
702 // Ensure the infobar will expire when we commit the pending navigation. | 674 // Ensure the infobar will expire when we commit the pending navigation. |
703 details.entry = web_contents()->GetController().GetActiveEntry(); | 675 details.entry = web_contents()->GetController().GetActiveEntry(); |
704 EXPECT_TRUE(infobar_0->ShouldExpire(details)); | 676 EXPECT_TRUE(infobar_0->ShouldExpire(details)); |
705 | 677 |
706 // Delete the tab contents. | 678 // Delete the tab contents. |
707 DeleteContents(); | 679 DeleteContents(); |
708 delete infobar_0; | 680 delete infobar_0; |
709 } | 681 } |
OLD | NEW |