Index: chrome/browser/ui/views/frame/web_contents_close_handler_unittest.cc |
diff --git a/chrome/browser/ui/views/frame/web_contents_close_handler_unittest.cc b/chrome/browser/ui/views/frame/web_contents_close_handler_unittest.cc |
index 0ff0896f95c0676ce0005b2e0b53d3549330f19e..950e5f10a63a906dd1964cff6637ef1670c48e1b 100644 |
--- a/chrome/browser/ui/views/frame/web_contents_close_handler_unittest.cc |
+++ b/chrome/browser/ui/views/frame/web_contents_close_handler_unittest.cc |
@@ -5,7 +5,7 @@ |
#include "chrome/browser/ui/views/frame/web_contents_close_handler.h" |
#include "base/macros.h" |
-#include "base/message_loop/message_loop.h" |
+#include "base/test/scoped_task_environment.h" |
#include "chrome/browser/ui/views/frame/web_contents_close_handler_delegate.h" |
#include "testing/gtest/include/gtest/gtest.h" |
@@ -13,9 +13,10 @@ class MockWebContentsCloseHandlerDelegate |
: public WebContentsCloseHandlerDelegate { |
public: |
explicit MockWebContentsCloseHandlerDelegate() |
- : got_clone_(false), |
- got_destroy_(false) { |
- } |
+ : scoped_task_environment_( |
+ base::test::ScopedTaskEnvironment::MainThreadType::UI), |
+ got_clone_(false), |
+ got_destroy_(false) {} |
~MockWebContentsCloseHandlerDelegate() override {} |
void Clear() { |
@@ -33,7 +34,7 @@ class MockWebContentsCloseHandlerDelegate |
void DestroyClonedLayer() override { got_destroy_ = true; } |
private: |
- base::MessageLoopForUI message_loop_; |
+ base::test::ScopedTaskEnvironment scoped_task_environment_; |
bool got_clone_; |
bool got_destroy_; |
@@ -90,4 +91,3 @@ TEST_F(WebContentsCloseHandlerTest, DontDestroyImmediatleyAfterCancel) { |
EXPECT_FALSE(close_handler_delegate_.got_clone()); |
EXPECT_TRUE(IsTimerRunning()); |
} |
- |