Use ScopedTaskEnvironment instead of MessageLoopForUI in device tests.

device/geolocation/wifi_data_provider_common_unittest.cc

 // Copyright (c) 2012 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/wifi_data_provider_common.h"
 
 #include <memory>
 
 #include "base/macros.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/strings/utf_string_conversions.h"
+#include "base/test/scoped_task_environment.h"
 #include "base/third_party/dynamic_annotations/dynamic_annotations.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "device/geolocation/wifi_data_provider_manager.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::_;
 using testing::AnyNumber;
 using testing::AtLeast;
 using testing::DoAll;
@@ skipping 52 matching lines @@
  private:
   ~WifiDataProviderCommonWithMock() override {}
 
   DISALLOW_COPY_AND_ASSIGN(WifiDataProviderCommonWithMock);
 };
 
 // Main test fixture
 class GeolocationWifiDataProviderCommonTest : public testing::Test {
  public:
   GeolocationWifiDataProviderCommonTest()
-      : wifi_data_callback_(base::Bind(&base::DoNothing)),
+      : scoped_task_environment_(
+            base::test::ScopedTaskEnvironment::MainThreadType::UI),
+        wifi_data_callback_(base::Bind(&base::DoNothing)),
         provider_(new WifiDataProviderCommonWithMock),
         wlan_api_(provider_->wlan_api_.get()),
         polling_policy_(provider_->polling_policy_.get()) {}
 
   void SetUp() override {
     provider_->AddCallback(&wifi_data_callback_);
   }
 
   void TearDown() override {
     provider_->RemoveCallback(&wifi_data_callback_);
     provider_->StopDataProvider();
   }
 
  protected:
-  const base::MessageLoopForUI message_loop_;
+  const base::test::ScopedTaskEnvironment scoped_task_environment_;
   WifiDataProviderManager::WifiDataUpdateCallback wifi_data_callback_;
   const scoped_refptr<WifiDataProviderCommonWithMock> provider_;
 
   MockWlanApi* const wlan_api_;
   MockPollingPolicy* const polling_policy_;
 };
 
 TEST_F(GeolocationWifiDataProviderCommonTest, CreateDestroy) {
   // Test fixture members were SetUp correctly.
   EXPECT_TRUE(provider_);
@@ skipping 75 matching lines @@
   provider_->StartDataProvider();
   run_loop.Run();
 
   WifiData data;
   EXPECT_TRUE(provider_->GetData(&data));
   ASSERT_EQ(1u, data.access_point_data.size());
   EXPECT_EQ(single_access_point.ssid, data.access_point_data.begin()->ssid);
 }
 
 }  // namespace device