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

ios/chrome/app/main_controller_unittest.mm

+// 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>
+
+#import "base/mac/bind_objc_block.h"
+#include "base/threading/thread.h"
+#import "ios/chrome/app/application_delegate/app_state.h"
+#import "ios/chrome/app/application_delegate/url_opener.h"
+#include "ios/chrome/app/main_controller_private.h"
+#import "ios/chrome/browser/tabs/tab_model.h"
+#import "ios/chrome/test/base/scoped_block_swizzler.h"
+#include "testing/platform_test.h"
+#import "third_party/ocmock/OCMock/OCMock.h"
+
+#pragma mark - MainController Testing Additions
+
+@interface MainController (TestingAdditions)
+- (id)initForTesting;
+@end
+
+@implementation MainController (TestingAdditions)
+- (id)initForTesting {
+  self = [self init];
+  if (self) {
+    [self setUpAsForegrounded];
+    id mainTabModel = [OCMockObject mockForClass:[TabModel class]];
+    [[mainTabModel stub] resetSessionMetrics];
+    [[mainTabModel stub] browserStateDestroyed];
+    [[mainTabModel stub] addObserver:[OCMArg any]];
+    [[mainTabModel stub] removeObserver:[OCMArg any]];
+    [[self browserViewInformation] setMainTabModel:mainTabModel];
+  }
+  return self;
+}
+
+@end
+
+#pragma mark - MainController Test
+
+namespace {
+
+// A block that takes the arguments of
+// +handleLaunchOptions:applicationActive:tabOpener:startupInformation: and
+// returns nothing.
+typedef void (^HandleLaunchOptions)(id self,
+                                    NSDictionary* options,
+                                    BOOL applicationActive,
+                                    id<TabOpening> tabOpener,
+                                    id<StartupInformation> startupInformation,
+                                    AppState* appState);
+
+class TabOpenerTest : public PlatformTest {
+ protected:
+  BOOL swizzleHasBeenCalled() { return swizzle_block_executed_; }
+
+  void swizzleHandleLaunchOptions(
+      NSDictionary* expectedLaunchOptions,
+      id<StartupInformation> expectedStartupInformation,
+      AppState* expectedAppState) {
+    swizzle_block_executed_ = NO;
+    swizzle_block_.reset(
+        [^(id self, NSDictionary* options, BOOL applicationActive,
+           id<TabOpening> tabOpener, id<StartupInformation> startupInformation,
+           AppState* appState) {
+          swizzle_block_executed_ = YES;
+          EXPECT_EQ(expectedLaunchOptions, options);
+          EXPECT_EQ(expectedStartupInformation, startupInformation);
+          EXPECT_EQ(main_controller_.get(), tabOpener);
+          EXPECT_EQ(expectedAppState, appState);
+        } copy]);
+    URL_opening_handle_launch_swizzler_.reset(new ScopedBlockSwizzler(
+        [URLOpener class], @selector(handleLaunchOptions:
+                                       applicationActive:
+                                               tabOpener:
+                                      startupInformation:
+                                                appState:),
+        swizzle_block_));
+  }
+
+  MainController* GetMainController() {
+    if (!main_controller_.get()) {
+      main_controller_.reset([[MainController alloc] initForTesting]);
+    }
+    return main_controller_.get();
+  }
+
+ private:
+  base::scoped_nsobject<MainController> main_controller_;
+  __block BOOL swizzle_block_executed_;
+  base::mac::ScopedBlock<HandleLaunchOptions> swizzle_block_;
+  std::unique_ptr<ScopedBlockSwizzler> URL_opening_handle_launch_swizzler_;
+};
+
+#pragma mark - Tests.
+
+// Tests that -newTabFromLaunchOptions calls +handleLaunchOption and reset
+// options.
+TEST_F(TabOpenerTest, openTabFromLaunchOptionsWithOptions) {
+  // Setup.
+  NSString* sourceApplication = @"com.apple.mobilesafari";
+  NSDictionary* launchOptions =
+      @{UIApplicationLaunchOptionsSourceApplicationKey : sourceApplication};
+
+  id startupInformationMock =
+      [OCMockObject mockForProtocol:@protocol(StartupInformation)];
+  id appStateMock = [OCMockObject mockForClass:[AppState class]];
+
+  swizzleHandleLaunchOptions(launchOptions, startupInformationMock,
+                             appStateMock);
+
+  id<TabOpening> tabOpener = GetMainController();
+
+  // Action.
+  [tabOpener openTabFromLaunchOptions:launchOptions
+                   startupInformation:startupInformationMock
+                             appState:appStateMock];
+
+  // Test.
+  EXPECT_TRUE(swizzleHasBeenCalled());
+}
+
+// Tests that -newTabFromLaunchOptions do nothing if launchOptions is nil.
+TEST_F(TabOpenerTest, openTabFromLaunchOptionsWithNil) {
+  // Setup.
+  id startupInformationMock =
+      [OCMockObject mockForProtocol:@protocol(StartupInformation)];
+  id appStateMock = [OCMockObject mockForClass:[AppState class]];
+
+  swizzleHandleLaunchOptions(nil, startupInformationMock, appStateMock);
+
+  id<TabOpening> tabOpener = GetMainController();
+
+  // Action.
+  [tabOpener openTabFromLaunchOptions:nil
+                   startupInformation:startupInformationMock
+                             appState:appStateMock];
+
+  // Test.
+  EXPECT_FALSE(swizzleHasBeenCalled());
+}
+}  // namespace