OLD | NEW |
| (Empty) |
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 "sync/notifier/p2p_notifier.h" | |
6 | |
7 #include <cstddef> | |
8 | |
9 #include "jingle/notifier/listener/fake_push_client.h" | |
10 #include "sync/internal_api/public/base/model_type.h" | |
11 #include "sync/internal_api/public/base/model_type_state_map.h" | |
12 #include "sync/notifier/fake_sync_notifier_observer.h" | |
13 #include "sync/notifier/object_id_state_map_test_util.h" | |
14 #include "testing/gtest/include/gtest/gtest.h" | |
15 | |
16 namespace syncer { | |
17 | |
18 namespace { | |
19 | |
20 class P2PNotifierTest : public testing::Test { | |
21 protected: | |
22 P2PNotifierTest() | |
23 : fake_push_client_(new notifier::FakePushClient()), | |
24 p2p_notifier_( | |
25 scoped_ptr<notifier::PushClient>(fake_push_client_), | |
26 NOTIFY_OTHERS), | |
27 next_sent_notification_to_reflect_(0) { | |
28 p2p_notifier_.RegisterHandler(&fake_observer_); | |
29 } | |
30 | |
31 virtual ~P2PNotifierTest() { | |
32 p2p_notifier_.UnregisterHandler(&fake_observer_); | |
33 } | |
34 | |
35 ModelTypeStateMap MakeStateMap(ModelTypeSet types) { | |
36 return ModelTypeSetToStateMap(types, std::string()); | |
37 } | |
38 | |
39 // Simulate receiving all the notifications we sent out since last | |
40 // time this was called. | |
41 void ReflectSentNotifications() { | |
42 const std::vector<notifier::Notification>& sent_notifications = | |
43 fake_push_client_->sent_notifications(); | |
44 for(size_t i = next_sent_notification_to_reflect_; | |
45 i < sent_notifications.size(); ++i) { | |
46 p2p_notifier_.OnIncomingNotification(sent_notifications[i]); | |
47 } | |
48 next_sent_notification_to_reflect_ = sent_notifications.size(); | |
49 } | |
50 | |
51 // Owned by |p2p_notifier_|. | |
52 notifier::FakePushClient* fake_push_client_; | |
53 P2PNotifier p2p_notifier_; | |
54 FakeSyncNotifierObserver fake_observer_; | |
55 | |
56 private: | |
57 size_t next_sent_notification_to_reflect_; | |
58 }; | |
59 | |
60 // Make sure the P2PNotificationTarget <-> string conversions work. | |
61 TEST_F(P2PNotifierTest, P2PNotificationTarget) { | |
62 for (int i = FIRST_NOTIFICATION_TARGET; | |
63 i <= LAST_NOTIFICATION_TARGET; ++i) { | |
64 P2PNotificationTarget target = static_cast<P2PNotificationTarget>(i); | |
65 const std::string& target_str = P2PNotificationTargetToString(target); | |
66 EXPECT_FALSE(target_str.empty()); | |
67 EXPECT_EQ(target, P2PNotificationTargetFromString(target_str)); | |
68 } | |
69 EXPECT_EQ(NOTIFY_SELF, P2PNotificationTargetFromString("unknown")); | |
70 } | |
71 | |
72 // Make sure notification targeting works correctly. | |
73 TEST_F(P2PNotifierTest, P2PNotificationDataIsTargeted) { | |
74 { | |
75 const P2PNotificationData notification_data( | |
76 "sender", NOTIFY_SELF, ModelTypeSet()); | |
77 EXPECT_TRUE(notification_data.IsTargeted("sender")); | |
78 EXPECT_FALSE(notification_data.IsTargeted("other1")); | |
79 EXPECT_FALSE(notification_data.IsTargeted("other2")); | |
80 } | |
81 { | |
82 const P2PNotificationData notification_data( | |
83 "sender", NOTIFY_OTHERS, ModelTypeSet()); | |
84 EXPECT_FALSE(notification_data.IsTargeted("sender")); | |
85 EXPECT_TRUE(notification_data.IsTargeted("other1")); | |
86 EXPECT_TRUE(notification_data.IsTargeted("other2")); | |
87 } | |
88 { | |
89 const P2PNotificationData notification_data( | |
90 "sender", NOTIFY_ALL, ModelTypeSet()); | |
91 EXPECT_TRUE(notification_data.IsTargeted("sender")); | |
92 EXPECT_TRUE(notification_data.IsTargeted("other1")); | |
93 EXPECT_TRUE(notification_data.IsTargeted("other2")); | |
94 } | |
95 } | |
96 | |
97 // Make sure the P2PNotificationData <-> string conversions work for a | |
98 // default-constructed P2PNotificationData. | |
99 TEST_F(P2PNotifierTest, P2PNotificationDataDefault) { | |
100 const P2PNotificationData notification_data; | |
101 EXPECT_TRUE(notification_data.IsTargeted("")); | |
102 EXPECT_FALSE(notification_data.IsTargeted("other1")); | |
103 EXPECT_FALSE(notification_data.IsTargeted("other2")); | |
104 EXPECT_TRUE(notification_data.GetChangedTypes().Empty()); | |
105 const std::string& notification_data_str = notification_data.ToString(); | |
106 EXPECT_EQ( | |
107 "{\"changedTypes\":[],\"notificationType\":\"notifySelf\"," | |
108 "\"senderId\":\"\"}", notification_data_str); | |
109 | |
110 P2PNotificationData notification_data_parsed; | |
111 EXPECT_TRUE(notification_data_parsed.ResetFromString(notification_data_str)); | |
112 EXPECT_TRUE(notification_data.Equals(notification_data_parsed)); | |
113 } | |
114 | |
115 // Make sure the P2PNotificationData <-> string conversions work for a | |
116 // non-default-constructed P2PNotificationData. | |
117 TEST_F(P2PNotifierTest, P2PNotificationDataNonDefault) { | |
118 const ModelTypeSet changed_types(BOOKMARKS, THEMES); | |
119 const P2PNotificationData notification_data( | |
120 "sender", NOTIFY_ALL, changed_types); | |
121 EXPECT_TRUE(notification_data.IsTargeted("sender")); | |
122 EXPECT_TRUE(notification_data.IsTargeted("other1")); | |
123 EXPECT_TRUE(notification_data.IsTargeted("other2")); | |
124 EXPECT_TRUE(notification_data.GetChangedTypes().Equals(changed_types)); | |
125 const std::string& notification_data_str = notification_data.ToString(); | |
126 EXPECT_EQ( | |
127 "{\"changedTypes\":[\"Bookmarks\",\"Themes\"]," | |
128 "\"notificationType\":\"notifyAll\"," | |
129 "\"senderId\":\"sender\"}", notification_data_str); | |
130 | |
131 P2PNotificationData notification_data_parsed; | |
132 EXPECT_TRUE(notification_data_parsed.ResetFromString(notification_data_str)); | |
133 EXPECT_TRUE(notification_data.Equals(notification_data_parsed)); | |
134 } | |
135 | |
136 // Set up the P2PNotifier, simulate a successful connection, and send | |
137 // a notification with the default target (NOTIFY_OTHERS). The | |
138 // observer should receive only a notification from the call to | |
139 // UpdateEnabledTypes(). | |
140 TEST_F(P2PNotifierTest, NotificationsBasic) { | |
141 const ModelTypeSet enabled_types(BOOKMARKS, PREFERENCES); | |
142 | |
143 p2p_notifier_.UpdateRegisteredIds(&fake_observer_, | |
144 ModelTypeSetToObjectIdSet(enabled_types)); | |
145 | |
146 p2p_notifier_.SetUniqueId("sender"); | |
147 | |
148 const char kEmail[] = "foo@bar.com"; | |
149 const char kToken[] = "token"; | |
150 p2p_notifier_.UpdateCredentials(kEmail, kToken); | |
151 { | |
152 notifier::Subscription expected_subscription; | |
153 expected_subscription.channel = kSyncP2PNotificationChannel; | |
154 expected_subscription.from = kEmail; | |
155 EXPECT_TRUE(notifier::SubscriptionListsEqual( | |
156 fake_push_client_->subscriptions(), | |
157 notifier::SubscriptionList(1, expected_subscription))); | |
158 } | |
159 EXPECT_EQ(kEmail, fake_push_client_->email()); | |
160 EXPECT_EQ(kToken, fake_push_client_->token()); | |
161 | |
162 ReflectSentNotifications(); | |
163 fake_push_client_->EnableNotifications(); | |
164 EXPECT_EQ(NO_NOTIFICATION_ERROR, | |
165 fake_observer_.GetNotificationsDisabledReason()); | |
166 | |
167 ReflectSentNotifications(); | |
168 EXPECT_EQ(1, fake_observer_.GetNotificationCount()); | |
169 EXPECT_THAT( | |
170 ModelTypeStateMapToObjectIdStateMap(MakeStateMap(enabled_types)), | |
171 Eq(fake_observer_.GetLastNotificationIdStateMap())); | |
172 EXPECT_EQ(REMOTE_NOTIFICATION, fake_observer_.GetLastNotificationSource()); | |
173 | |
174 // Sent with target NOTIFY_OTHERS so should not be propagated to | |
175 // |fake_observer_|. | |
176 { | |
177 ModelTypeSet changed_types(THEMES, APPS); | |
178 p2p_notifier_.SendNotification(changed_types); | |
179 } | |
180 | |
181 ReflectSentNotifications(); | |
182 EXPECT_EQ(1, fake_observer_.GetNotificationCount()); | |
183 } | |
184 | |
185 // Set up the P2PNotifier and send out notifications with various | |
186 // target settings. The notifications received by the observer should | |
187 // be consistent with the target settings. | |
188 TEST_F(P2PNotifierTest, SendNotificationData) { | |
189 const ModelTypeSet enabled_types(BOOKMARKS, PREFERENCES, THEMES); | |
190 const ModelTypeSet changed_types(THEMES, APPS); | |
191 const ModelTypeSet expected_types(THEMES); | |
192 | |
193 p2p_notifier_.UpdateRegisteredIds(&fake_observer_, | |
194 ModelTypeSetToObjectIdSet(enabled_types)); | |
195 | |
196 p2p_notifier_.SetUniqueId("sender"); | |
197 p2p_notifier_.UpdateCredentials("foo@bar.com", "fake_token"); | |
198 | |
199 ReflectSentNotifications(); | |
200 fake_push_client_->EnableNotifications(); | |
201 EXPECT_EQ(NO_NOTIFICATION_ERROR, | |
202 fake_observer_.GetNotificationsDisabledReason()); | |
203 | |
204 ReflectSentNotifications(); | |
205 EXPECT_EQ(1, fake_observer_.GetNotificationCount()); | |
206 EXPECT_THAT( | |
207 ModelTypeStateMapToObjectIdStateMap(MakeStateMap(enabled_types)), | |
208 Eq(fake_observer_.GetLastNotificationIdStateMap())); | |
209 EXPECT_EQ(REMOTE_NOTIFICATION, fake_observer_.GetLastNotificationSource()); | |
210 | |
211 // Should be dropped. | |
212 p2p_notifier_.SendNotificationDataForTest(P2PNotificationData()); | |
213 ReflectSentNotifications(); | |
214 EXPECT_EQ(1, fake_observer_.GetNotificationCount()); | |
215 | |
216 const ObjectIdStateMap& expected_ids = | |
217 ModelTypeStateMapToObjectIdStateMap(MakeStateMap(expected_types)); | |
218 | |
219 // Should be propagated. | |
220 p2p_notifier_.SendNotificationDataForTest( | |
221 P2PNotificationData("sender", NOTIFY_SELF, changed_types)); | |
222 ReflectSentNotifications(); | |
223 EXPECT_EQ(2, fake_observer_.GetNotificationCount()); | |
224 EXPECT_THAT( | |
225 expected_ids, | |
226 Eq(fake_observer_.GetLastNotificationIdStateMap())); | |
227 | |
228 // Should be dropped. | |
229 p2p_notifier_.SendNotificationDataForTest( | |
230 P2PNotificationData("sender2", NOTIFY_SELF, changed_types)); | |
231 ReflectSentNotifications(); | |
232 EXPECT_EQ(2, fake_observer_.GetNotificationCount()); | |
233 | |
234 // Should be dropped. | |
235 p2p_notifier_.SendNotificationDataForTest( | |
236 P2PNotificationData("sender", NOTIFY_SELF, ModelTypeSet())); | |
237 ReflectSentNotifications(); | |
238 EXPECT_EQ(2, fake_observer_.GetNotificationCount()); | |
239 | |
240 // Should be dropped. | |
241 p2p_notifier_.SendNotificationDataForTest( | |
242 P2PNotificationData("sender", NOTIFY_OTHERS, changed_types)); | |
243 ReflectSentNotifications(); | |
244 EXPECT_EQ(2, fake_observer_.GetNotificationCount()); | |
245 | |
246 // Should be propagated. | |
247 p2p_notifier_.SendNotificationDataForTest( | |
248 P2PNotificationData("sender2", NOTIFY_OTHERS, changed_types)); | |
249 ReflectSentNotifications(); | |
250 EXPECT_EQ(3, fake_observer_.GetNotificationCount()); | |
251 EXPECT_THAT( | |
252 expected_ids, | |
253 Eq(fake_observer_.GetLastNotificationIdStateMap())); | |
254 | |
255 // Should be dropped. | |
256 p2p_notifier_.SendNotificationDataForTest( | |
257 P2PNotificationData("sender2", NOTIFY_OTHERS, ModelTypeSet())); | |
258 ReflectSentNotifications(); | |
259 EXPECT_EQ(3, fake_observer_.GetNotificationCount()); | |
260 | |
261 // Should be propagated. | |
262 p2p_notifier_.SendNotificationDataForTest( | |
263 P2PNotificationData("sender", NOTIFY_ALL, changed_types)); | |
264 ReflectSentNotifications(); | |
265 EXPECT_EQ(4, fake_observer_.GetNotificationCount()); | |
266 EXPECT_THAT( | |
267 expected_ids, | |
268 Eq(fake_observer_.GetLastNotificationIdStateMap())); | |
269 | |
270 // Should be propagated. | |
271 p2p_notifier_.SendNotificationDataForTest( | |
272 P2PNotificationData("sender2", NOTIFY_ALL, changed_types)); | |
273 ReflectSentNotifications(); | |
274 EXPECT_EQ(5, fake_observer_.GetNotificationCount()); | |
275 EXPECT_THAT( | |
276 expected_ids, | |
277 Eq(fake_observer_.GetLastNotificationIdStateMap())); | |
278 | |
279 // Should be dropped. | |
280 p2p_notifier_.SendNotificationDataForTest( | |
281 P2PNotificationData("sender2", NOTIFY_ALL, ModelTypeSet())); | |
282 ReflectSentNotifications(); | |
283 EXPECT_EQ(5, fake_observer_.GetNotificationCount()); | |
284 } | |
285 | |
286 } // namespace | |
287 | |
288 } // namespace syncer | |
OLD | NEW |