OLD | NEW |
(Empty) | |
| 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 |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "content/browser/geolocation/geolocation_provider_impl.h" |
| 6 |
| 7 #include <memory> |
| 8 |
| 9 #include "base/bind.h" |
| 10 #include "base/bind_helpers.h" |
| 11 #include "base/location.h" |
| 12 #include "base/macros.h" |
| 13 #include "base/memory/ptr_util.h" |
| 14 #include "base/memory/ref_counted.h" |
| 15 #include "base/run_loop.h" |
| 16 #include "base/single_thread_task_runner.h" |
| 17 #include "base/strings/string16.h" |
| 18 #include "base/time/time.h" |
| 19 #include "content/browser/geolocation/mock_location_arbitrator.h" |
| 20 #include "content/public/browser/access_token_store.h" |
| 21 #include "content/public/browser/browser_thread.h" |
| 22 #include "content/public/test/test_browser_thread.h" |
| 23 #include "testing/gmock/include/gmock/gmock.h" |
| 24 #include "testing/gtest/include/gtest/gtest.h" |
| 25 |
| 26 using testing::MakeMatcher; |
| 27 using testing::Matcher; |
| 28 using testing::MatcherInterface; |
| 29 using testing::MatchResultListener; |
| 30 |
| 31 namespace content { |
| 32 |
| 33 class LocationProviderForTestArbitrator : public GeolocationProviderImpl { |
| 34 public: |
| 35 LocationProviderForTestArbitrator() : mock_arbitrator_(NULL) {} |
| 36 ~LocationProviderForTestArbitrator() override {} |
| 37 |
| 38 // Only valid for use on the geolocation thread. |
| 39 MockLocationArbitrator* mock_arbitrator() const { |
| 40 return mock_arbitrator_; |
| 41 } |
| 42 |
| 43 protected: |
| 44 // GeolocationProviderImpl implementation: |
| 45 std::unique_ptr<LocationArbitrator> CreateArbitrator() override; |
| 46 |
| 47 private: |
| 48 // An alias to the arbitrator stored in the super class, where it is owned. |
| 49 MockLocationArbitrator* mock_arbitrator_; |
| 50 }; |
| 51 |
| 52 std::unique_ptr<LocationArbitrator> |
| 53 LocationProviderForTestArbitrator::CreateArbitrator() { |
| 54 DCHECK(mock_arbitrator_ == NULL); |
| 55 mock_arbitrator_ = new MockLocationArbitrator; |
| 56 return base::WrapUnique(mock_arbitrator_); |
| 57 } |
| 58 |
| 59 class GeolocationObserver { |
| 60 public: |
| 61 virtual ~GeolocationObserver() {} |
| 62 virtual void OnLocationUpdate(const Geoposition& position) = 0; |
| 63 }; |
| 64 |
| 65 class MockGeolocationObserver : public GeolocationObserver { |
| 66 public: |
| 67 MOCK_METHOD1(OnLocationUpdate, void(const Geoposition& position)); |
| 68 }; |
| 69 |
| 70 class AsyncMockGeolocationObserver : public MockGeolocationObserver { |
| 71 public: |
| 72 void OnLocationUpdate(const Geoposition& position) override { |
| 73 MockGeolocationObserver::OnLocationUpdate(position); |
| 74 base::MessageLoop::current()->QuitWhenIdle(); |
| 75 } |
| 76 }; |
| 77 |
| 78 class MockGeolocationCallbackWrapper { |
| 79 public: |
| 80 MOCK_METHOD1(Callback, void(const Geoposition& position)); |
| 81 }; |
| 82 |
| 83 class GeopositionEqMatcher |
| 84 : public MatcherInterface<const Geoposition&> { |
| 85 public: |
| 86 explicit GeopositionEqMatcher(const Geoposition& expected) |
| 87 : expected_(expected) {} |
| 88 |
| 89 bool MatchAndExplain(const Geoposition& actual, |
| 90 MatchResultListener* listener) const override { |
| 91 return actual.latitude == expected_.latitude && |
| 92 actual.longitude == expected_.longitude && |
| 93 actual.altitude == expected_.altitude && |
| 94 actual.accuracy == expected_.accuracy && |
| 95 actual.altitude_accuracy == expected_.altitude_accuracy && |
| 96 actual.heading == expected_.heading && |
| 97 actual.speed == expected_.speed && |
| 98 actual.timestamp == expected_.timestamp && |
| 99 actual.error_code == expected_.error_code && |
| 100 actual.error_message == expected_.error_message; |
| 101 } |
| 102 |
| 103 void DescribeTo(::std::ostream* os) const override { |
| 104 *os << "which matches the expected position"; |
| 105 } |
| 106 |
| 107 void DescribeNegationTo(::std::ostream* os) const override { |
| 108 *os << "which does not match the expected position"; |
| 109 } |
| 110 |
| 111 private: |
| 112 Geoposition expected_; |
| 113 |
| 114 DISALLOW_COPY_AND_ASSIGN(GeopositionEqMatcher); |
| 115 }; |
| 116 |
| 117 Matcher<const Geoposition&> GeopositionEq(const Geoposition& expected) { |
| 118 return MakeMatcher(new GeopositionEqMatcher(expected)); |
| 119 } |
| 120 |
| 121 class GeolocationProviderTest : public testing::Test { |
| 122 protected: |
| 123 GeolocationProviderTest() |
| 124 : message_loop_(), |
| 125 ui_thread_(BrowserThread::UI, &message_loop_), |
| 126 provider_(new LocationProviderForTestArbitrator) { |
| 127 } |
| 128 |
| 129 ~GeolocationProviderTest() override {} |
| 130 |
| 131 LocationProviderForTestArbitrator* provider() { return provider_.get(); } |
| 132 |
| 133 // Called on test thread. |
| 134 bool ProvidersStarted(); |
| 135 void SendMockLocation(const Geoposition& position); |
| 136 |
| 137 private: |
| 138 // Called on provider thread. |
| 139 void GetProvidersStarted(bool* started); |
| 140 |
| 141 base::MessageLoop message_loop_; |
| 142 TestBrowserThread ui_thread_; |
| 143 std::unique_ptr<LocationProviderForTestArbitrator> provider_; |
| 144 }; |
| 145 |
| 146 |
| 147 bool GeolocationProviderTest::ProvidersStarted() { |
| 148 DCHECK(provider_->IsRunning()); |
| 149 DCHECK(base::MessageLoop::current() == &message_loop_); |
| 150 bool started; |
| 151 provider_->task_runner()->PostTaskAndReply( |
| 152 FROM_HERE, base::Bind(&GeolocationProviderTest::GetProvidersStarted, |
| 153 base::Unretained(this), &started), |
| 154 base::MessageLoop::QuitWhenIdleClosure()); |
| 155 base::RunLoop().Run(); |
| 156 return started; |
| 157 } |
| 158 |
| 159 void GeolocationProviderTest::GetProvidersStarted(bool* started) { |
| 160 DCHECK(provider_->task_runner()->BelongsToCurrentThread()); |
| 161 *started = provider_->mock_arbitrator()->providers_started(); |
| 162 } |
| 163 |
| 164 void GeolocationProviderTest::SendMockLocation(const Geoposition& position) { |
| 165 DCHECK(provider_->IsRunning()); |
| 166 DCHECK(base::MessageLoop::current() == &message_loop_); |
| 167 provider_->task_runner()->PostTask( |
| 168 FROM_HERE, base::Bind(&GeolocationProviderImpl::OnLocationUpdate, |
| 169 base::Unretained(provider_.get()), position)); |
| 170 } |
| 171 |
| 172 // Regression test for http://crbug.com/59377 |
| 173 TEST_F(GeolocationProviderTest, OnPermissionGrantedWithoutObservers) { |
| 174 EXPECT_FALSE(provider()->user_did_opt_into_location_services_for_testing()); |
| 175 provider()->UserDidOptIntoLocationServices(); |
| 176 EXPECT_TRUE(provider()->user_did_opt_into_location_services_for_testing()); |
| 177 } |
| 178 |
| 179 void DummyFunction(const Geoposition& position) { |
| 180 } |
| 181 |
| 182 TEST_F(GeolocationProviderTest, StartStop) { |
| 183 EXPECT_FALSE(provider()->IsRunning()); |
| 184 GeolocationProviderImpl::LocationUpdateCallback callback = |
| 185 base::Bind(&DummyFunction); |
| 186 std::unique_ptr<content::GeolocationProvider::Subscription> subscription = |
| 187 provider()->AddLocationUpdateCallback(callback, false); |
| 188 EXPECT_TRUE(provider()->IsRunning()); |
| 189 EXPECT_TRUE(ProvidersStarted()); |
| 190 |
| 191 subscription.reset(); |
| 192 |
| 193 EXPECT_FALSE(ProvidersStarted()); |
| 194 EXPECT_TRUE(provider()->IsRunning()); |
| 195 } |
| 196 |
| 197 TEST_F(GeolocationProviderTest, StalePositionNotSent) { |
| 198 Geoposition first_position; |
| 199 first_position.latitude = 12; |
| 200 first_position.longitude = 34; |
| 201 first_position.accuracy = 56; |
| 202 first_position.timestamp = base::Time::Now(); |
| 203 |
| 204 AsyncMockGeolocationObserver first_observer; |
| 205 GeolocationProviderImpl::LocationUpdateCallback first_callback = base::Bind( |
| 206 &MockGeolocationObserver::OnLocationUpdate, |
| 207 base::Unretained(&first_observer)); |
| 208 EXPECT_CALL(first_observer, OnLocationUpdate(GeopositionEq(first_position))); |
| 209 std::unique_ptr<content::GeolocationProvider::Subscription> subscription = |
| 210 provider()->AddLocationUpdateCallback(first_callback, false); |
| 211 SendMockLocation(first_position); |
| 212 base::RunLoop().Run(); |
| 213 |
| 214 subscription.reset(); |
| 215 |
| 216 Geoposition second_position; |
| 217 second_position.latitude = 13; |
| 218 second_position.longitude = 34; |
| 219 second_position.accuracy = 56; |
| 220 second_position.timestamp = base::Time::Now(); |
| 221 |
| 222 AsyncMockGeolocationObserver second_observer; |
| 223 |
| 224 // After adding a second observer, check that no unexpected position update |
| 225 // is sent. |
| 226 EXPECT_CALL(second_observer, OnLocationUpdate(testing::_)).Times(0); |
| 227 GeolocationProviderImpl::LocationUpdateCallback second_callback = base::Bind( |
| 228 &MockGeolocationObserver::OnLocationUpdate, |
| 229 base::Unretained(&second_observer)); |
| 230 std::unique_ptr<content::GeolocationProvider::Subscription> subscription2 = |
| 231 provider()->AddLocationUpdateCallback(second_callback, false); |
| 232 base::RunLoop().RunUntilIdle(); |
| 233 |
| 234 // The second observer should receive the new position now. |
| 235 EXPECT_CALL(second_observer, |
| 236 OnLocationUpdate(GeopositionEq(second_position))); |
| 237 SendMockLocation(second_position); |
| 238 base::RunLoop().Run(); |
| 239 |
| 240 subscription2.reset(); |
| 241 EXPECT_FALSE(ProvidersStarted()); |
| 242 } |
| 243 |
| 244 TEST_F(GeolocationProviderTest, OverrideLocationForTesting) { |
| 245 Geoposition position; |
| 246 position.error_code = Geoposition::ERROR_CODE_POSITION_UNAVAILABLE; |
| 247 provider()->OverrideLocationForTesting(position); |
| 248 // Adding an observer when the location is overridden should synchronously |
| 249 // update the observer with our overridden position. |
| 250 MockGeolocationObserver mock_observer; |
| 251 EXPECT_CALL(mock_observer, OnLocationUpdate(GeopositionEq(position))); |
| 252 GeolocationProviderImpl::LocationUpdateCallback callback = base::Bind( |
| 253 &MockGeolocationObserver::OnLocationUpdate, |
| 254 base::Unretained(&mock_observer)); |
| 255 std::unique_ptr<content::GeolocationProvider::Subscription> subscription = |
| 256 provider()->AddLocationUpdateCallback(callback, false); |
| 257 subscription.reset(); |
| 258 // Wait for the providers to be stopped now that all clients are gone. |
| 259 EXPECT_FALSE(ProvidersStarted()); |
| 260 } |
| 261 |
| 262 } // namespace content |
OLD | NEW |