Upstream Chrome on iOS source code [9/11].

ios/chrome/browser/ui/tabs/tab_strip_controller_unittest.mm

Index: ios/chrome/browser/ui/tabs/tab_strip_controller_unittest.mm
diff --git a/ios/chrome/browser/ui/tabs/tab_strip_controller_unittest.mm b/ios/chrome/browser/ui/tabs/tab_strip_controller_unittest.mm
new file mode 100644
index 0000000000000000000000000000000000000000..5c313125433f44ddbad9f7b6913f9482bf6e72bb
--- /dev/null
+++ b/ios/chrome/browser/ui/tabs/tab_strip_controller_unittest.mm
@@ -0,0 +1,168 @@
+// Copyright 2012 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#import <Foundation/Foundation.h>
+
+#include <memory>
+
+#include "base/mac/scoped_nsobject.h"
+#include "ios/chrome/browser/browser_state/test_chrome_browser_state.h"
+#import "ios/chrome/browser/sessions/test_session_service.h"
+#import "ios/chrome/browser/tabs/tab.h"
+#import "ios/chrome/browser/tabs/tab_model.h"
+#import "ios/chrome/browser/ui/tabs/tab_strip_controller.h"
+#import "ios/chrome/browser/ui/tabs/tab_strip_view.h"
+#import "ios/chrome/browser/ui/tabs/tab_view.h"
+#include "ios/chrome/browser/ui/ui_util.h"
+#include "ios/web/public/test/test_web_thread_bundle.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/gtest_mac.h"
+#include "testing/platform_test.h"
+#import "third_party/ocmock/OCMock/OCMock.h"
+#import "third_party/ocmock/gtest_support.h"
+
+@interface ArrayBackedTabModel : TabModel {
+ @private
+  base::scoped_nsobject<NSMutableArray> tabsForTesting_;
+}
+
+@end
+
+@implementation ArrayBackedTabModel
+
+- (instancetype)initWithSessionWindow:(SessionWindowIOS*)window
+                       sessionService:(SessionServiceIOS*)service
+                         browserState:(ios::ChromeBrowserState*)browserState {
+  if ((self = [super initWithSessionWindow:window
+                            sessionService:service
+                              browserState:browserState])) {
+    tabsForTesting_.reset([[NSMutableArray alloc] initWithCapacity:5]);
+  }
+  return self;
+}
+
+- (void)addTabForTesting:(Tab*)tab {
+  [tabsForTesting_ addObject:tab];
+}
+
+- (Tab*)tabAtIndex:(NSUInteger)index {
+  return (Tab*)[tabsForTesting_ objectAtIndex:index];
+}
+
+- (NSUInteger)indexOfTab:(Tab*)tab {
+  return [tabsForTesting_ indexOfObject:tab];
+}
+
+- (BOOL)isEmpty {
+  return [tabsForTesting_ count] == 0;
+}
+
+- (NSUInteger)count {
+  return [tabsForTesting_ count];
+}
+
+// Pass along fast enumeration calls to the tab array.
+- (NSUInteger)countByEnumeratingWithState:(NSFastEnumerationState*)state
+                                  objects:(id*)stackbuf
+                                    count:(NSUInteger)len {
+  return [tabsForTesting_ countByEnumeratingWithState:state
+                                              objects:stackbuf
+                                                count:len];
+}
+
+- (void)closeTabAtIndex:(NSUInteger)index {
+  DCHECK(index < [tabsForTesting_ count]);
+  [self closeTab:[tabsForTesting_ objectAtIndex:index]];
+}
+
+@end
+
+namespace {
+
+class TabStripControllerTest : public PlatformTest {
+ protected:
+  void SetUp() override {
+    if (!IsIPadIdiom())
+      return;
+
+    // Need a ChromeBrowserState for the tab model.
+    TestChromeBrowserState::Builder test_cbs_builder;
+    chrome_browser_state_ = test_cbs_builder.Build();
+
+    // Setup mock TabModel, sessionService, and Tabs.
+    base::scoped_nsobject<TestSessionService> test_service(
+        [[TestSessionService alloc] init]);
+    real_tab_model_.reset([[ArrayBackedTabModel alloc]
+        initWithSessionWindow:nil
+               sessionService:test_service
+                 browserState:chrome_browser_state_.get()]);
+    id tabModel = [OCMockObject partialMockForObject:real_tab_model_];
+    id tab1 = [OCMockObject mockForClass:[Tab class]];
+    id tab2 = [OCMockObject mockForClass:[Tab class]];
+
+    // Populate the tab model.
+    [real_tab_model_ addTabForTesting:tab1];
+    [real_tab_model_ addTabForTesting:tab2];
+
+    // Stub methods for TabModel.
+    [[[tabModel stub] andDo:^(NSInvocation* invocation) {
+      // Return the raw pointer to the C++ object.
+      ios::ChromeBrowserState* browser_state = chrome_browser_state_.get();
+      [invocation setReturnValue:&browser_state];
+    }] browserState];
+    [[tabModel stub] addObserver:[OCMArg any]];
+    [[tabModel stub] removeObserver:[OCMArg any]];
+
+    // Stub methods for Tabs.
+    [[[tab1 stub] andReturn:@"Tab Title 1"] title];
+    [[[tab2 stub] andReturn:@"Tab Title 2"] title];
+    [[[tab1 stub] andReturn:nil] favicon];
+    [[[tab2 stub] andReturn:nil] favicon];
+    [[tab1 stub] close];
+    [[tab2 stub] close];
+
+    tabModel_.reset([tabModel retain]);
+    tab1_.reset([tab1 retain]);
+    tab2_.reset([tab2 retain]);
+    controller_.reset([[TabStripController alloc]
+        initWithTabModel:(TabModel*)tabModel_.get()
+                   style:TabStrip::kStyleDark]);
+
+    // Force the view to load.
+    UIWindow* window = [[UIWindow alloc] initWithFrame:CGRectZero];
+    [window addSubview:[controller_ view]];
+    window_.reset(window);
+  }
+  void TearDown() override {
+    if (!IsIPadIdiom())
+      return;
+    [real_tab_model_ browserStateDestroyed];
+  }
+
+  web::TestWebThreadBundle thread_bundle_;
+  base::scoped_nsobject<OCMockObject> tab1_;
+  base::scoped_nsobject<OCMockObject> tab2_;
+  std::unique_ptr<TestChromeBrowserState> chrome_browser_state_;
+  base::scoped_nsobject<OCMockObject> tabModel_;
+  base::scoped_nsobject<TabStripController> controller_;
+  base::scoped_nsobject<UIWindow> window_;
+  base::scoped_nsobject<ArrayBackedTabModel> real_tab_model_;
+};
+
+TEST_F(TabStripControllerTest, LoadAndDisplay) {
+  if (!IsIPadIdiom())
+    return;
+
+  // If this doesn't crash, we're good.
+  ASSERT_OCMOCK_VERIFY(tabModel_);
+  ASSERT_OCMOCK_VERIFY(tab1_);
+  ASSERT_OCMOCK_VERIFY(tab2_);
+
+  // There should be two TabViews and one new tab button nested within the
+  // parent view (which contains exactly one scroll view).
+  EXPECT_EQ(3U,
+            [[[[[controller_ view] subviews] objectAtIndex:0] subviews] count]);
+}
+
+}  // namespace