Use the same design for all suggestions

ios/chrome/browser/ui/content_suggestions/cells/content_suggestions_item_unittest.mm

+// Copyright 2017 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/content_suggestions/cells/content_suggestions_item.h"
+
+#include "base/time/time.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#import "third_party/ocmock/OCMock/OCMock.h"
+#import "third_party/ocmock/gtest_support.h"
+#include "url/gurl.h"
+
+#if !defined(__has_feature) || !__has_feature(objc_arc)
+#error "This file requires ARC support."
+#endif
+
+namespace {
+
+// Tests that configureCell: sets all the fields of the cell except the image
+// and fetches the image through the delegate.
+TEST(ContentSuggestionsItemTest, CellIsConfiguredWithoutImage) {
+  // Setup.
+  NSString* title = @"testTitle";
+  NSString* subtitle = @"testSubtitle";
+  GURL url = GURL("http://chromium.org");
+  NSString* publisher = @"publisherName";
+  base::Time publishTime = base::Time::Now();
+  id delegateMock = OCMProtocolMock(@protocol(ContentSuggestionsItemDelegate));
+  ContentSuggestionsItem* item =
+      [[ContentSuggestionsItem alloc] initWithType:0
+                                             title:title
+                                          subtitle:subtitle
+                                          delegate:delegateMock
+                                               url:url];
+  item.hasImage = YES;
+  item.publisher = publisher;
+  item.publishDate = publishTime;
+  item.availableOffline = YES;
+  OCMExpect([delegateMock loadImageForSuggestionItem:item]);
+  ContentSuggestionsCell* cell = [[[item cellClass] alloc] init];
+  ASSERT_EQ([ContentSuggestionsCell class], [cell class]);
+  ASSERT_EQ(url, item.URL);
+  ASSERT_EQ(nil, item.image);
+  id cellMock = OCMPartialMock(cell);
+  OCMExpect([cellMock setContentImage:item.image]);
+  OCMExpect([cellMock setSubtitleText:subtitle]);
+  OCMExpect([cellMock setAdditionalInformationWithPublisherName:publisher
+                                                           date:publishTime
+                                            offlineAvailability:YES]);
+
+  // Action.
+  [item configureCell:cell];
+
+  // Tests.
+  EXPECT_OCMOCK_VERIFY(cellMock);
+  EXPECT_EQ(title, cell.titleLabel.text);
+  EXPECT_OCMOCK_VERIFY(delegateMock);
+}
+
+// Tests that configureCell: does not call the delegate if it fetched the image
+// once.
+TEST(ContentSuggestionsItemTest, DontFetchImageIsImageIsBeingFetched) {
+  // Setup.
+  NSString* title = @"testTitle";
+  NSString* subtitle = @"testSubtitle";
+  GURL url = GURL("http://chromium.org");
+  id niceDelegateMock =
+      OCMProtocolMock(@protocol(ContentSuggestionsItemDelegate));
+  ContentSuggestionsItem* item =
+      [[ContentSuggestionsItem alloc] initWithType:0
+                                             title:title
+                                          subtitle:subtitle
+                                          delegate:niceDelegateMock
+                                               url:url];
+  item.hasImage = YES;
+  item.image = [[UIImage alloc] init];
+
+  OCMExpect([niceDelegateMock loadImageForSuggestionItem:item]);
+  ContentSuggestionsCell* cell = [[[item cellClass] alloc] init];
+  ASSERT_NE(nil, item.image);
+  [item configureCell:cell];
+  ASSERT_OCMOCK_VERIFY(niceDelegateMock);
+
+  id strictDelegateMock =
+      OCMStrictProtocolMock(@protocol(ContentSuggestionsItemDelegate));
+  item.delegate = strictDelegateMock;
+  id cellMock = OCMPartialMock(cell);
+  OCMExpect([cellMock setContentImage:item.image]);
+
+  // Action.
+  [item configureCell:cell];
+
+  // Tests.
+  EXPECT_OCMOCK_VERIFY(cellMock);
+  EXPECT_EQ(title, cell.titleLabel.text);
+}
+
+// Tests that the delegate is not called when |hasImage| is set to NO. If the
+// delegate is called an exception is raised.
+TEST(ContentSuggestionsItemTest, NoDelegateCallWhenHasNotImage) {
+  // Setup.
+  NSString* title = @"testTitle";
+  NSString* subtitle = @"testSubtitle";
+  GURL url = GURL("http://chromium.org");
+  // Strict mock. Raise exception if the load method is called.
+  id delegateMock =
+      OCMStrictProtocolMock(@protocol(ContentSuggestionsItemDelegate));
+  ContentSuggestionsItem* item =
+      [[ContentSuggestionsItem alloc] initWithType:0
+                                             title:title
+                                          subtitle:subtitle
+                                          delegate:delegateMock
+                                               url:url];
+  item.hasImage = NO;
+  ContentSuggestionsCell* cell = [[[item cellClass] alloc] init];
+
+  // Action.
+  [item configureCell:cell];
+}
+
+}  // namespace