Gets rid of the LocationArbitrator interface, in preference for LocationProvider.

device/geolocation/geolocation_provider_impl_unittest.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "device/geolocation/geolocation_provider_impl.h"
 
 #include <memory>
 
+#include "base/at_exit.h"
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/location.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/memory/ref_counted.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/strings/string16.h"
 #include "base/time/time.h"
 #include "device/geolocation/access_token_store.h"
-#include "device/geolocation/mock_location_arbitrator.h"
+#include "device/geolocation/mock_location_provider.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::MakeMatcher;
 using testing::Matcher;
 using testing::MatcherInterface;
 using testing::MatchResultListener;
 
 namespace device {
-
-class LocationProviderForTestArbitrator : public GeolocationProviderImpl {
- public:
-  LocationProviderForTestArbitrator() : mock_arbitrator_(NULL) {}
-  ~LocationProviderForTestArbitrator() override {}
-
-  // Only valid for use on the geolocation thread.
-  MockLocationArbitrator* mock_arbitrator() const { return mock_arbitrator_; }
-
- protected:
-  // GeolocationProviderImpl implementation:
-  std::unique_ptr<LocationArbitrator> CreateArbitrator() override;
-
- private:
-  // An alias to the arbitrator stored in the super class, where it is owned.
-  MockLocationArbitrator* mock_arbitrator_;
-};
-
-std::unique_ptr<LocationArbitrator>
-LocationProviderForTestArbitrator::CreateArbitrator() {
-  DCHECK(mock_arbitrator_ == NULL);
-  mock_arbitrator_ = new MockLocationArbitrator;
-  return base::WrapUnique(mock_arbitrator_);
-}
+namespace {
 
 class GeolocationObserver {
  public:
   virtual ~GeolocationObserver() {}
   virtual void OnLocationUpdate(const Geoposition& position) = 0;
 };
 
 class MockGeolocationObserver : public GeolocationObserver {
  public:
   MOCK_METHOD1(OnLocationUpdate, void(const Geoposition& position));
@@ skipping 42 matching lines @@
  private:
   Geoposition expected_;
 
   DISALLOW_COPY_AND_ASSIGN(GeopositionEqMatcher);
 };
 
 Matcher<const Geoposition&> GeopositionEq(const Geoposition& expected) {
   return MakeMatcher(new GeopositionEqMatcher(expected));
 }
 
+void DummyFunction(const Geoposition& position) {}
+
+}  // namespace
+
 class GeolocationProviderTest : public testing::Test {
  protected:
-  GeolocationProviderTest()
-      : message_loop_(), provider_(new LocationProviderForTestArbitrator) {}
+  GeolocationProviderTest() : arbitrator_(new MockLocationProvider) {
+    provider()->SetArbitratorForTesting(arbitrator_);
+  }
 
   ~GeolocationProviderTest() override {}
 
-  LocationProviderForTestArbitrator* provider() { return provider_.get(); }
+  GeolocationProviderImpl* provider() {
+    return GeolocationProviderImpl::GetInstance();
+  }
+
+  MockLocationProvider* arbitrator() { return arbitrator_; }
 
   // Called on test thread.
   bool ProvidersStarted();
   void SendMockLocation(const Geoposition& position);
 
  private:
   // Called on provider thread.
   void GetProvidersStarted(bool* started);
 
+  // |at_exit| must be initialized before all other variables to handle
+  // correctly reinitializing GeolocationProviderImpl (Singleton)

[Wez, 2016/08/12 00:33:44]

It needs to be before all other variables so that it is available for
Singletons, etc to register with, to be torn-down when the test completes.

[CJ, 2016/08/12 20:22:46]

Done.

+  // |message_loop_| must come next to include all other variable construction
+  // on the correct thread.

[Wez, 2016/08/12 00:33:44]

nit: Move this comment down to the message_loop_ member

[CJ, 2016/08/12 20:22:46]

Done.

+  base::AtExitManager at_exit_;
   base::MessageLoopForUI message_loop_;
-  std::unique_ptr<LocationProviderForTestArbitrator> provider_;
+
+  // Owned by the GeolocationProviderImpl class.
+  MockLocationProvider* arbitrator_;
 };
 
 bool GeolocationProviderTest::ProvidersStarted() {
-  DCHECK(provider_->IsRunning());
+  DCHECK(provider()->IsRunning());
   DCHECK(base::MessageLoop::current() == &message_loop_);
+
   bool started;
-  provider_->task_runner()->PostTaskAndReply(
+  provider()->task_runner()->PostTaskAndReply(
       FROM_HERE, base::Bind(&GeolocationProviderTest::GetProvidersStarted,
                             base::Unretained(this), &started),
       base::MessageLoop::QuitWhenIdleClosure());
   base::RunLoop().Run();
   return started;
 }
 
 void GeolocationProviderTest::GetProvidersStarted(bool* started) {
-  DCHECK(provider_->task_runner()->BelongsToCurrentThread());
-  *started = provider_->mock_arbitrator()->providers_started();
+  DCHECK(provider()->task_runner()->BelongsToCurrentThread());
+  *started = arbitrator()->providers_started();
 }
 
 void GeolocationProviderTest::SendMockLocation(const Geoposition& position) {
-  DCHECK(provider_->IsRunning());
+  DCHECK(provider()->IsRunning());
   DCHECK(base::MessageLoop::current() == &message_loop_);
-  provider_->task_runner()->PostTask(
+  provider()->task_runner()->PostTask(
       FROM_HERE, base::Bind(&GeolocationProviderImpl::OnLocationUpdate,
-                            base::Unretained(provider_.get()), position));
+                            base::Unretained(provider()), position));
 }
 
 // Regression test for http://crbug.com/59377
 TEST_F(GeolocationProviderTest, OnPermissionGrantedWithoutObservers) {
+  provider()->SetArbitratorForTesting(nullptr);
   EXPECT_FALSE(provider()->user_did_opt_into_location_services_for_testing());
   provider()->UserDidOptIntoLocationServices();
   EXPECT_TRUE(provider()->user_did_opt_into_location_services_for_testing());
 }
 
-void DummyFunction(const Geoposition& position) {}
-
 TEST_F(GeolocationProviderTest, StartStop) {
   EXPECT_FALSE(provider()->IsRunning());
   GeolocationProviderImpl::LocationUpdateCallback callback =
       base::Bind(&DummyFunction);
   std::unique_ptr<GeolocationProvider::Subscription> subscription =
       provider()->AddLocationUpdateCallback(callback, false);
   EXPECT_TRUE(provider()->IsRunning());
   EXPECT_TRUE(ProvidersStarted());
 
   subscription.reset();
@@ skipping 61 matching lines @@
       base::Bind(&MockGeolocationObserver::OnLocationUpdate,
                  base::Unretained(&mock_observer));
   std::unique_ptr<GeolocationProvider::Subscription> subscription =
       provider()->AddLocationUpdateCallback(callback, false);
   subscription.reset();
   // Wait for the providers to be stopped now that all clients are gone.
   EXPECT_FALSE(ProvidersStarted());
 }
 
 }  // namespace device