| 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_;
|
| };
|
|
|