Use ScopedTaskEnvironment instead of MessageLoopForUI in chromeos tests.

chromeos/network/firewall_hole_unittest.cc

 // Copyright 2015 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 "chromeos/network/firewall_hole.h"
 
 #include "base/bind.h"
 #include "base/memory/ptr_util.h"
 #include "base/run_loop.h"
+#include "base/test/scoped_task_environment.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "chromeos/dbus/mock_permission_broker_client.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using chromeos::DBusThreadManager;
 using chromeos::FirewallHole;
 using chromeos::MockPermissionBrokerClient;
 using testing::_;
 
 namespace {
 
 ACTION_TEMPLATE(InvokeCallback,
                 HAS_1_TEMPLATE_PARAMS(int, k),
                 AND_1_VALUE_PARAMS(p1)) {
   ::std::tr1::get<k>(args).Run(p1);
 }
 
 }  // namespace
 
 class FirewallHoleTest : public testing::Test {
  public:
-  FirewallHoleTest() {}
+  FirewallHoleTest()
+      : scoped_task_environment_(
+            base::test::ScopedTaskEnvironment::MainThreadType::UI) {}
   ~FirewallHoleTest() override {}
 
   void SetUp() override {
     mock_permission_broker_client_ = new MockPermissionBrokerClient();
     DBusThreadManager::GetSetterForTesting()->SetPermissionBrokerClient(
         base::WrapUnique(mock_permission_broker_client_));
   }
 
   void TearDown() override { DBusThreadManager::Shutdown(); }
 
   void AssertOpenSuccess(std::unique_ptr<FirewallHole> hole) {
     EXPECT_TRUE(hole.get() != nullptr);
     run_loop_.Quit();
   }
 
   void AssertOpenFailure(std::unique_ptr<FirewallHole> hole) {
     EXPECT_TRUE(hole.get() == nullptr);
     run_loop_.Quit();
   }
 
  private:
-  base::MessageLoopForUI message_loop_;
+  base::test::ScopedTaskEnvironment scoped_task_environment_;
 
  protected:
   base::RunLoop run_loop_;
   MockPermissionBrokerClient* mock_permission_broker_client_ = nullptr;
 };
 
 TEST_F(FirewallHoleTest, GrantTcpPortAccess) {
   EXPECT_CALL(*mock_permission_broker_client_,
               RequestTcpPortAccess(1234, "foo0", _, _))
       .WillOnce(InvokeCallback<3>(true));
@@ skipping 33 matching lines @@
 TEST_F(FirewallHoleTest, DenyUdpPortAccess) {
   EXPECT_CALL(*mock_permission_broker_client_,
               RequestUdpPortAccess(1234, "foo0", _, _))
       .WillOnce(InvokeCallback<3>(false));
 
   FirewallHole::Open(
       FirewallHole::PortType::UDP, 1234, "foo0",
       base::Bind(&FirewallHoleTest::AssertOpenFailure, base::Unretained(this)));
   run_loop_.Run();
 }