Use ScopedTaskEnvironment instead of MessageLoopForUI in chrome tests.

chrome/browser/chromeos/settings/install_attributes_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 "chrome/browser/chromeos/settings/install_attributes.h"
 
 #include <memory>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/memory/ptr_util.h"
 #include "base/path_service.h"
 #include "base/run_loop.h"
+#include "base/test/scoped_task_environment.h"
 #include "chrome/browser/chromeos/policy/proto/install_attributes.pb.h"
 #include "chromeos/chromeos_paths.h"
 #include "chromeos/cryptohome/cryptohome_util.h"
 #include "chromeos/dbus/cryptohome_client.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "google_apis/gaia/gaia_auth_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace chromeos {
 
@@ skipping 15 matching lines @@
 
 }  // namespace
 
 static const char kTestDomain[] = "example.com";
 static const char kTestRealm[] = "realm.example.com";
 static const char kTestDeviceId[] = "133750519";
 static const char kTestUserDeprecated[] = "test@example.com";
 
 class InstallAttributesTest : public testing::Test {
  protected:
-  InstallAttributesTest() {}
+  InstallAttributesTest()
+      : scoped_task_environment_(
+            base::test::ScopedTaskEnvironment::MainThreadType::UI) {}
 
   void SetUp() override {
     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
     ASSERT_TRUE(PathService::OverrideAndCreateIfNeeded(
         FILE_INSTALL_ATTRIBUTES, GetTempPath(), true, false));
     DBusThreadManager::Initialize();
     install_attributes_ = base::MakeUnique<InstallAttributes>(
         DBusThreadManager::Get()->GetCryptohomeClient());
   }
 
   void TearDown() override { DBusThreadManager::Shutdown(); }
 
   base::FilePath GetTempPath() const {
     base::FilePath temp_path = base::MakeAbsoluteFilePath(temp_dir_.GetPath());
     return temp_path.Append("install_attrs_test");
   }
 
   void SetAttribute(
       cryptohome::SerializedInstallAttributes* install_attrs_proto,
       const std::string& name,
       const std::string& value) {
     cryptohome::SerializedInstallAttributes::Attribute* attribute;
     attribute = install_attrs_proto->add_attributes();
     attribute->set_name(name);
     attribute->set_value(value);
   }
 
-  base::MessageLoopForUI message_loop_;
+  base::test::ScopedTaskEnvironment scoped_task_environment_;
   base::ScopedTempDir temp_dir_;
   std::unique_ptr<InstallAttributes> install_attributes_;
 
   InstallAttributes::LockResult LockDeviceAndWaitForResult(
       policy::DeviceMode device_mode,
       const std::string& domain,
       const std::string& realm,
       const std::string& device_id) {
     base::RunLoop loop;
     InstallAttributes::LockResult result;
@@ skipping 216 matching lines @@
   chromeos::DBusMethodCallStatus status =
       chromeos::DBusMethodCallStatus::DBUS_METHOD_CALL_FAILURE;
   install_attributes_->SetBlockDevmodeInTpm(
       true, base::Bind(&OnSetBlockDevmode, &status));
   base::RunLoop().RunUntilIdle();
 
   EXPECT_EQ(chromeos::DBusMethodCallStatus::DBUS_METHOD_CALL_SUCCESS, status);
 }
 
 }  // namespace chromeos