Chromium Code Reviews| 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) |