Index: device/u2f/u2f_register_unittest.cc |
diff --git a/device/u2f/u2f_register_unittest.cc b/device/u2f/u2f_register_unittest.cc |
index 6b183da56e302e3a9645a0f0a54a966360a08a1b..5bb55e461aaf531295a81361cbb61bfcf0f56a11 100644 |
--- a/device/u2f/u2f_register_unittest.cc |
+++ b/device/u2f/u2f_register_unittest.cc |
@@ -5,6 +5,7 @@ |
#include <list> |
#include "base/run_loop.h" |
+#include "base/test/scoped_task_environment.h" |
#include "base/test/test_io_thread.h" |
#include "device/base/mock_device_client.h" |
#include "device/hid/mock_hid_service.h" |
@@ -17,7 +18,10 @@ namespace device { |
class U2fRegisterTest : public testing::Test { |
public: |
- U2fRegisterTest() : io_thread_(base::TestIOThread::kAutoStart) {} |
+ U2fRegisterTest() |
+ : scoped_task_environment_( |
+ base::test::ScopedTaskEnvironment::MainThreadType::UI), |
+ io_thread_(base::TestIOThread::kAutoStart) {} |
void SetUp() override { |
MockHidService* hid_service = device_client_.hid_service(); |
@@ -25,7 +29,7 @@ class U2fRegisterTest : public testing::Test { |
} |
protected: |
- base::MessageLoopForUI message_loop_; |
+ base::test::ScopedTaskEnvironment scoped_task_environment_; |
base::TestIOThread io_thread_; |
device::MockDeviceClient device_client_; |
}; |