Index: chrome/browser/ui/cocoa/toolbar/app_toolbar_button_cell_unittest.mm |
diff --git a/chrome/browser/ui/cocoa/toolbar/app_toolbar_button_cell_unittest.mm b/chrome/browser/ui/cocoa/toolbar/app_toolbar_button_cell_unittest.mm |
index ff6c506688588889cc9b6be10dfcb9cd27d9c1f3..bf7fe122483e9aff952e0e5b4c66caf1b89d7550 100644 |
--- a/chrome/browser/ui/cocoa/toolbar/app_toolbar_button_cell_unittest.mm |
+++ b/chrome/browser/ui/cocoa/toolbar/app_toolbar_button_cell_unittest.mm |
@@ -5,7 +5,7 @@ |
#import "chrome/browser/ui/cocoa/toolbar/app_toolbar_button_cell.h" |
#include "base/macros.h" |
-#include "base/message_loop/message_loop.h" |
+#include "base/test/scoped_task_environment.h" |
#import "chrome/browser/ui/cocoa/test/cocoa_test_helper.h" |
@interface TestAppToolbarButton : NSButton |
@@ -21,7 +21,9 @@ |
class AppToolbarButtonCellTest : public CocoaTest { |
protected: |
- AppToolbarButtonCellTest() { |
+ AppToolbarButtonCellTest() |
+ : scoped_task_environment_( |
+ base::test::ScopedTaskEnvironment::MainThreadType::UI) { |
base::scoped_nsobject<NSButton> button([[TestAppToolbarButton alloc] |
initWithFrame:NSMakeRect(0, 0, 29, 29)]); |
button_ = button; |
@@ -30,7 +32,8 @@ class AppToolbarButtonCellTest : public CocoaTest { |
NSButton* button_; |
base::scoped_nsobject<AppToolbarButtonCell> cell_; |
- base::MessageLoopForUI message_loop_; // Needed for gfx::Animation. |
+ base::test::ScopedTaskEnvironment |
+ scoped_task_environment_; // Needed for gfx::Animation. |
sky
2017/04/27 22:17:46
Move the comment above the member.
fdoray
2017/05/01 17:44:59
Done.
|
private: |
DISALLOW_COPY_AND_ASSIGN(AppToolbarButtonCellTest); |