OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/notifications/notification_test_util.h" | 5 #include "chrome/browser/notifications/notification_test_util.h" |
6 | 6 |
7 MockNotificationDelegate::MockNotificationDelegate(const std::string& id) | 7 MockNotificationDelegate::MockNotificationDelegate(const std::string& id) |
8 : id_(id) {} | 8 : id_(id) {} |
9 | 9 |
10 MockNotificationDelegate::~MockNotificationDelegate() {} | 10 MockNotificationDelegate::~MockNotificationDelegate() {} |
(...skipping 16 matching lines...) Expand all Loading... |
27 return notifications_[index].first; | 27 return notifications_[index].first; |
28 } | 28 } |
29 | 29 |
30 void StubNotificationUIManager::SetNotificationAddedCallback( | 30 void StubNotificationUIManager::SetNotificationAddedCallback( |
31 const base::Closure& callback) { | 31 const base::Closure& callback) { |
32 notification_added_callback_ = callback; | 32 notification_added_callback_ = callback; |
33 } | 33 } |
34 | 34 |
35 void StubNotificationUIManager::Add(const Notification& notification, | 35 void StubNotificationUIManager::Add(const Notification& notification, |
36 Profile* profile) { | 36 Profile* profile) { |
37 notifications_.push_back(std::make_pair(notification, profile)); | 37 notifications_.push_back(std::make_pair( |
| 38 notification, NotificationUIManager::GetProfileID(profile))); |
38 | 39 |
39 if (!notification_added_callback_.is_null()) { | 40 if (!notification_added_callback_.is_null()) { |
40 notification_added_callback_.Run(); | 41 notification_added_callback_.Run(); |
41 notification_added_callback_.Reset(); | 42 notification_added_callback_.Reset(); |
42 } | 43 } |
43 | 44 |
44 // Fire the Display() event on the delegate. | 45 // Fire the Display() event on the delegate. |
45 notification.delegate()->Display(); | 46 notification.delegate()->Display(); |
46 } | 47 } |
47 | 48 |
(...skipping 27 matching lines...) Expand all Loading... |
75 iter->first.delegate()->Close(false /* by_user */); | 76 iter->first.delegate()->Close(false /* by_user */); |
76 notifications_.erase(iter); | 77 notifications_.erase(iter); |
77 return true; | 78 return true; |
78 } | 79 } |
79 | 80 |
80 return false; | 81 return false; |
81 } | 82 } |
82 | 83 |
83 std::set<std::string> | 84 std::set<std::string> |
84 StubNotificationUIManager::GetAllIdsByProfileAndSourceOrigin( | 85 StubNotificationUIManager::GetAllIdsByProfileAndSourceOrigin( |
85 Profile* profile, | 86 ProfileID profile_id, |
86 const GURL& source) { | 87 const GURL& source) { |
87 std::set<std::string> delegate_ids; | 88 std::set<std::string> delegate_ids; |
88 for (const auto& pair : notifications_) { | 89 for (const auto& pair : notifications_) { |
89 if (pair.second == profile && pair.first.origin_url() == source) | 90 if (pair.second == profile_id && pair.first.origin_url() == source) |
90 delegate_ids.insert(pair.first.delegate_id()); | 91 delegate_ids.insert(pair.first.delegate_id()); |
91 } | 92 } |
92 return delegate_ids; | 93 return delegate_ids; |
93 } | 94 } |
94 | 95 |
95 std::set<std::string> StubNotificationUIManager::GetAllIdsByProfile( | 96 std::set<std::string> StubNotificationUIManager::GetAllIdsByProfile( |
96 Profile* profile) { | 97 ProfileID profile_id) { |
97 std::set<std::string> delegate_ids; | 98 std::set<std::string> delegate_ids; |
98 for (const auto& pair : notifications_) { | 99 for (const auto& pair : notifications_) { |
99 if (pair.second == profile) | 100 if (pair.second == profile_id) |
100 delegate_ids.insert(pair.first.delegate_id()); | 101 delegate_ids.insert(pair.first.delegate_id()); |
101 } | 102 } |
102 return delegate_ids; | 103 return delegate_ids; |
103 } | 104 } |
104 | 105 |
105 bool StubNotificationUIManager::CancelAllBySourceOrigin( | 106 bool StubNotificationUIManager::CancelAllBySourceOrigin( |
106 const GURL& source_origin) { | 107 const GURL& source_origin) { |
107 NOTIMPLEMENTED(); | 108 NOTIMPLEMENTED(); |
108 return false; | 109 return false; |
109 } | 110 } |
110 | 111 |
111 bool StubNotificationUIManager::CancelAllByProfile(ProfileID profile_id) { | 112 bool StubNotificationUIManager::CancelAllByProfile(ProfileID profile_id) { |
112 NOTIMPLEMENTED(); | 113 NOTIMPLEMENTED(); |
113 return false; | 114 return false; |
114 } | 115 } |
115 | 116 |
116 void StubNotificationUIManager::CancelAll() { | 117 void StubNotificationUIManager::CancelAll() { |
117 for (const auto& pair : notifications_) | 118 for (const auto& pair : notifications_) |
118 pair.first.delegate()->Close(false /* by_user */); | 119 pair.first.delegate()->Close(false /* by_user */); |
119 notifications_.clear(); | 120 notifications_.clear(); |
120 } | 121 } |
OLD | NEW |