| Index: chrome/browser/ui/cocoa/wrench_menu/wrench_menu_controller_unittest.mm
|
| diff --git a/chrome/browser/ui/cocoa/wrench_menu/wrench_menu_controller_unittest.mm b/chrome/browser/ui/cocoa/wrench_menu/wrench_menu_controller_unittest.mm
|
| index 34511790d67cbb7ec7eb77daf79a2879fbebdbd5..146b8748e3888ba0bdbe4346bd337b906af98493 100644
|
| --- a/chrome/browser/ui/cocoa/wrench_menu/wrench_menu_controller_unittest.mm
|
| +++ b/chrome/browser/ui/cocoa/wrench_menu/wrench_menu_controller_unittest.mm
|
| @@ -7,6 +7,7 @@
|
| #include "chrome/browser/ui/cocoa/browser_test_helper.h"
|
| #import "chrome/browser/ui/cocoa/cocoa_test_helper.h"
|
| #import "chrome/browser/ui/cocoa/toolbar/toolbar_controller.h"
|
| +#include "chrome/browser/ui/cocoa/run_loop_testing.h"
|
| #import "chrome/browser/ui/cocoa/view_resizer_pong.h"
|
| #import "chrome/browser/ui/cocoa/wrench_menu/wrench_menu_controller.h"
|
| #include "chrome/browser/ui/toolbar/wrench_menu_model.h"
|
| @@ -14,14 +15,6 @@
|
| #include "testing/gtest/include/gtest/gtest.h"
|
| #include "testing/platform_test.h"
|
|
|
| -// Override to avoid dealing with run loops in the testing environment.
|
| -@implementation WrenchMenuController (UnitTesting)
|
| -- (void)dispatchCommandInternal:(NSInteger)tag {
|
| - [self wrenchMenuModel]->ExecuteCommand(tag);
|
| -}
|
| -@end
|
| -
|
| -
|
| namespace {
|
|
|
| class MockWrenchMenuModel : public WrenchMenuModel {
|
| @@ -79,6 +72,7 @@ TEST_F(WrenchMenuControllerTest, DispatchSimple) {
|
| [controller() setModel:&fake_model_];
|
|
|
| [controller() dispatchWrenchMenuCommand:button.get()];
|
| + chrome::testing::NSRunLoopRunAllPending();
|
| }
|
|
|
| } // namespace
|
|
|