Expose thumbnails of pages to iOS share extensions.

ios/chrome/browser/ui/browser_view_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 <PassKit/PassKit.h>
 
 #include <memory>
 
 #include "base/command_line.h"
@@ skipping 48 matching lines @@
 #import "ios/web/web_state/web_state_impl.h"
 #import "net/base/mac/url_conversions.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/gtest_mac.h"
 #import "third_party/ocmock/OCMock/OCMock.h"
 #include "third_party/ocmock/gtest_support.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/l10n/l10n_util_mac.h"
+#include "ui/base/test/ios/ui_image_test_utils.h"
 
 using web::NavigationManagerImpl;
 using web::WebStateImpl;
 
 // Private methods in BrowserViewController to test.
 @interface BrowserViewController (
     Testing)<CRWNativeContentProvider, PassKitDialogProvider, ShareToDelegate>
 - (void)pageLoadStarted:(NSNotification*)notification;
 - (void)pageLoadComplete:(NSNotification*)notification;
 - (void)tabSelected:(Tab*)tab;
@@ skipping 322 matching lines @@
   EXPECT_OCMOCK_VERIFY(tabMock);
 }
 
 // Verifies that BVC invokes -shareURL on ShareController with the correct
 // parameters in response to the IDC_SHARE_PAGE command.
 TEST_F(BrowserViewControllerTest, TestSharePageCommandHandling) {
   GURL expectedUrl("http://www.testurl.net");
   NSString* expectedTitle = @"title";
   [(BVCTestTabMock*)tab_.get() setUrl:expectedUrl];
   OCMockObject* tabMock = (OCMockObject*)tab_.get();
+  ios::ChromeBrowserState* ptr = chrome_browser_state_.get();
+  [[[tabMock stub] andReturnValue:OCMOCK_VALUE(ptr)] browserState];
   [[[tabMock stub] andReturn:expectedTitle] title];
   [[[tabMock stub] andReturn:expectedTitle] originalTitle];
-  base::scoped_nsobject<ShareToData> expectedShareData([[ShareToData alloc]
-          initWithURL:expectedUrl
-                title:expectedTitle
-      isOriginalTitle:YES
-      isPagePrintable:YES]);
+
+  UIImage* tabSnapshot = ui::test::uiimage_utils::UIImageWithSizeAndSolidColor(
+      CGSizeMake(300, 400), [UIColor blueColor]);
+  [[[tabMock stub] andReturn:tabSnapshot] generateSnapshotWithOverlay:NO
+                                                     visibleFrameOnly:YES];
   OCMockObject* shareControllerMock = (OCMockObject*)shareController_.get();
   // Passing non zero/nil |fromRect| and |inView| parameters to satisfy protocol
   // requirements.
+  BOOL (^shareDataChecker)
+  (id value) = ^BOOL(id value) {
+    if (![value isMemberOfClass:ShareToData.class])
+      return NO;
+    ShareToData* shareToData = (ShareToData*)value;
+    CGSize size = CGSizeMake(40, 40);
+    BOOL thumbnailDataIsEqual = ui::test::uiimage_utils::UIImagesAreEqual(
+        shareToData.thumbnailGenerator(size),
+        ui::test::uiimage_utils::UIImageWithSizeAndSolidColor(
+            size, [UIColor blueColor]));
+    return shareToData.url == expectedUrl &&
+           [shareToData.title isEqual:expectedTitle] &&
+           shareToData.isOriginalTitle == YES &&
+           shareToData.isPagePrintable == NO && thumbnailDataIsEqual;
+  };
+
   [[shareControllerMock expect]
-        shareWithData:expectedShareData
+        shareWithData:[OCMArg checkWithBlock:shareDataChecker]
            controller:bvc_
          browserState:chrome_browser_state_.get()
       shareToDelegate:bvc_
              fromRect:[bvc_ testing_shareButtonAnchorRect]
                inView:[OCMArg any]];
   [bvc_ chromeExecuteCommand:GetCommandWithTag(IDC_SHARE_PAGE)];
   EXPECT_OCMOCK_VERIFY(shareControllerMock);
 }
 
 // Verifies that BVC does not invoke -shareURL on ShareController in response
@@ skipping 94 matching lines @@
   OCMockObject* shareControllerMock = (OCMockObject*)shareController_.get();
   [[shareControllerMock expect] cancelShareAnimated:NO];
   EXPECT_CALL(*this, OnCompletionCalled());
   [bvc_ clearPresentedStateWithCompletion:^{
     this->OnCompletionCalled();
   }];
   EXPECT_OCMOCK_VERIFY(shareControllerMock);
 }
 
 }  // namespace