src/chrome/browser/status_icons/status_tray_unittest.cc

   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 "base/compiler_specific.h"
   6 #include "base/strings/string_util.h"
   7 #include "base/strings/utf_string_conversions.h"
   8 #include "chrome/browser/status_icons/status_icon.h"
   9 #include "chrome/browser/status_icons/status_tray.h"
  10 #include "grit/chrome_unscaled_resources.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12 #include "ui/base/resource/resource_bundle.h"
  13 #include "ui/gfx/image/image_skia.h"
  14
  15 class MockStatusIcon : public StatusIcon {
  16   void SetImage(const gfx::ImageSkia& image) override {}
  17   void SetToolTip(const base::string16& tool_tip) override {}
  18   void DisplayBalloon(const gfx::ImageSkia& icon,
  19                       const base::string16& title,
  20                       const base::string16& contents) override {}
  21   void UpdatePlatformContextMenu(StatusIconMenuModel* menu) override {}
  22 };
  23
  24 class TestStatusTray : public StatusTray {
  25  public:
  26   StatusIcon* CreatePlatformStatusIcon(
  27       StatusIconType type,
  28       const gfx::ImageSkia& image,
  29       const base::string16& tool_tip) override {
  30     return new MockStatusIcon();
  31   }
  32
  33   const StatusIcons& GetStatusIconsForTest() const { return status_icons(); }
  34 };
  35
  36 TEST(StatusTrayTest, Create) {
  37   // Check for creation and leaks.
  38   TestStatusTray tray;
  39   ui::ResourceBundle& rb = ui::ResourceBundle::GetSharedInstance();
  40   gfx::ImageSkia* image = rb.GetImageSkiaNamed(IDR_STATUS_TRAY_ICON);
  41   tray.CreateStatusIcon(
  42       StatusTray::OTHER_ICON, *image, base::ASCIIToUTF16("tool tip"));
  43   EXPECT_EQ(1U, tray.GetStatusIconsForTest().size());
  44 }
  45
  46 // Make sure that removing an icon removes it from the list.
  47 TEST(StatusTrayTest, CreateRemove) {
  48   TestStatusTray tray;
  49   ui::ResourceBundle& rb = ui::ResourceBundle::GetSharedInstance();
  50   gfx::ImageSkia* image = rb.GetImageSkiaNamed(IDR_STATUS_TRAY_ICON);
  51   StatusIcon* icon = tray.CreateStatusIcon(
  52       StatusTray::OTHER_ICON, *image, base::ASCIIToUTF16("tool tip"));
  53   EXPECT_EQ(1U, tray.GetStatusIconsForTest().size());
  54   tray.RemoveStatusIcon(icon);
  55   EXPECT_EQ(0U, tray.GetStatusIconsForTest().size());
  56 }