chrome/browser/status_icons/status_icon_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 "chrome/browser/status_icons/status_icon.h"
   6
   7 #include "base/compiler_specific.h"
   8 #include "chrome/browser/status_icons/status_icon_observer.h"
   9 #include "testing/gmock/include/gmock/gmock.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 class MockStatusIconObserver : public StatusIconObserver {
  13  public:
  14   MOCK_METHOD0(OnStatusIconClicked, void());
  15 };
  16
  17 // Define pure virtual functions so we can test base class functionality.
  18 class TestStatusIcon : public StatusIcon {
  19  public:
  20   TestStatusIcon() {}
  21   void SetImage(const gfx::ImageSkia& image) override {}
  22   void SetToolTip(const base::string16& tool_tip) override {}
  23   void UpdatePlatformContextMenu(StatusIconMenuModel* menu) override {}
  24   void DisplayBalloon(const gfx::ImageSkia& icon,
  25                       const base::string16& title,
  26                       const base::string16& contents) override {}
  27 };
  28
  29 TEST(StatusIconTest, ObserverAdd) {
  30   // Make sure that observers are invoked when we click items.
  31   TestStatusIcon icon;
  32   MockStatusIconObserver observer, observer2;
  33   EXPECT_CALL(observer, OnStatusIconClicked()).Times(2);
  34   EXPECT_CALL(observer2, OnStatusIconClicked());
  35   icon.AddObserver(&observer);
  36   icon.DispatchClickEvent();
  37   icon.AddObserver(&observer2);
  38   icon.DispatchClickEvent();
  39   icon.RemoveObserver(&observer);
  40   icon.RemoveObserver(&observer2);
  41 }
  42
  43 TEST(StatusIconTest, ObserverRemove) {
  44   // Make sure that observers are no longer invoked after they are removed.
  45   TestStatusIcon icon;
  46   MockStatusIconObserver observer;
  47   EXPECT_CALL(observer, OnStatusIconClicked());
  48   icon.AddObserver(&observer);
  49   icon.DispatchClickEvent();
  50   icon.RemoveObserver(&observer);
  51   icon.DispatchClickEvent();
  52 }