Adds a first-run balloon to the Windows notification center.

chrome/browser/ui/views/status_icons/status_tray_win_unittest.cc

 // Copyright (c) 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.
 
 #include "chrome/browser/ui/views/status_icons/status_tray_win.h"
 
+#include <commctrl.h>
+
 #include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/status_icons/status_icon_observer.h"
 #include "chrome/browser/ui/views/status_icons/status_icon_win.h"
 #include "grit/chrome_unscaled_resources.h"
-#include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/base/models/simple_menu_model.h"
 #include "ui/base/resource/resource_bundle.h"
 
-class SkBitmap;
+class FakeStatusIconObserver : public StatusIconObserver {
+ public:
+  FakeStatusIconObserver()
+      : balloon_clicked_(false),
+        status_icon_click_count_(0) {}
+  virtual void OnStatusIconClicked() {
+    ++status_icon_click_count_;
+  }
+  virtual void OnBalloonClicked() { balloon_clicked_ = true; }
+  bool balloon_clicked() const { return balloon_clicked_; }
+  size_t status_icon_click_count() const {
+    return status_icon_click_count_;
+  }
 
-class MockStatusIconObserver : public StatusIconObserver {
- public:
-  MOCK_METHOD0(OnStatusIconClicked, void());
+ private:
+  size_t status_icon_click_count_;
+  bool balloon_clicked_;
 };
 
 TEST(StatusTrayWinTest, CreateTray) {
   // Just tests creation/destruction.
   StatusTrayWin tray;
 }
 
 TEST(StatusTrayWinTest, CreateIconAndMenu) {
   // Create an icon, set the images, tooltip, and context menu, then shut it
   // down.
   StatusTrayWin tray;
   StatusIcon* icon = tray.CreateStatusIcon();
   ui::ResourceBundle& rb = ui::ResourceBundle::GetSharedInstance();
   gfx::ImageSkia* image = rb.GetImageSkiaNamed(IDR_STATUS_TRAY_ICON);
   icon->SetImage(*image);
   icon->SetPressedImage(*image);
   icon->SetToolTip(ASCIIToUTF16("tool tip"));
   ui::SimpleMenuModel* menu = new ui::SimpleMenuModel(NULL);
   menu->AddItem(0, L"foo");
   icon->SetContextMenu(menu);
 }
 
 #if !defined(USE_AURA)  // http://crbug.com/156370
 TEST(StatusTrayWinTest, ClickOnIcon) {
   // Create an icon, send a fake click event, make sure observer is called.
   StatusTrayWin tray;
   StatusIconWin* icon = static_cast<StatusIconWin*>(tray.CreateStatusIcon());
-  MockStatusIconObserver observer;
+  FakeStatusIconObserver observer;
   icon->AddObserver(&observer);
-  EXPECT_CALL(observer, OnStatusIconClicked());
   // Mimic a click.
   tray.WndProc(NULL, icon->message_id(), icon->icon_id(), WM_LBUTTONDOWN);
   // Mimic a right-click - observer should not be called.
   tray.WndProc(NULL, icon->message_id(), icon->icon_id(), WM_RBUTTONDOWN);
+  EXPECT_EQ(1, observer.status_icon_click_count());
+  icon->RemoveObserver(&observer);
+}
+
+TEST(StatusTrayWinTest, ClickOnBalloon) {
+  // Create an icon, send a fake click event, make sure observer is called.
+  StatusTrayWin tray;
+  StatusIconWin* icon = static_cast<StatusIconWin*>(tray.CreateStatusIcon());
+  FakeStatusIconObserver observer;
+  icon->AddObserver(&observer);
+  // Mimic a click.
+  tray.WndProc(
+      NULL, icon->message_id(), icon->icon_id(), TB_INDETERMINATE);
+  EXPECT_TRUE(observer.balloon_clicked());
   icon->RemoveObserver(&observer);
 }
 #endif  // !defined(USE_AURA)