content/browser/geofencing/geofencing_manager_unittest.cc

   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 "base/callback.h"
   6 #include "base/message_loop/message_loop.h"
   7 #include "content/browser/geofencing/geofencing_manager.h"
   8 #include "content/browser/geofencing/geofencing_service.h"
   9 #include "content/browser/service_worker/service_worker_context_wrapper.h"
  10 #include "content/public/test/test_browser_thread_bundle.h"
  11 #include "content/public/test/test_utils.h"
  12 #include "testing/gmock/include/gmock/gmock.h"
  13 #include "testing/gtest/include/gtest/gtest.h"
  14 #include "third_party/WebKit/public/platform/WebCircularGeofencingRegion.h"
  15
  16 using blink::WebCircularGeofencingRegion;
  17 typedef std::map<std::string, WebCircularGeofencingRegion> RegionMap;
  18
  19 namespace {
  20
  21 static const char* kTestRegionId = "region-id";
  22 static const int64 kTestServiceWorkerRegistrationId = 123;
  23 static const int64 kTestServiceWorkerRegistrationId2 = 456;
  24 static const int64 kTestGeofencingRegistrationId = 42;
  25 static const int64 kTestGeofencingRegistrationId2 = 43;
  26
  27 bool RegionsMatch(const WebCircularGeofencingRegion& expected,
  28                   const WebCircularGeofencingRegion& arg) {
  29   return testing::Matches(expected.latitude)(arg.latitude) &&
  30          testing::Matches(expected.longitude)(arg.longitude) &&
  31          testing::Matches(expected.radius)(arg.radius);
  32 }
  33 }
  34
  35 namespace content {
  36
  37 class TestGeofencingService : public GeofencingService {
  38  public:
  39   MOCK_METHOD0(IsServiceAvailable, bool());
  40   MOCK_METHOD2(RegisterRegion,
  41                int64(const WebCircularGeofencingRegion& region,
  42                      GeofencingRegistrationDelegate* delegate));
  43   MOCK_METHOD1(UnregisterRegion, void(int64 geofencing_registration_id));
  44 };
  45
  46 ACTION_P(SaveDelegate, delegate) {
  47   *delegate = arg1;
  48 }
  49
  50 ACTION_P(QuitRunner, runner) {
  51   runner->Quit();
  52 }
  53
  54 MATCHER_P(WebCircularGeofencingRegionEq, expected, "") {
  55   return RegionsMatch(expected, arg);
  56 }
  57
  58 class StatusCatcher {
  59  public:
  60   StatusCatcher() : was_called_(false), runner_(new MessageLoopRunner()) {}
  61
  62   void Done(GeofencingStatus status) {
  63     CHECK(!was_called_);
  64     result_ = status;
  65     was_called_ = true;
  66     runner_->Quit();
  67   }
  68
  69   GeofencingStatus Wait() {
  70     runner_->Run();
  71     CHECK(was_called_);
  72     return result_;
  73   }
  74
  75  private:
  76   bool was_called_;
  77   GeofencingStatus result_;
  78   scoped_refptr<MessageLoopRunner> runner_;
  79 };
  80
  81 class GeofencingManagerTest : public testing::Test {
  82  public:
  83   GeofencingManagerTest() : service_(nullptr) {
  84     test_region_.latitude = 37.421999;
  85     test_region_.longitude = -122.084015;
  86     test_region_.radius = 100;
  87     expected_regions_[kTestRegionId] = test_region_;
  88   }
  89
  90   void SetUp() override {
  91     service_ = new TestGeofencingService();
  92     ON_CALL(*service_, IsServiceAvailable())
  93         .WillByDefault(testing::Return(false));
  94     manager_ = new GeofencingManager(nullptr /* ServiceWorkerContextWrapper */);
  95     manager_->SetServiceForTesting(service_);
  96   }
  97
  98   void TearDown() override {
  99     manager_ = nullptr;
 100     delete service_;
 101     service_ = nullptr;
 102   }
 103
 104   void SetHasProviderForTests() {
 105     ON_CALL(*service_, IsServiceAvailable())
 106         .WillByDefault(testing::Return(true));
 107   }
 108
 109   GeofencingStatus RegisterRegionSync(
 110       int64 service_worker_registration_id,
 111       const std::string& id,
 112       const WebCircularGeofencingRegion& region) {
 113     StatusCatcher result;
 114     manager_->RegisterRegion(
 115         service_worker_registration_id,
 116         id,
 117         region,
 118         base::Bind(&StatusCatcher::Done, base::Unretained(&result)));
 119     return result.Wait();
 120   }
 121
 122   GeofencingStatus RegisterRegionSyncWithServiceResult(
 123       int64 service_worker_registration_id,
 124       const std::string& id,
 125       const WebCircularGeofencingRegion& region,
 126       GeofencingStatus service_status,
 127       int64 geofencing_registration_id) {
 128     StatusCatcher result;
 129     GeofencingRegistrationDelegate* delegate = 0;
 130     EXPECT_CALL(
 131         *service_,
 132         RegisterRegion(WebCircularGeofencingRegionEq(region), testing::_))
 133         .WillOnce(testing::DoAll(SaveDelegate(&delegate),
 134                                  testing::Return(geofencing_registration_id)));
 135     manager_->RegisterRegion(
 136         service_worker_registration_id,
 137         id,
 138         region,
 139         base::Bind(&StatusCatcher::Done, base::Unretained(&result)));
 140     CHECK(delegate);
 141     delegate->RegistrationFinished(geofencing_registration_id, service_status);
 142     return result.Wait();
 143   }
 144
 145   GeofencingStatus UnregisterRegionSync(int64 service_worker_registration_id,
 146                                         const std::string& id,
 147                                         bool should_call_service,
 148                                         int64 geofencing_registration_id = 0) {
 149     StatusCatcher result;
 150     if (should_call_service) {
 151       EXPECT_CALL(*service_, UnregisterRegion(geofencing_registration_id));
 152     }
 153     manager_->UnregisterRegion(
 154         service_worker_registration_id,
 155         id,
 156         base::Bind(&StatusCatcher::Done, base::Unretained(&result)));
 157     return result.Wait();
 158   }
 159
 160   void VerifyRegions(int64 service_worker_registration_id,
 161                      const RegionMap& expected_regions) {
 162     RegionMap regions;
 163     EXPECT_EQ(GEOFENCING_STATUS_OK,
 164               manager_->GetRegisteredRegions(service_worker_registration_id,
 165                                              &regions));
 166     EXPECT_EQ(expected_regions.size(), regions.size());
 167     for (RegionMap::const_iterator it = expected_regions.begin();
 168          it != expected_regions.end();
 169          ++it) {
 170       EXPECT_THAT(regions[it->first],
 171                   WebCircularGeofencingRegionEq(it->second));
 172     }
 173   }
 174
 175  protected:
 176   TestBrowserThreadBundle threads_;
 177   TestGeofencingService* service_;
 178   scoped_refptr<GeofencingManager> manager_;
 179
 180   WebCircularGeofencingRegion test_region_;
 181   RegionMap expected_regions_;
 182 };
 183
 184 TEST_F(GeofencingManagerTest, RegisterRegion_NoService) {
 185   EXPECT_EQ(GEOFENCING_STATUS_OPERATION_FAILED_SERVICE_NOT_AVAILABLE,
 186             RegisterRegionSync(
 187                 kTestServiceWorkerRegistrationId, kTestRegionId, test_region_));
 188 }
 189
 190 TEST_F(GeofencingManagerTest, UnregisterRegion_NoService) {
 191   EXPECT_EQ(GEOFENCING_STATUS_OPERATION_FAILED_SERVICE_NOT_AVAILABLE,
 192             UnregisterRegionSync(
 193                 kTestServiceWorkerRegistrationId, kTestRegionId, false));
 194 }
 195
 196 TEST_F(GeofencingManagerTest, GetRegisteredRegions_NoService) {
 197   RegionMap regions;
 198   EXPECT_EQ(GEOFENCING_STATUS_OPERATION_FAILED_SERVICE_NOT_AVAILABLE,
 199             manager_->GetRegisteredRegions(kTestServiceWorkerRegistrationId,
 200                                            &regions));
 201   EXPECT_TRUE(regions.empty());
 202 }
 203
 204 TEST_F(GeofencingManagerTest, RegisterRegion_FailsInService) {
 205   SetHasProviderForTests();
 206   EXPECT_EQ(
 207       GEOFENCING_STATUS_ERROR,
 208       RegisterRegionSyncWithServiceResult(kTestServiceWorkerRegistrationId,
 209                                           kTestRegionId,
 210                                           test_region_,
 211                                           GEOFENCING_STATUS_ERROR,
 212                                           -1));
 213 }
 214
 215 TEST_F(GeofencingManagerTest, RegisterRegion_SucceedsInService) {
 216   SetHasProviderForTests();
 217   EXPECT_EQ(
 218       GEOFENCING_STATUS_OK,
 219       RegisterRegionSyncWithServiceResult(kTestServiceWorkerRegistrationId,
 220                                           kTestRegionId,
 221                                           test_region_,
 222                                           GEOFENCING_STATUS_OK,
 223                                           kTestGeofencingRegistrationId));
 224   VerifyRegions(kTestServiceWorkerRegistrationId, expected_regions_);
 225 }
 226
 227 TEST_F(GeofencingManagerTest, RegisterRegion_AlreadyRegistered) {
 228   SetHasProviderForTests();
 229   EXPECT_EQ(
 230       GEOFENCING_STATUS_OK,
 231       RegisterRegionSyncWithServiceResult(kTestServiceWorkerRegistrationId,
 232                                           kTestRegionId,
 233                                           test_region_,
 234                                           GEOFENCING_STATUS_OK,
 235                                           kTestGeofencingRegistrationId));
 236   VerifyRegions(kTestServiceWorkerRegistrationId, expected_regions_);
 237
 238   WebCircularGeofencingRegion region2;
 239   region2.latitude = 43.2;
 240   region2.longitude = 1.45;
 241   region2.radius = 8.5;
 242   EXPECT_EQ(GEOFENCING_STATUS_ERROR,
 243             RegisterRegionSync(
 244                 kTestServiceWorkerRegistrationId, kTestRegionId, region2));
 245   VerifyRegions(kTestServiceWorkerRegistrationId, expected_regions_);
 246 }
 247
 248 TEST_F(GeofencingManagerTest, UnregisterRegion_NotRegistered) {
 249   SetHasProviderForTests();
 250   EXPECT_EQ(GEOFENCING_STATUS_UNREGISTRATION_FAILED_NOT_REGISTERED,
 251             UnregisterRegionSync(
 252                 kTestServiceWorkerRegistrationId, kTestRegionId, false));
 253 }
 254
 255 TEST_F(GeofencingManagerTest, UnregisterRegion_Success) {
 256   SetHasProviderForTests();
 257
 258   EXPECT_EQ(
 259       GEOFENCING_STATUS_OK,
 260       RegisterRegionSyncWithServiceResult(kTestServiceWorkerRegistrationId,
 261                                           kTestRegionId,
 262                                           test_region_,
 263                                           GEOFENCING_STATUS_OK,
 264                                           kTestGeofencingRegistrationId));
 265
 266   EXPECT_EQ(GEOFENCING_STATUS_OK,
 267             UnregisterRegionSync(kTestServiceWorkerRegistrationId,
 268                                  kTestRegionId,
 269                                  true,
 270                                  kTestGeofencingRegistrationId));
 271   VerifyRegions(kTestServiceWorkerRegistrationId, RegionMap());
 272 }
 273
 274 TEST_F(GeofencingManagerTest, GetRegisteredRegions_RegistrationInProgress) {
 275   SetHasProviderForTests();
 276   StatusCatcher result;
 277   GeofencingRegistrationDelegate* delegate = nullptr;
 278
 279   EXPECT_CALL(
 280       *service_,
 281       RegisterRegion(WebCircularGeofencingRegionEq(test_region_), testing::_))
 282       .WillOnce(testing::DoAll(SaveDelegate(&delegate),
 283                                testing::Return(kTestGeofencingRegistrationId)));
 284   manager_->RegisterRegion(
 285       kTestServiceWorkerRegistrationId,
 286       kTestRegionId,
 287       test_region_,
 288       base::Bind(&StatusCatcher::Done, base::Unretained(&result)));
 289
 290   // At this point the manager should have tried registering the region with
 291   // the service, resulting in |delegate| being set. Until the callback is
 292   // called the registration is not complete though.
 293   EXPECT_NE(delegate, nullptr);
 294   VerifyRegions(kTestServiceWorkerRegistrationId, RegionMap());
 295
 296   // Now call the callback, and verify the registration completed succesfully.
 297   delegate->RegistrationFinished(kTestGeofencingRegistrationId,
 298                                  GEOFENCING_STATUS_OK);
 299   EXPECT_EQ(GEOFENCING_STATUS_OK, result.Wait());
 300   VerifyRegions(kTestServiceWorkerRegistrationId, expected_regions_);
 301 }
 302
 303 TEST_F(GeofencingManagerTest, UnregisterRegion_RegistrationInProgress) {
 304   SetHasProviderForTests();
 305   StatusCatcher result;
 306   GeofencingRegistrationDelegate* delegate = nullptr;
 307
 308   EXPECT_CALL(
 309       *service_,
 310       RegisterRegion(WebCircularGeofencingRegionEq(test_region_), testing::_))
 311       .WillOnce(testing::DoAll(SaveDelegate(&delegate),
 312                                testing::Return(kTestGeofencingRegistrationId)));
 313   manager_->RegisterRegion(
 314       kTestServiceWorkerRegistrationId,
 315       kTestRegionId,
 316       test_region_,
 317       base::Bind(&StatusCatcher::Done, base::Unretained(&result)));
 318
 319   // At this point the manager should have tried registering the region with
 320   // the service, resulting in |delegate| being set. Until the callback is
 321   // called the registration is not complete though.
 322   EXPECT_NE(delegate, nullptr);
 323
 324   EXPECT_EQ(GEOFENCING_STATUS_UNREGISTRATION_FAILED_NOT_REGISTERED,
 325             UnregisterRegionSync(
 326                 kTestServiceWorkerRegistrationId, kTestRegionId, false));
 327 }
 328
 329 TEST_F(GeofencingManagerTest, GetRegisteredRegions_NoRegions) {
 330   SetHasProviderForTests();
 331   VerifyRegions(kTestServiceWorkerRegistrationId, RegionMap());
 332 }
 333
 334 TEST_F(GeofencingManagerTest, RegisterRegion_SeparateServiceWorkers) {
 335   SetHasProviderForTests();
 336
 337   EXPECT_EQ(
 338       GEOFENCING_STATUS_OK,
 339       RegisterRegionSyncWithServiceResult(kTestServiceWorkerRegistrationId,
 340                                           kTestRegionId,
 341                                           test_region_,
 342                                           GEOFENCING_STATUS_OK,
 343                                           kTestGeofencingRegistrationId));
 344
 345   VerifyRegions(kTestServiceWorkerRegistrationId, expected_regions_);
 346   VerifyRegions(kTestServiceWorkerRegistrationId2, RegionMap());
 347
 348   EXPECT_EQ(
 349       GEOFENCING_STATUS_OK,
 350       RegisterRegionSyncWithServiceResult(kTestServiceWorkerRegistrationId2,
 351                                           kTestRegionId,
 352                                           test_region_,
 353                                           GEOFENCING_STATUS_OK,
 354                                           kTestGeofencingRegistrationId2));
 355
 356   VerifyRegions(kTestServiceWorkerRegistrationId, expected_regions_);
 357   VerifyRegions(kTestServiceWorkerRegistrationId2, expected_regions_);
 358 }
 359
 360 TEST_F(GeofencingManagerTest, UnregisterRegion_SeparateServiceWorkers) {
 361   SetHasProviderForTests();
 362
 363   EXPECT_EQ(
 364       GEOFENCING_STATUS_OK,
 365       RegisterRegionSyncWithServiceResult(kTestServiceWorkerRegistrationId,
 366                                           kTestRegionId,
 367                                           test_region_,
 368                                           GEOFENCING_STATUS_OK,
 369                                           kTestGeofencingRegistrationId));
 370   EXPECT_EQ(
 371       GEOFENCING_STATUS_OK,
 372       RegisterRegionSyncWithServiceResult(kTestServiceWorkerRegistrationId2,
 373                                           kTestRegionId,
 374                                           test_region_,
 375                                           GEOFENCING_STATUS_OK,
 376                                           kTestGeofencingRegistrationId2));
 377
 378   EXPECT_EQ(GEOFENCING_STATUS_OK,
 379             UnregisterRegionSync(kTestServiceWorkerRegistrationId,
 380                                  kTestRegionId,
 381                                  true,
 382                                  kTestGeofencingRegistrationId));
 383
 384   VerifyRegions(kTestServiceWorkerRegistrationId, RegionMap());
 385   VerifyRegions(kTestServiceWorkerRegistrationId2, expected_regions_);
 386
 387   EXPECT_EQ(GEOFENCING_STATUS_OK,
 388             UnregisterRegionSync(kTestServiceWorkerRegistrationId2,
 389                                  kTestRegionId,
 390                                  true,
 391                                  kTestGeofencingRegistrationId2));
 392
 393   VerifyRegions(kTestServiceWorkerRegistrationId, RegionMap());
 394   VerifyRegions(kTestServiceWorkerRegistrationId2, RegionMap());
 395 }
 396
 397 TEST_F(GeofencingManagerTest, ShutdownCleansRegistrations) {
 398   SetHasProviderForTests();
 399   scoped_refptr<MessageLoopRunner> runner(new MessageLoopRunner());
 400   EXPECT_EQ(
 401       GEOFENCING_STATUS_OK,
 402       RegisterRegionSyncWithServiceResult(kTestServiceWorkerRegistrationId,
 403                                           kTestRegionId,
 404                                           test_region_,
 405                                           GEOFENCING_STATUS_OK,
 406                                           kTestGeofencingRegistrationId));
 407
 408   EXPECT_CALL(*service_, UnregisterRegion(kTestGeofencingRegistrationId))
 409       .WillOnce(QuitRunner(runner));
 410   manager_->Shutdown();
 411   runner->Run();
 412 }
 413
 414 }  // namespace content