Index: chrome/browser/safe_browsing/browser_feature_extractor_unittest.cc |
diff --git a/chrome/browser/safe_browsing/browser_feature_extractor_unittest.cc b/chrome/browser/safe_browsing/browser_feature_extractor_unittest.cc |
index 4c2a58fb7093caf092907852da7b06c3a2b2ae1e..1d0b4bd287a99476e1240b5b9249463526425013 100644 |
--- a/chrome/browser/safe_browsing/browser_feature_extractor_unittest.cc |
+++ b/chrome/browser/safe_browsing/browser_feature_extractor_unittest.cc |
@@ -179,7 +179,8 @@ class BrowserFeatureExtractorTest : public ChromeRenderViewHostTestHarness { |
// Feature extraction takes ownership of the request object |
// and passes it along to the done callback in the end. |
StartExtractMalwareFeatures(request); |
- base::MessageLoopForUI::current()->Run(); |
+ ASSERT_TRUE(base::MessageLoopForUI::IsCurrent()); |
+ base::RunLoop().Run(); |
EXPECT_EQ(1U, success_.count(request)); |
EXPECT_TRUE(success_[request]); |
} |