Expose thumbnails of pages to iOS share extensions.

ios/chrome/browser/ui/activity_services/activity_service_controller_unittest.mm

 // Copyright 2014 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 "ios/chrome/browser/ui/activity_services/activity_service_controller.h"
 
 #import <MobileCoreServices/MobileCoreServices.h>
 
 #include "base/mac/scoped_nsobject.h"
 #import "base/test/ios/wait_util.h"
@@ skipping 26 matching lines @@
  protected:
   void SetUp() override {
     PlatformTest::SetUp();
     parentController_.reset(
         [[UIViewController alloc] initWithNibName:nil bundle:nil]);
     [[UIApplication sharedApplication] keyWindow].rootViewController =
         parentController_;
     shareToDelegate_.reset(
         [[OCMockObject mockForProtocol:@protocol(ShareToDelegate)] retain]);
     shareData_.reset([[ShareToData alloc]
-            initWithURL:GURL("https://chromium.org")
-                  title:@""
-        isOriginalTitle:YES
-        isPagePrintable:YES]);
+               initWithURL:GURL("https://chromium.org")
+                     title:@""
+           isOriginalTitle:YES
+           isPagePrintable:YES
+        thumbnailGenerator:DummyThumbnailGeneratorBlock()]);
   }
 
   void TearDown() override {
     [[UIApplication sharedApplication] keyWindow].rootViewController = nil;
     PlatformTest::TearDown();
   }
 
+  ThumbnailGeneratorBlock DummyThumbnailGeneratorBlock() {
+    return ^UIImage*(CGSize const& size) { return nil; };
+  }
+
   id<ShareToDelegate> GetShareToDelegate() {
     return static_cast<id<ShareToDelegate>>(shareToDelegate_.get());
   }
 
   CGRect AnchorRect() {
     // On iPad, UIPopovers must be anchored to rectangles that have a non zero
     // size.
     return CGRectMake(0, 0, 1, 1);
   }
 
@@ skipping 160 matching lines @@
 
 // Verifies that an UIActivityImageSource is sent to the
 // UIActivityViewController if and only if the ShareToData contains an image.
 TEST_F(ActivityServiceControllerTest, ActivityItemsForData) {
   base::scoped_nsobject<ActivityServiceController> activityController(
       [[ActivityServiceController alloc] init]);
 
   // ShareToData does not contain an image, so the result items array will not
   // contain an image source.
   base::scoped_nsobject<ShareToData> data([[ShareToData alloc]
-          initWithURL:GURL("https://chromium.org")
-                title:@"foo"
-      isOriginalTitle:YES
-      isPagePrintable:YES]);
+             initWithURL:GURL("https://chromium.org")
+                   title:@"foo"
+         isOriginalTitle:YES
+         isPagePrintable:YES
+      thumbnailGenerator:DummyThumbnailGeneratorBlock()]);
   NSArray* items = [activityController activityItemsForData:data];
   EXPECT_FALSE(ArrayContainsImageSource(items));
 
   // Adds an image to the ShareToData object and call -activityItemsForData:
   // again. Verifies that the result items array contains an image source.
   [data setImage:[UIImage imageNamed:@"activity_services_print"]];
   items = [activityController activityItemsForData:data];
   EXPECT_TRUE(ArrayContainsImageSource(items));
 }
 
 // Verifies that when App Extension support is enabled, the URL string is
 // passed in a dictionary as part of the Activity Items to the App Extension.
 TEST_F(ActivityServiceControllerTest, ActivityItemsForDataWithPasswordAppEx) {
   base::scoped_nsobject<ActivityServiceController> activityController(
       [[ActivityServiceController alloc] init]);
   base::scoped_nsobject<ShareToData> data([[ShareToData alloc]
-          initWithURL:GURL("https://chromium.org/login.html")
-                title:@"kung fu fighting"
-      isOriginalTitle:YES
-      isPagePrintable:YES]);
+             initWithURL:GURL("https://chromium.org/login.html")
+                   title:@"kung fu fighting"
+         isOriginalTitle:YES
+         isPagePrintable:YES
+      thumbnailGenerator:DummyThumbnailGeneratorBlock()]);
   NSArray* items = [activityController activityItemsForData:data];
   NSString* findLoginAction =
       (NSString*)activity_services::kUTTypeAppExtensionFindLoginAction;
   // Gets the list of NSExtensionItem objects returned by the array of
   // id<UIActivityItemSource> objects returned by -activityItemsForData:.
   NSArray* extensionItems = FindItemsForActivityType(
       items, activity_services::kAppExtensionOnePassword);
   ASSERT_EQ(1U, [extensionItems count]);
   NSExtensionItem* item = extensionItems[0];
   EXPECT_EQ(1U, item.attachments.count);
