chrome/browser/notifications/notification_test_util.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/notifications/notification_test_util.h"
   6
   7 MockNotificationDelegate::MockNotificationDelegate(const std::string& id)
   8     : id_(id) {}
   9
  10 MockNotificationDelegate::~MockNotificationDelegate() {}
  11
  12 std::string MockNotificationDelegate::id() const { return id_; }
  13
  14 // -----------------------------------------------------------------------------
  15
  16 StubNotificationUIManager::StubNotificationUIManager() {}
  17
  18 StubNotificationUIManager::~StubNotificationUIManager() {}
  19
  20 unsigned int StubNotificationUIManager::GetNotificationCount() const {
  21   return notifications_.size();
  22 }
  23
  24 const Notification& StubNotificationUIManager::GetNotificationAt(
  25     unsigned int index) const {
  26   DCHECK_GT(GetNotificationCount(), index);
  27   return notifications_[index].first;
  28 }
  29
  30 void StubNotificationUIManager::SetNotificationAddedCallback(
  31     const base::Closure& callback) {
  32   notification_added_callback_ = callback;
  33 }
  34
  35 void StubNotificationUIManager::Add(const Notification& notification,
  36                                     Profile* profile) {
  37   notifications_.push_back(std::make_pair(notification, profile));
  38
  39   if (!notification_added_callback_.is_null()) {
  40     notification_added_callback_.Run();
  41     notification_added_callback_.Reset();
  42   }
  43
  44   // Fire the Display() event on the delegate.
  45   notification.delegate()->Display();
  46 }
  47
  48 bool StubNotificationUIManager::Update(const Notification& notification,
  49                                        Profile* profile) {
  50   return false;
  51 }
  52
  53 const Notification* StubNotificationUIManager::FindById(
  54     const std::string& delegate_id,
  55     ProfileID profile_id) const {
  56   auto iter = notifications_.begin();
  57   for (; iter != notifications_.end(); ++iter) {
  58     if (iter->first.delegate_id() != delegate_id || iter->second != profile_id)
  59       continue;
  60
  61     return &iter->first;
  62   }
  63
  64   return nullptr;
  65 }
  66
  67 bool StubNotificationUIManager::CancelById(const std::string& delegate_id,
  68                                            ProfileID profile_id) {
  69   auto iter = notifications_.begin();
  70   for (; iter != notifications_.end(); ++iter) {
  71     if (iter->first.delegate_id() != delegate_id ||
  72         iter->second != profile_id)
  73       continue;
  74
  75     iter->first.delegate()->Close(false /* by_user */);
  76     notifications_.erase(iter);
  77     return true;
  78   }
  79
  80   return false;
  81 }
  82
  83 std::set<std::string>
  84 StubNotificationUIManager::GetAllIdsByProfileAndSourceOrigin(
  85     Profile* profile,
  86     const GURL& source) {
  87   std::set<std::string> delegate_ids;
  88   for (const auto& pair : notifications_) {
  89     if (pair.second == profile && pair.first.origin_url() == source)
  90       delegate_ids.insert(pair.first.delegate_id());
  91   }
  92   return delegate_ids;
  93 }
  94
  95 std::set<std::string> StubNotificationUIManager::GetAllIdsByProfile(
  96     Profile* profile) {
  97   std::set<std::string> delegate_ids;
  98   for (const auto& pair : notifications_) {
  99     if (pair.second == profile)
 100       delegate_ids.insert(pair.first.delegate_id());
 101   }
 102   return delegate_ids;
 103 }
 104
 105 bool StubNotificationUIManager::CancelAllBySourceOrigin(
 106     const GURL& source_origin) {
 107   NOTIMPLEMENTED();
 108   return false;
 109 }
 110
 111 bool StubNotificationUIManager::CancelAllByProfile(ProfileID profile_id) {
 112   NOTIMPLEMENTED();
 113   return false;
 114 }
 115
 116 void StubNotificationUIManager::CancelAll() {
 117   for (const auto& pair : notifications_)
 118     pair.first.delegate()->Close(false /* by_user */);
 119   notifications_.clear();
 120 }