Index: ui/compositor/test/test_suite.cc |
diff --git a/ui/compositor/test/test_suite.cc b/ui/compositor/test/test_suite.cc |
index c315a574f741d6d49256369dce06ae16a6c3ee42..b9ec38bcafe94222bbfea60a1979a65b84fdfa60 100644 |
--- a/ui/compositor/test/test_suite.cc |
+++ b/ui/compositor/test/test_suite.cc |
@@ -5,7 +5,8 @@ |
#include "ui/compositor/test/test_suite.h" |
#include "base/command_line.h" |
-#include "base/message_loop/message_loop.h" |
+#include "base/memory/ptr_util.h" |
+#include "base/test/scoped_task_environment.h" |
#include "build/build_config.h" |
#include "ui/compositor/compositor.h" |
#include "ui/compositor/compositor_switches.h" |
@@ -43,11 +44,13 @@ void CompositorTestSuite::Initialize() { |
display::win::SetDefaultDeviceScaleFactor(1.0f); |
#endif |
- message_loop_.reset(new base::MessageLoopForUI); |
+ scoped_task_environment_ = |
+ base::MakeUnique<base::test::ScopedTaskEnvironment>( |
+ base::test::ScopedTaskEnvironment::MainThreadType::UI); |
} |
void CompositorTestSuite::Shutdown() { |
- message_loop_.reset(); |
+ scoped_task_environment_.reset(); |
base::TestSuite::Shutdown(); |
} |