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

ios/chrome/app/spotlight/spotlight_manager_unittest.mm

Index: ios/chrome/app/spotlight/spotlight_manager_unittest.mm
diff --git a/ios/chrome/app/spotlight/spotlight_manager_unittest.mm b/ios/chrome/app/spotlight/spotlight_manager_unittest.mm
new file mode 100644
index 0000000000000000000000000000000000000000..52e85697f0223f676e2b76e13e658e3cacd4b7b7
--- /dev/null
+++ b/ios/chrome/app/spotlight/spotlight_manager_unittest.mm
@@ -0,0 +1,204 @@
+// Copyright 2015 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.
+
+#include <memory>
+
+#import <CoreSpotlight/CoreSpotlight.h>
+#import <Foundation/Foundation.h>
+
+#include "base/location.h"
+#include "base/mac/scoped_nsobject.h"
+#include "base/message_loop/message_loop.h"
+#include "base/single_thread_task_runner.h"
+#include "base/strings/sys_string_conversions.h"
+#include "base/strings/utf_string_conversions.h"
+#include "base/threading/thread_task_runner_handle.h"
+#include "components/bookmarks/browser/bookmark_model.h"
+#include "components/bookmarks/test/bookmark_test_helpers.h"
+#include "components/bookmarks/test/test_bookmark_client.h"
+#include "components/favicon/core/favicon_client.h"
+#include "components/favicon/core/favicon_service.h"
+#include "components/favicon/core/large_icon_service.h"
+#include "components/favicon_base/fallback_icon_style.h"
+#import "ios/chrome/app/spotlight/bookmarks_spotlight_manager.h"
+#import "ios/chrome/app/spotlight/spotlight_manager.h"
+#import "ios/chrome/app/spotlight/spotlight_util.h"
+#include "ios/chrome/browser/bookmarks/bookmark_model_factory.h"
+#include "ios/public/provider/chrome/browser/chrome_browser_provider.h"
+#include "ios/public/provider/chrome/browser/spotlight/spotlight_provider.h"
+#import "net/base/mac/url_conversions.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/gtest_mac.h"
+
+const char kDummyIconUrl[] = "http://www.example.com/touch_icon.png";
+
+favicon_base::FaviconRawBitmapResult CreateTestBitmap(int w, int h) {
+  favicon_base::FaviconRawBitmapResult result;
+  result.expired = false;
+
+  CGRect rect = CGRectMake(0, 0, w, h);
+  UIGraphicsBeginImageContext(rect.size);
+  CGContextRef context = UIGraphicsGetCurrentContext();
+  CGContextFillRect(context, rect);
+  UIImage* favicon = UIGraphicsGetImageFromCurrentImageContext();
+  UIGraphicsEndImageContext();
+
+  NSData* png = UIImagePNGRepresentation(favicon);
+  scoped_refptr<base::RefCountedBytes> data(new base::RefCountedBytes(
+      static_cast<const unsigned char*>([png bytes]), [png length]));
+
+  result.bitmap_data = data;
+  result.pixel_size = gfx::Size(w, h);
+  result.icon_url = GURL(kDummyIconUrl);
+  result.icon_type = favicon_base::TOUCH_ICON;
+  CHECK(result.is_valid());
+  return result;
+}
+
+// A mock FaviconService that emits pre-programmed response.
+class MockFaviconService : public favicon::FaviconService {
+ public:
+  MockFaviconService() : FaviconService(nullptr, nullptr) {}
+
+  ~MockFaviconService() override {}
+
+  base::CancelableTaskTracker::TaskId GetLargestRawFaviconForPageURL(
+      const GURL& page_url,
+      const std::vector<int>& icon_types,
+      int minimum_size_in_pixels,
+      const favicon_base::FaviconRawBitmapCallback& callback,
+      base::CancelableTaskTracker* tracker) override {
+    favicon_base::FaviconRawBitmapResult mock_result = CreateTestBitmap(24, 24);
+    return tracker->PostTask(base::ThreadTaskRunnerHandle::Get().get(),
+                             FROM_HERE, base::Bind(callback, mock_result));
+  }
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MockFaviconService);
+};
+
+// This class provides access to LargeIconService internals, using the current
+// thread's task runner for testing.
+class TestLargeIconService : public favicon::LargeIconService {
+ public:
+  explicit TestLargeIconService(MockFaviconService* mock_favicon_service)
+      : LargeIconService(mock_favicon_service,
+                         base::ThreadTaskRunnerHandle::Get()) {}
+  ~TestLargeIconService() override {}
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(TestLargeIconService);
+};
+
+class SpotlightManagerTest : public testing::Test {
+ protected:
+  SpotlightManagerTest() {
+    mock_favicon_service_.reset(new MockFaviconService());
+    large_icon_service_.reset(
+        new TestLargeIconService(mock_favicon_service_.get()));
+    model_ = bookmarks::TestBookmarkClient::CreateModel();
+    mock_favicon_service_.reset(new MockFaviconService());
+    large_icon_service_.reset(
+        new TestLargeIconService(mock_favicon_service_.get()));
+    bookmarksSpotlightManager_.reset([[BookmarksSpotlightManager alloc]
+        initWithLargeIconService:large_icon_service_.get()
+                   bookmarkModel:model_.get()]);
+  }
+
+  base::MessageLoop loop_;
+  std::unique_ptr<MockFaviconService> mock_favicon_service_;
+  std::unique_ptr<TestLargeIconService> large_icon_service_;
+  base::CancelableTaskTracker cancelable_task_tracker_;
+  std::unique_ptr<bookmarks::BookmarkModel> model_;
+  base::scoped_nsobject<BookmarksSpotlightManager> bookmarksSpotlightManager_;
+};
+
+TEST_F(SpotlightManagerTest, testSpotlightID) {
+  // Creating CSSearchableItem requires Spotlight to be available on the device.
+  if (!spotlight::IsSpotlightAvailable())
+    return;
+  GURL url("http://url.com");
+  NSString* spotlightId =
+      [bookmarksSpotlightManager_ spotlightIDForURL:url title:@"title"];
+  NSString* expectedSpotlightId =
+      [NSString stringWithFormat:@"%@.9c6b643df2a0c990",
+                                 spotlight::StringFromSpotlightDomain(
+                                     spotlight::DOMAIN_BOOKMARKS)];
+  EXPECT_NSEQ(spotlightId, expectedSpotlightId);
+}
+
+TEST_F(SpotlightManagerTest, testParentKeywordsForNode) {
+  const bookmarks::BookmarkNode* root = model_->bookmark_bar_node();
+  static const std::string model_string("a 1:[ b c ] d 2:[ 21:[ e ] f g ] h");
+  bookmarks::test::AddNodesFromModelString(model_.get(), root, model_string);
+  const bookmarks::BookmarkNode* eNode =
+      root->GetChild(3)->GetChild(0)->GetChild(0);
+  base::scoped_nsobject<NSMutableArray> keywords([[NSMutableArray alloc] init]);
+  [bookmarksSpotlightManager_ getParentKeywordsForNode:eNode inArray:keywords];
+  EXPECT_EQ([keywords count], 2u);
+  EXPECT_TRUE([[keywords objectAtIndex:0] isEqualToString:@"21"]);
+  EXPECT_TRUE([[keywords objectAtIndex:1] isEqualToString:@"2"]);
+}
+
+TEST_F(SpotlightManagerTest, testBookmarksCreateSpotlightItemsWithUrl) {
+  // Creating CSSearchableItem requires Spotlight to be available on the device.
+  if (!spotlight::IsSpotlightAvailable())
+    return;
+
+  const bookmarks::BookmarkNode* root = model_->bookmark_bar_node();
+  static const std::string model_string("a 1:[ b c ] d 2:[ 21:[ e ] f g ] h");
+  bookmarks::test::AddNodesFromModelString(model_.get(), root, model_string);
+  const bookmarks::BookmarkNode* eNode =
+      root->GetChild(3)->GetChild(0)->GetChild(0);
+
+  NSString* spotlightID = [bookmarksSpotlightManager_
+      spotlightIDForURL:eNode->url()
+                  title:base::SysUTF16ToNSString(eNode->GetTitle())];
+  base::scoped_nsobject<NSMutableArray> keywords([[NSMutableArray alloc] init]);
+  [bookmarksSpotlightManager_ getParentKeywordsForNode:eNode inArray:keywords];
+  NSArray* items = [bookmarksSpotlightManager_
+      spotlightItemsWithURL:eNode->url()
+                    favicon:nil
+               defaultTitle:base::SysUTF16ToNSString(eNode->GetTitle())];
+  EXPECT_TRUE([items count] == 1);
+  CSSearchableItem* item = [items objectAtIndex:0];
+  EXPECT_NSEQ([item uniqueIdentifier], spotlightID);
+  EXPECT_NSEQ([[item attributeSet] title], @"e");
+  EXPECT_NSEQ([[[item attributeSet] URL] absoluteString], @"http://e.com/");
+  [bookmarksSpotlightManager_ addKeywords:keywords.get() toSearchableItem:item];
+  // We use the set intersection to verify that the item from the Spotlight
+  // manager
+  // contains all the newly added Keywords.
+  NSMutableSet* spotlightManagerKeywords =
+      [NSMutableSet setWithArray:[[item attributeSet] keywords]];
+  NSSet* testModelKeywords = [NSSet setWithArray:keywords];
+  [spotlightManagerKeywords intersectSet:testModelKeywords];
+  EXPECT_NSEQ(spotlightManagerKeywords, testModelKeywords);
+}
+
+TEST_F(SpotlightManagerTest, testDefaultKeywordsExist) {
+  // Creating CSSearchableItem requires Spotlight to be available on the device.
+  if (!spotlight::IsSpotlightAvailable())
+    return;
+
+  const bookmarks::BookmarkNode* root = model_->bookmark_bar_node();
+  static const std::string model_string("a 1:[ b c ] d 2:[ 21:[ e ] f g ] h");
+  bookmarks::test::AddNodesFromModelString(model_.get(), root, model_string);
+  const bookmarks::BookmarkNode* aNode = root->GetChild(0);
+  NSArray* items = [bookmarksSpotlightManager_
+      spotlightItemsWithURL:aNode->url()
+                    favicon:nil
+               defaultTitle:base::SysUTF16ToNSString(aNode->GetTitle())];
+  EXPECT_TRUE([items count] == 1);
+  CSSearchableItem* item = [items objectAtIndex:0];
+  NSSet* spotlightManagerKeywords =
+      [NSSet setWithArray:[[item attributeSet] keywords]];
+  EXPECT_TRUE([spotlightManagerKeywords count] > 0);
+  // Check static/hardcoded keywords exist
+  NSSet* hardCodedKeywordsSet =
+      [NSSet setWithArray:ios::GetChromeBrowserProvider()
+                              ->GetSpotlightProvider()
+                              ->GetAdditionalKeywords()];
+  EXPECT_TRUE([hardCodedKeywordsSet isSubsetOfSet:spotlightManagerKeywords]);
+}