Web Intent Picker UI (implemented as a constrained dialog, linux only)

chrome/browser/ui/intents/web_intent_picker_controller_unittest.cc

Index: chrome/browser/ui/intents/web_intent_picker_controller_unittest.cc
diff --git a/chrome/browser/ui/intents/web_intent_picker_controller_unittest.cc b/chrome/browser/ui/intents/web_intent_picker_controller_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..89a2e1413328f3f4d032131f9fb63dbaefd40433
--- /dev/null
+++ b/chrome/browser/ui/intents/web_intent_picker_controller_unittest.cc
@@ -0,0 +1,273 @@
+// Copyright (c) 2011 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 "testing/gtest/include/gtest/gtest.h"
+
+#include "base/bind.h"
+#include "base/callback.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/message_loop.h"
+#include "base/utf_string_conversions.h"
+#include "chrome/browser/favicon/favicon_service.h"
+#include "chrome/browser/intents/web_intent_data.h"
+#include "chrome/browser/intents/web_intents_registry.h"
+#include "chrome/browser/intents/web_intents_registry_factory.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/ui/intents/web_intent_picker.h"
+#include "chrome/browser/ui/intents/web_intent_picker_controller.h"
+#include "chrome/browser/ui/intents/web_intent_picker_factory.h"
+#include "chrome/browser/ui/tab_contents/test_tab_contents_wrapper.h"
+#include "chrome/browser/webdata/web_data_service.h"
+#include "chrome/test/base/testing_profile.h"
+#include "content/browser/browser_thread.h"
+#include "content/browser/tab_contents/constrained_window.h"
+#include "content/browser/tab_contents/tab_contents.h"
+#include "content/browser/tab_contents/test_tab_contents.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "ui/gfx/codec/png_codec.h"
+
+using testing::_;
+using testing::AtMost;
+using testing::DoAll;
+using testing::Invoke;
+using testing::Return;
+using testing::SaveArg;
+
+namespace {
+
+const string16 kAction1(ASCIIToUTF16("http://www.example.com/share"));
+const string16 kAction2(ASCIIToUTF16("http://www.example.com/foobar"));
+const string16 kType(ASCIIToUTF16("image/png"));
+const GURL kServiceURL1("http://www.google.com");
+const GURL kServiceURL2("http://www.chromium.org");
+const GURL kServiceURL3("http://www.test.com");
+
+// Fill the given bmp with valid png data.
+void FillDataToBitmap(int w, int h, SkBitmap* bmp) {
+  bmp->setConfig(SkBitmap::kARGB_8888_Config, w, h);
+  bmp->allocPixels();
+
+  unsigned char* src_data =
+      reinterpret_cast<unsigned char*>(bmp->getAddr32(0, 0));
+  for (int i = 0; i < w * h; i++) {
+    src_data[i * 4 + 0] = static_cast<unsigned char>(i % 255);
+    src_data[i * 4 + 1] = static_cast<unsigned char>(i % 255);
+    src_data[i * 4 + 2] = static_cast<unsigned char>(i % 255);
+    src_data[i * 4 + 3] = static_cast<unsigned char>(i % 255);
+  }
+}
+
+// Fill the given data buffer with valid png data.
+void FillBitmap(int w, int h, std::vector<unsigned char>* output) {
+  SkBitmap bitmap;
+  FillDataToBitmap(w, h, &bitmap);
+  gfx::PNGCodec::EncodeBGRASkBitmap(bitmap, false, output);
+}
+
+GURL MakeFaviconURL(const GURL& url) {
+  return GURL(url.spec() + "/favicon.png");
+}
+
+MATCHER_P(VectorIsOfSize, n, "") {
+  return arg.size() == static_cast<size_t>(n);
+}
+
+} // namespace
+
+class WebIntentPickerMock : public WebIntentPicker {
+ public:
+  MOCK_METHOD1(SetServiceURLs, void(const std::vector<GURL>& urls));
+  MOCK_METHOD2(SetServiceIcon, void(size_t index, const SkBitmap& icon));
+  MOCK_METHOD1(SetDefaultServiceIcon, void(size_t index));
+  MOCK_METHOD0(Show, void(void));
+  MOCK_METHOD0(Close, void(void));
+};
+
+class WebIntentPickerFactoryMock : public WebIntentPickerFactory {
+ public:
+  MOCK_METHOD2(Create,
+               WebIntentPicker*(TabContents* tab_contents,
+                                WebIntentPickerDelegate* delegate));
+  MOCK_METHOD1(ClosePicker, void(WebIntentPicker* picker));
+};
+
+class TestWebIntentPickerController : public WebIntentPickerController {
+ public:
+  TestWebIntentPickerController(TabContents* tab_contents,
+                                WebIntentPickerFactory* factory)
+      : WebIntentPickerController(tab_contents, factory) {
+  }
+
+  MOCK_METHOD1(OnServiceChosen, void(size_t index));
+  MOCK_METHOD0(OnCancelled, void(void));
+
+  // helper functions to forward to the base class.
+  void BaseOnServiceChosen(size_t index) {
+    WebIntentPickerController::OnServiceChosen(index);
+  }
+
+  void BaseOnCancelled() {
+    WebIntentPickerController::OnCancelled();
+  }
+};
+
+class WebIntentPickerControllerTest : public TabContentsWrapperTestHarness {
+ public:
+  WebIntentPickerControllerTest()
+      : ui_thread_(BrowserThread::UI, &message_loop_),
+        db_thread_(BrowserThread::DB, &message_loop_),
+        picker_factory_(NULL),
+        delegate_(NULL) {
+  }
+
+  virtual void SetUp() {
+    TabContentsWrapperTestHarness::SetUp();
+
+    profile_->CreateFaviconService();
+    profile_->CreateWebDataService(true);
+    web_data_service_ = profile_->GetWebDataService(Profile::EXPLICIT_ACCESS);
+    favicon_service_ = profile_->GetFaviconService(Profile::EXPLICIT_ACCESS);
+    WebIntentsRegistry *registry =
+        WebIntentsRegistryFactory::GetForProfile(profile_.get());
+    registry->Initialize(web_data_service_);
+
+    picker_factory_ = new WebIntentPickerFactoryMock();
+    controller_.reset(new TestWebIntentPickerController(contents(),
+                                                  picker_factory_));
+  }
+
+  virtual void TearDown() {
+    controller_.reset();
+
+    TabContentsWrapperTestHarness::TearDown();
+  }
+
+ protected:
+  void AddWebIntentService(const string16& action,
+                           const GURL& service_url) {
+    WebIntentData web_intent_data;
+    web_intent_data.action = action;
+    web_intent_data.type = kType;
+    web_intent_data.service_url = service_url;
+    web_data_service_->AddWebIntent(web_intent_data);
+  }
+
+  void AddFaviconForURL(const GURL& url) {
+    std::vector<unsigned char> image_data;
+    FillBitmap(16, 16, &image_data);
+
+    favicon_service_->SetFavicon(url,
+                                 MakeFaviconURL(url),
+                                 image_data,
+                                 history::FAVICON);
+  }
+
+  void SetPickerExpectations(int expected_service_count,
+                             int expected_default_favicons) {
+    EXPECT_CALL(*picker_factory_, Create(_, _)).
+        WillOnce(DoAll(SaveArg<1>(&delegate_), Return(&picker_)));
+    EXPECT_CALL(picker_,
+                SetServiceURLs(VectorIsOfSize(expected_service_count))).
+        Times(1);
+    EXPECT_CALL(picker_, SetDefaultServiceIcon(_)).
+        Times(expected_default_favicons);
+  }
+
+  void CheckPendingAsync() {
+    if (controller_->pending_async_count() > 0) {
+      MessageLoop::current()->PostTask(
+          FROM_HERE,
+          base::Bind(&WebIntentPickerControllerTest::CheckPendingAsync,
+                     base::Unretained(this)));
+      return;
+    }
+
+    MessageLoop::current()->Quit();
+  }
+
+  void WaitForDialogToShow() {
+    CheckPendingAsync();
+    MessageLoop::current()->Run();
+  }
+
+  BrowserThread ui_thread_;
+  BrowserThread db_thread_;
+  WebIntentPickerMock picker_;
+  WebIntentPickerFactoryMock* picker_factory_; // controller_ takes ownership.
+  scoped_ptr<TestWebIntentPickerController> controller_;
+  WebIntentPickerDelegate* delegate_;
+  WebDataService* web_data_service_;
+  FaviconService* favicon_service_;
+};
+
+TEST_F(WebIntentPickerControllerTest, ShowDialogWith3Services) {
+  SetPickerExpectations(3, 3);
+  AddWebIntentService(kAction1, kServiceURL1);
+  AddWebIntentService(kAction1, kServiceURL2);
+  AddWebIntentService(kAction1, kServiceURL3);
+
+  controller_->ShowDialog(kAction1, kType);
+  WaitForDialogToShow();
+}
+
+TEST_F(WebIntentPickerControllerTest, ShowDialogWithNoServices) {
+  SetPickerExpectations(0, 0);
+
+  EXPECT_CALL(picker_, SetServiceIcon(_, _)).Times(0);
+
+  controller_->ShowDialog(kAction1, kType);
+  WaitForDialogToShow();
+}
+
+// TODO(binji) SetServiceIcon isn't called unless I create the HistoryService,
+// but when I do, the test hangs...
+TEST_F(WebIntentPickerControllerTest, DISABLED_ShowFavicon) {
+  SetPickerExpectations(3, 2);
+  AddWebIntentService(kAction1, kServiceURL1);
+  AddWebIntentService(kAction1, kServiceURL2);
+  AddWebIntentService(kAction1, kServiceURL3);
+  AddFaviconForURL(kServiceURL1);
+  AddFaviconForURL(kServiceURL3);
+
+  EXPECT_CALL(picker_, SetServiceIcon(0, _)).Times(1);
+  EXPECT_CALL(picker_, SetDefaultServiceIcon(1)).Times(1);
+  EXPECT_CALL(picker_, SetServiceIcon(2, _)).Times(1);
+
+  controller_->ShowDialog(kAction1, kType);
+  WaitForDialogToShow();
+}
+
+TEST_F(WebIntentPickerControllerTest, ChooseService) {
+  SetPickerExpectations(2, 2);
+  AddWebIntentService(kAction1, kServiceURL1);
+  AddWebIntentService(kAction1, kServiceURL2);
+
+  EXPECT_CALL(*controller_, OnServiceChosen(0))
+      .WillOnce(Invoke(controller_.get(),
+                       &TestWebIntentPickerController::BaseOnServiceChosen));
+  EXPECT_CALL(*controller_, OnCancelled())
+      .Times(0);
+  EXPECT_CALL(*picker_factory_, ClosePicker(_));
+
+  controller_->ShowDialog(kAction1, kType);
+  WaitForDialogToShow();
+  delegate_->OnServiceChosen(0);
+}
+
+TEST_F(WebIntentPickerControllerTest, Cancel) {
+  SetPickerExpectations(2, 2);
+  AddWebIntentService(kAction1, kServiceURL1);
+  AddWebIntentService(kAction1, kServiceURL2);
+
+  EXPECT_CALL(*controller_, OnServiceChosen(0))
+      .Times(0);
+  EXPECT_CALL(*controller_, OnCancelled())
+      .WillOnce(Invoke(controller_.get(),
+                       &TestWebIntentPickerController::BaseOnCancelled));
+  EXPECT_CALL(*picker_factory_, ClosePicker(_));
+
+  controller_->ShowDialog(kAction1, kType);
+  WaitForDialogToShow();
+  delegate_->OnCancelled();
+}