Index: extensions/shell/test/shell_test.cc |
diff --git a/extensions/shell/test/shell_test.cc b/extensions/shell/test/shell_test.cc |
index 96ebb2a69441fd8adaee050cd2cafb93f0d97113..ff6dc69abf8d4b6efcf81ce92898b9e37f72a41f 100644 |
--- a/extensions/shell/test/shell_test.cc |
+++ b/extensions/shell/test/shell_test.cc |
@@ -8,7 +8,9 @@ |
#include "base/command_line.h" |
#include "base/files/file_path.h" |
#include "base/logging.h" |
+#include "base/message_loop/message_loop.h" |
#include "base/path_service.h" |
+#include "base/run_loop.h" |
#include "build/build_config.h" |
#include "components/version_info/version_info.h" |
#include "content/public/common/content_switches.h" |
@@ -57,7 +59,8 @@ void AppShellTest::SetUpOnMainThread() { |
} |
void AppShellTest::RunTestOnMainThreadLoop() { |
- base::MessageLoopForUI::current()->RunUntilIdle(); |
+ DCHECK(base::MessageLoopForUI::IsCurrent()); |
+ base::RunLoop().RunUntilIdle(); |
SetUpOnMainThread(); |