[ios clean] Creates ToolsMenu model.

ios/clean/chrome/browser/ui/tools/tools_mediator_unittest.mm

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 ca08ce0dd8a6a04222d4cef5f32002a34bfd60dd..7928f4aeb0ee466751a642daed1b61fbeaf0bc33 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,8 @@
 #import "ios/clean/chrome/browser/ui/tools/tools_mediator.h"
 
 #import "ios/clean/chrome/browser/ui/tools/tools_consumer.h"
+#import "ios/clean/chrome/browser/ui/tools/tools_mediator_private.h"
+#import "ios/clean/chrome/browser/ui/tools/tools_menu_item.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"
@@ -21,7 +23,7 @@ class ToolsMediatorTest : public PlatformTest {
   ToolsMediator* mediator_;
 };
 
-TEST_F(ToolsMediatorTest, TestSetConsumer) {
+TEST_F(ToolsMediatorTest, TestShowOverFlowControls) {
   id consumer = OCMProtocolMock(@protocol(ToolsConsumer));
   id configuration = OCMClassMock([ToolsMenuConfiguration class]);
   OCMStub([configuration isInTabSwitcher]).andReturn(YES);
@@ -33,4 +35,68 @@ TEST_F(ToolsMediatorTest, TestSetConsumer) {
   [[consumer verify] setDisplayOverflowControls:NO];
 }
 
+TEST_F(ToolsMediatorTest, TestHideOverFlowControls) {
+  id consumer = OCMProtocolMock(@protocol(ToolsConsumer));
+  id configuration = OCMClassMock([ToolsMenuConfiguration class]);
+  OCMStub([configuration isInTabSwitcher]).andReturn(NO);
+
+  mediator_ = [[ToolsMediator alloc] initWithConsumer:consumer
+                                     andConfiguration:configuration];
+
+  [[consumer verify] setToolsMenuItems:[OCMArg any]];
+  [[consumer verify] setDisplayOverflowControls:YES];
+}
+
+TEST_F(ToolsMediatorTest, TestMenuItemsForNonTabSwitcherNonIncognito) {
+  id consumer = OCMProtocolMock(@protocol(ToolsConsumer));
+  id configuration = OCMClassMock([ToolsMenuConfiguration class]);
+  OCMStub([configuration isInTabSwitcher]).andReturn(NO);
+  OCMStub([configuration isInIncognito]).andReturn(NO);
+
+  mediator_ = [[ToolsMediator alloc] initWithConsumer:consumer
+                                     andConfiguration:configuration];
+
+  EXPECT_EQ(7ul, [mediator_.menuItemsArray count]);
+}
+
+TEST_F(ToolsMediatorTest, TestMenuItemsForNonTabSwitcherIncognito) {
+  id consumer = OCMProtocolMock(@protocol(ToolsConsumer));
+  id configuration = OCMClassMock([ToolsMenuConfiguration class]);
+  OCMStub([configuration isInTabSwitcher]).andReturn(NO);
+  OCMStub([configuration isInIncognito]).andReturn(YES);
+
+  mediator_ = [[ToolsMediator alloc] initWithConsumer:consumer
+                                     andConfiguration:configuration];
+
+  EXPECT_EQ(7ul, [mediator_.menuItemsArray count]);
+}
+
+TEST_F(ToolsMediatorTest, TestMenuItemsForTabSwitcherNonIncognito) {
+  id consumer = OCMProtocolMock(@protocol(ToolsConsumer));
+  id configuration = OCMClassMock([ToolsMenuConfiguration class]);
+  OCMStub([configuration isInTabSwitcher]).andReturn(YES);
+  OCMStub([configuration isInIncognito]).andReturn(YES);
+
+  mediator_ = [[ToolsMediator alloc] initWithConsumer:consumer
+                                     andConfiguration:configuration];
+
+  ToolsMenuItem* closeAllTabsItem = mediator_.menuItemsArray[2];
+  EXPECT_EQ(@"Close All Incognito Tabs", closeAllTabsItem.title);
+  EXPECT_EQ(5ul, [mediator_.menuItemsArray count]);
+}
+
+TEST_F(ToolsMediatorTest, TestMenuItemsForTabSwitcherIncognito) {
+  id consumer = OCMProtocolMock(@protocol(ToolsConsumer));
+  id configuration = OCMClassMock([ToolsMenuConfiguration class]);
+  OCMStub([configuration isInTabSwitcher]).andReturn(YES);
+  OCMStub([configuration isInIncognito]).andReturn(NO);
+
+  mediator_ = [[ToolsMediator alloc] initWithConsumer:consumer
+                                     andConfiguration:configuration];
+
+  ToolsMenuItem* closeAllTabsItem = mediator_.menuItemsArray[2];
+  EXPECT_EQ(@"Close All Tabs", closeAllTabsItem.title);
+  EXPECT_EQ(5ul, [mediator_.menuItemsArray count]);
+}
+
 }  // namespace