| 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..25603a383a5fa5bd4572ef6ec65f21aa9d63ed4a 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,9 @@ class AppToolbarButtonCellTest : public CocoaTest {
|
|
|
| NSButton* button_;
|
| base::scoped_nsobject<AppToolbarButtonCell> cell_;
|
| - base::MessageLoopForUI message_loop_; // Needed for gfx::Animation.
|
| +
|
| + // Needed for gfx::Animation.
|
| + base::test::ScopedTaskEnvironment scoped_task_environment_;
|
|
|
| private:
|
| DISALLOW_COPY_AND_ASSIGN(AppToolbarButtonCellTest);
|
|
|