status_tray_win_unittest.cc revision 5d1f7b1de12d16ceb2c938c56701a3e8bfa558f7
11320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// Copyright (c) 2012 The Chromium Authors. All rights reserved. 21320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// Use of this source code is governed by a BSD-style license that can be 31320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// found in the LICENSE file. 41320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 51320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "chrome/browser/ui/views/status_icons/status_tray_win.h" 61320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 71320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include <commctrl.h> 81320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 91320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/strings/string_util.h" 101320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/strings/utf_string_conversions.h" 111320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "chrome/browser/status_icons/status_icon_menu_model.h" 121320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "chrome/browser/status_icons/status_icon_observer.h" 13#include "chrome/browser/ui/views/status_icons/status_icon_win.h" 14#include "grit/chrome_unscaled_resources.h" 15#include "testing/gtest/include/gtest/gtest.h" 16#include "ui/base/resource/resource_bundle.h" 17#include "ui/gfx/image/image_skia.h" 18 19class FakeStatusIconObserver : public StatusIconObserver { 20 public: 21 FakeStatusIconObserver() 22 : balloon_clicked_(false), 23 status_icon_click_count_(0) {} 24 virtual void OnStatusIconClicked() { 25 ++status_icon_click_count_; 26 } 27 virtual void OnBalloonClicked() { balloon_clicked_ = true; } 28 bool balloon_clicked() const { return balloon_clicked_; } 29 size_t status_icon_click_count() const { 30 return status_icon_click_count_; 31 } 32 33 private: 34 size_t status_icon_click_count_; 35 bool balloon_clicked_; 36}; 37 38TEST(StatusTrayWinTest, CreateTray) { 39 // Just tests creation/destruction. 40 StatusTrayWin tray; 41} 42 43TEST(StatusTrayWinTest, CreateIconAndMenu) { 44 // Create an icon, set the images, tooltip, and context menu, then shut it 45 // down. 46 StatusTrayWin tray; 47 ui::ResourceBundle& rb = ui::ResourceBundle::GetSharedInstance(); 48 gfx::ImageSkia* image = rb.GetImageSkiaNamed(IDR_STATUS_TRAY_ICON); 49 StatusIcon* icon = tray.CreateStatusIcon( 50 StatusTray::OTHER_ICON, *image, base::ASCIIToUTF16("tool tip")); 51 icon->SetPressedImage(*image); 52 scoped_ptr<StatusIconMenuModel> menu(new StatusIconMenuModel(NULL)); 53 menu->AddItem(0, L"foo"); 54 icon->SetContextMenu(menu.Pass()); 55} 56 57#if !defined(USE_AURA) // http://crbug.com/156370 58TEST(StatusTrayWinTest, ClickOnIcon) { 59 // Create an icon, send a fake click event, make sure observer is called. 60 StatusTrayWin tray; 61 ui::ResourceBundle& rb = ui::ResourceBundle::GetSharedInstance(); 62 gfx::ImageSkia* image = rb.GetImageSkiaNamed(IDR_STATUS_TRAY_ICON); 63 64 StatusIconWin* icon = static_cast<StatusIconWin*>(tray.CreateStatusIcon( 65 StatusTray::OTHER_ICON, *image, base::ASCIIToUTF16("tool tip"))); 66 FakeStatusIconObserver observer; 67 icon->AddObserver(&observer); 68 // Mimic a click. 69 tray.WndProc(NULL, icon->message_id(), icon->icon_id(), WM_LBUTTONDOWN); 70 // Mimic a right-click - observer should not be called. 71 tray.WndProc(NULL, icon->message_id(), icon->icon_id(), WM_RBUTTONDOWN); 72 EXPECT_EQ(1, observer.status_icon_click_count()); 73 icon->RemoveObserver(&observer); 74} 75 76TEST(StatusTrayWinTest, ClickOnBalloon) { 77 // Create an icon, send a fake click event, make sure observer is called. 78 StatusTrayWin tray; 79 ui::ResourceBundle& rb = ui::ResourceBundle::GetSharedInstance(); 80 gfx::ImageSkia* image = rb.GetImageSkiaNamed(IDR_STATUS_TRAY_ICON); 81 82 StatusIconWin* icon = static_cast<StatusIconWin*>(tray.CreateStatusIcon( 83 StatusTray::OTHER_ICON, *image, base::ASCIIToUTF16("tool tip"))); 84 FakeStatusIconObserver observer; 85 icon->AddObserver(&observer); 86 // Mimic a click. 87 tray.WndProc( 88 NULL, icon->message_id(), icon->icon_id(), TB_INDETERMINATE); 89 EXPECT_TRUE(observer.balloon_clicked()); 90 icon->RemoveObserver(&observer); 91} 92 93TEST(StatusTrayWinTest, HandleOldIconId) { 94 StatusTrayWin tray; 95 ui::ResourceBundle& rb = ui::ResourceBundle::GetSharedInstance(); 96 gfx::ImageSkia* image = rb.GetImageSkiaNamed(IDR_STATUS_TRAY_ICON); 97 98 StatusIconWin* icon = static_cast<StatusIconWin*>(tray.CreateStatusIcon( 99 StatusTray::OTHER_ICON, *image, base::ASCIIToUTF16("tool tip"))); 100 UINT message_id = icon->message_id(); 101 UINT icon_id = icon->icon_id(); 102 103 tray.RemoveStatusIcon(icon); 104 tray.WndProc(NULL, message_id, icon_id, WM_LBUTTONDOWN); 105} 106#endif // !defined(USE_AURA) 107