@@ skipping 37 matching lines @@
   EXPECT_NSEQ(@"kung fu fighting", title);
 }
 
 // Verifies that a Share extension can fetch a URL when Password App Extension
 // is enabled.
 TEST_F(ActivityServiceControllerTest,
        ActivityItemsForDataWithPasswordAppExReturnsURL) {
   base::scoped_nsobject<ActivityServiceController> activityController(
       [[ActivityServiceController alloc] init]);
   base::scoped_nsobject<ShareToData> data([[ShareToData alloc]
-          initWithURL:GURL("https://chromium.org/login.html")
-                title:@"kung fu fighting"
-      isOriginalTitle:YES
-      isPagePrintable:YES]);
+             initWithURL:GURL("https://chromium.org/login.html")
+                   title:@"kung fu fighting"
+         isOriginalTitle:YES
+         isPagePrintable:YES
+      thumbnailGenerator:DummyThumbnailGeneratorBlock()]);
   NSArray* items = [activityController activityItemsForData:data];
   NSString* shareAction = @"com.apple.UIKit.activity.PostToFacebook";
   NSArray* urlItems =
       FindItemsEqualsToUTType(items, shareAction, @"public.url");
   ASSERT_EQ(1U, [urlItems count]);
   id<UIActivityItemSource> itemSource = urlItems[0];
   id mockActivityViewController =
       [OCMockObject niceMockForClass:[UIActivityViewController class]];
   id item = [itemSource activityViewController:mockActivityViewController
                            itemForActivityType:shareAction];
@@ skipping 86 matching lines @@
 }
 
 // Verifies that the PrintActivity is sent to the UIActivityViewController if
 // and only if the activity is "printable".
 TEST_F(ActivityServiceControllerTest, ApplicationActivitiesForData) {
   base::scoped_nsobject<ActivityServiceController> activityController(
       [[ActivityServiceController alloc] init]);
 
   // Verify printable data.
   base::scoped_nsobject<ShareToData> data([[ShareToData alloc]
-          initWithURL:GURL("https://chromium.org/printable")
-                title:@"bar"
-      isOriginalTitle:YES
-      isPagePrintable:YES]);
+             initWithURL:GURL("https://chromium.org/printable")
+                   title:@"bar"
+         isOriginalTitle:YES
+         isPagePrintable:YES
+      thumbnailGenerator:DummyThumbnailGeneratorBlock()]);
 
   NSArray* items =
       [activityController applicationActivitiesForData:data controller:nil];
   NSUInteger expected_items_count =
       reading_list::switches::IsReadingListEnabled() ? 2U : 1U;
   ASSERT_EQ(expected_items_count, [items count]);
   EXPECT_EQ([PrintActivity class], [[items objectAtIndex:0] class]);
 
   // Verify non-printable data.
   data.reset([[ShareToData alloc]
-          initWithURL:GURL("https://chromium.org/unprintable")
-                title:@"baz"
-      isOriginalTitle:YES
-      isPagePrintable:NO]);
+             initWithURL:GURL("https://chromium.org/unprintable")
+                   title:@"baz"
+         isOriginalTitle:YES
+         isPagePrintable:NO
+      thumbnailGenerator:DummyThumbnailGeneratorBlock()]);
   items = [activityController applicationActivitiesForData:data controller:nil];
   EXPECT_EQ(expected_items_count - 1, [items count]);
 }
 
 TEST_F(ActivityServiceControllerTest, FindLoginActionTypeConformsToPublicURL) {
   // If this test fails, it is probably due to missing or incorrect
   // UTImportedTypeDeclarations in Info.plist. Note that there are
   // two Info.plist,
   // - ios/chrome/app/resources/Info.plist for Chrome app
   // - testing/gtest_ios/unittest-Info.plist for ios_chrome_unittests
@@ skipping 14 matching lines @@
   // public.url UTType in order to allow Share actions (e.g. Facebook, Twitter,
   // etc) to appear on UIActivityViewController opened by Chrome).
   CFStringRef chromeFindLoginAction = reinterpret_cast<CFStringRef>(
       activity_services::kUTTypeAppExtensionFindLoginAction);
   EXPECT_TRUE(UTTypeConformsTo(chromeFindLoginAction, kUTTypeURL));
   EXPECT_TRUE(
       UTTypeConformsTo(chromeFindLoginAction, onePasswordFindLoginAction));
 }
 
 }  // namespace