Index: net/http/transport_security_persister_unittest.cc |
diff --git a/net/http/transport_security_persister_unittest.cc b/net/http/transport_security_persister_unittest.cc |
index 5809c92719f223075c1eb6511dc9d4f6fe0ee042..9d93601e950c66bb1777705c4dfb6c33ac40760a 100644 |
--- a/net/http/transport_security_persister_unittest.cc |
+++ b/net/http/transport_security_persister_unittest.cc |
@@ -13,6 +13,8 @@ |
#include "base/files/file_util.h" |
#include "base/files/scoped_temp_dir.h" |
#include "base/message_loop/message_loop.h" |
+#include "base/run_loop.h" |
+#include "base/threading/thread_task_runner_handle.h" |
#include "net/http/transport_security_state.h" |
#include "testing/gtest/include/gtest/gtest.h" |
@@ -28,14 +30,15 @@ class TransportSecurityPersisterTest : public testing::Test { |
} |
~TransportSecurityPersisterTest() override { |
- base::MessageLoopForIO::current()->RunUntilIdle(); |
+ EXPECT_TRUE(base::MessageLoopForIO::IsCurrent()); |
+ base::RunLoop().RunUntilIdle(); |
} |
void SetUp() override { |
ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
+ ASSERT_TRUE(base::MessageLoopForIO::IsCurrent()); |
persister_.reset(new TransportSecurityPersister( |
- &state_, temp_dir_.path(), |
- base::MessageLoopForIO::current()->task_runner(), false)); |
+ &state_, temp_dir_.path(), base::ThreadTaskRunnerHandle::Get(), false)); |
} |
protected: |