status_tray_win_unittest.cc revision 3551c9c881056c480085172ff9840cab31610854 
1// Copyright (c) 2012 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "chrome/browser/ui/views/status_icons/status_tray_win.h"
6
7#include <commctrl.h>
8
9#include "base/strings/string_util.h"
10#include "base/strings/utf_string_conversions.h"
11#include "chrome/browser/status_icons/status_icon_observer.h"
12#include "chrome/browser/ui/views/status_icons/status_icon_win.h"
13#include "grit/chrome_unscaled_resources.h"
14#include "testing/gtest/include/gtest/gtest.h"
15#include "ui/base/models/simple_menu_model.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, ASCIIToUTF16("tool tip"));
51  icon->SetPressedImage(*image);
52  ui::SimpleMenuModel* menu = new ui::SimpleMenuModel(NULL);
53  menu->AddItem(0, L"foo");
54  icon->SetContextMenu(menu);
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, 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, 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, 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