Index: chrome/browser/ui/views/status_icons/status_tray_win_unittest.cc |
diff --git a/chrome/browser/ui/views/status_icons/status_tray_win_unittest.cc b/chrome/browser/ui/views/status_icons/status_tray_win_unittest.cc |
index 7ff83094e65025554f9424cd6fb099b77b204480..e9f76432c773744df60a5b08727125edc65aed3b 100644 |
--- a/chrome/browser/ui/views/status_icons/status_tray_win_unittest.cc |
+++ b/chrome/browser/ui/views/status_icons/status_tray_win_unittest.cc |
@@ -14,13 +14,17 @@ |
#include "ui/base/models/simple_menu_model.h" |
#include "ui/base/resource/resource_bundle.h" |
-class SkBitmap; |
- |
class MockStatusIconObserver : public StatusIconObserver { |
public: |
MOCK_METHOD0(OnStatusIconClicked, void()); |
}; |
+class MockBalloonObserver : public StatusIconObserver { |
+ public: |
+ virtual void OnStatusIconClicked() {} |
+ MOCK_METHOD0(OnBalloonClicked, void()); |
sky
2013/06/20 17:22:01
gmock is generally frowned upon in chrome.
dewittj
2013/06/20 17:51:51
Okay, I was following the existing style but can e
|
+}; |
+ |
TEST(StatusTrayWinTest, CreateTray) { |
// Just tests creation/destruction. |
StatusTrayWin tray; |
@@ -55,4 +59,16 @@ TEST(StatusTrayWinTest, ClickOnIcon) { |
tray.WndProc(NULL, icon->message_id(), icon->icon_id(), WM_RBUTTONDOWN); |
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()); |
+ MockBalloonObserver observer; |
+ EXPECT_CALL(observer, OnBalloonClicked()); |
+ icon->AddObserver(&observer); |
+ // Mimic a click. |
+ tray.WndProc(NULL, icon->message_id(), icon->icon_id(), TB_INDETERMINATE); |
+ icon->RemoveObserver(&observer); |
+} |
#endif // !defined(USE_AURA) |