| Index: ios/clean/chrome/browser/ui/tools/tools_mediator_unittest.mm
|
| diff --git a/ios/clean/chrome/browser/ui/tools/tools_mediator_unittest.mm b/ios/clean/chrome/browser/ui/tools/tools_mediator_unittest.mm
|
| index 5329a2f8be50af07f8a1226d1f2c73059255fd11..ca08ce0dd8a6a04222d4cef5f32002a34bfd60dd 100644
|
| --- a/ios/clean/chrome/browser/ui/tools/tools_mediator_unittest.mm
|
| +++ b/ios/clean/chrome/browser/ui/tools/tools_mediator_unittest.mm
|
| @@ -5,6 +5,7 @@
|
| #import "ios/clean/chrome/browser/ui/tools/tools_mediator.h"
|
|
|
| #import "ios/clean/chrome/browser/ui/tools/tools_consumer.h"
|
| +#import "ios/shared/chrome/browser/ui/tools_menu/tools_menu_configuration.h"
|
| #include "testing/platform_test.h"
|
| #import "third_party/ocmock/OCMock/OCMock.h"
|
| #include "third_party/ocmock/gtest_support.h"
|
| @@ -22,9 +23,14 @@ class ToolsMediatorTest : public PlatformTest {
|
|
|
| TEST_F(ToolsMediatorTest, TestSetConsumer) {
|
| id consumer = OCMProtocolMock(@protocol(ToolsConsumer));
|
| - mediator_ = [[ToolsMediator alloc] initWithConsumer:consumer];
|
| + id configuration = OCMClassMock([ToolsMenuConfiguration class]);
|
| + OCMStub([configuration isInTabSwitcher]).andReturn(YES);
|
| +
|
| + mediator_ = [[ToolsMediator alloc] initWithConsumer:consumer
|
| + andConfiguration:configuration];
|
|
|
| [[consumer verify] setToolsMenuItems:[OCMArg any]];
|
| + [[consumer verify] setDisplayOverflowControls:NO];
|
| }
|
|
|
| } // namespace
|
|
|