src/content/browser/notification_service_impl_unittest.cc

   1 // Copyright (c) 2011 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 "content/browser/notification_service_impl.h"
   6
   7 #include "content/public/browser/notification_observer.h"
   8 #include "content/public/browser/notification_registrar.h"
   9 #include "content/public/browser/notification_types.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 namespace content {
  13
  14 namespace {
  15
  16 // Bogus class to act as a NotificationSource for the messages.
  17 class TestSource {};
  18
  19 class TestObserver : public NotificationObserver {
  20  public:
  21   TestObserver() : notification_count_(0) {}
  22
  23   int notification_count() const { return notification_count_; }
  24
  25   void Observe(int type,
  26                const NotificationSource& source,
  27                const NotificationDetails& details) override {
  28     ++notification_count_;
  29   }
  30
  31  private:
  32   int notification_count_;
  33 };
  34
  35 const int kNotification1 = 1;
  36 const int kNotification2 = 2;
  37
  38 }  // namespace
  39
  40
  41 class NotificationServiceImplTest : public testing::Test {
  42  protected:
  43   NotificationRegistrar registrar_;
  44 };
  45
  46 TEST_F(NotificationServiceImplTest, Basic) {
  47   TestSource test_source;
  48   TestSource other_source;
  49
  50   // Check the equality operators defined for NotificationSource
  51   EXPECT_TRUE(Source<TestSource>(&test_source) ==
  52               Source<TestSource>(&test_source));
  53   EXPECT_TRUE(Source<TestSource>(&test_source) !=
  54               Source<TestSource>(&other_source));
  55
  56   TestObserver all_types_all_sources;
  57   TestObserver idle_all_sources;
  58   TestObserver all_types_test_source;
  59   TestObserver idle_test_source;
  60
  61   // Make sure it doesn't freak out when there are no observers.
  62   NotificationService* service = NotificationService::current();
  63   service->Notify(kNotification1,
  64                   Source<TestSource>(&test_source),
  65                   NotificationService::NoDetails());
  66
  67   registrar_.Add(&all_types_all_sources, NOTIFICATION_ALL,
  68                  NotificationService::AllSources());
  69   registrar_.Add(&idle_all_sources, kNotification1,
  70                  NotificationService::AllSources());
  71   registrar_.Add(&all_types_test_source, NOTIFICATION_ALL,
  72                  Source<TestSource>(&test_source));
  73   registrar_.Add(&idle_test_source, kNotification1,
  74                  Source<TestSource>(&test_source));
  75
  76   EXPECT_EQ(0, all_types_all_sources.notification_count());
  77   EXPECT_EQ(0, idle_all_sources.notification_count());
  78   EXPECT_EQ(0, all_types_test_source.notification_count());
  79   EXPECT_EQ(0, idle_test_source.notification_count());
  80
  81   service->Notify(kNotification1,
  82                   Source<TestSource>(&test_source),
  83                   NotificationService::NoDetails());
  84
  85   EXPECT_EQ(1, all_types_all_sources.notification_count());
  86   EXPECT_EQ(1, idle_all_sources.notification_count());
  87   EXPECT_EQ(1, all_types_test_source.notification_count());
  88   EXPECT_EQ(1, idle_test_source.notification_count());
  89
  90   service->Notify(kNotification2,
  91                   Source<TestSource>(&test_source),
  92                   NotificationService::NoDetails());
  93
  94   EXPECT_EQ(2, all_types_all_sources.notification_count());
  95   EXPECT_EQ(1, idle_all_sources.notification_count());
  96   EXPECT_EQ(2, all_types_test_source.notification_count());
  97   EXPECT_EQ(1, idle_test_source.notification_count());
  98
  99   service->Notify(kNotification1,
 100                   Source<TestSource>(&other_source),
 101                   NotificationService::NoDetails());
 102
 103   EXPECT_EQ(3, all_types_all_sources.notification_count());
 104   EXPECT_EQ(2, idle_all_sources.notification_count());
 105   EXPECT_EQ(2, all_types_test_source.notification_count());
 106   EXPECT_EQ(1, idle_test_source.notification_count());
 107
 108   service->Notify(kNotification2,
 109                   Source<TestSource>(&other_source),
 110                   NotificationService::NoDetails());
 111
 112   EXPECT_EQ(4, all_types_all_sources.notification_count());
 113   EXPECT_EQ(2, idle_all_sources.notification_count());
 114   EXPECT_EQ(2, all_types_test_source.notification_count());
 115   EXPECT_EQ(1, idle_test_source.notification_count());
 116
 117   // Try send with NULL source.
 118   service->Notify(kNotification1,
 119                   NotificationService::AllSources(),
 120                   NotificationService::NoDetails());
 121
 122   EXPECT_EQ(5, all_types_all_sources.notification_count());
 123   EXPECT_EQ(3, idle_all_sources.notification_count());
 124   EXPECT_EQ(2, all_types_test_source.notification_count());
 125   EXPECT_EQ(1, idle_test_source.notification_count());
 126
 127   registrar_.RemoveAll();
 128
 129   service->Notify(kNotification1,
 130                   Source<TestSource>(&test_source),
 131                   NotificationService::NoDetails());
 132
 133   EXPECT_EQ(5, all_types_all_sources.notification_count());
 134   EXPECT_EQ(3, idle_all_sources.notification_count());
 135   EXPECT_EQ(2, all_types_test_source.notification_count());
 136   EXPECT_EQ(1, idle_test_source.notification_count());
 137 }
 138
 139 TEST_F(NotificationServiceImplTest, MultipleRegistration) {
 140   TestSource test_source;
 141
 142   TestObserver idle_test_source;
 143
 144   NotificationService* service = NotificationService::current();
 145
 146   registrar_.Add(&idle_test_source, kNotification1,
 147                  Source<TestSource>(&test_source));
 148   registrar_.Add(&idle_test_source, NOTIFICATION_ALL,
 149                  Source<TestSource>(&test_source));
 150
 151   service->Notify(kNotification1,
 152                   Source<TestSource>(&test_source),
 153                   NotificationService::NoDetails());
 154   EXPECT_EQ(2, idle_test_source.notification_count());
 155
 156   registrar_.Remove(&idle_test_source, kNotification1,
 157                     Source<TestSource>(&test_source));
 158
 159   service->Notify(kNotification1,
 160                  Source<TestSource>(&test_source),
 161                  NotificationService::NoDetails());
 162   EXPECT_EQ(3, idle_test_source.notification_count());
 163
 164   registrar_.Remove(&idle_test_source, NOTIFICATION_ALL,
 165                     Source<TestSource>(&test_source));
 166
 167   service->Notify(kNotification1,
 168                   Source<TestSource>(&test_source),
 169                   NotificationService::NoDetails());
 170   EXPECT_EQ(3, idle_test_source.notification_count());
 171 }
 172
 173 }  // namespace content