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 "sync/notifier/invalidation_notifier.h" | 5 #include "sync/notifier/invalidation_notifier.h" |
6 | 6 |
7 #include "base/memory/scoped_ptr.h" | 7 #include "base/memory/scoped_ptr.h" |
8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
9 #include "jingle/notifier/base/fake_base_task.h" | 9 #include "jingle/notifier/base/fake_base_task.h" |
10 #include "jingle/notifier/base/notifier_options.h" | 10 #include "jingle/notifier/base/notifier_options.h" |
11 #include "jingle/notifier/listener/fake_push_client.h" | 11 #include "jingle/notifier/listener/fake_push_client.h" |
12 #include "net/url_request/url_request_test_util.h" | 12 #include "net/url_request/url_request_test_util.h" |
13 #include "sync/internal_api/public/base/model_type.h" | 13 #include "sync/internal_api/public/base/model_type.h" |
14 #include "sync/internal_api/public/base/model_type_payload_map.h" | 14 #include "sync/internal_api/public/base/model_type_payload_map.h" |
15 #include "sync/internal_api/public/util/weak_handle.h" | 15 #include "sync/internal_api/public/util/weak_handle.h" |
16 #include "sync/notifier/invalidation_state_tracker.h" | 16 #include "sync/notifier/invalidation_state_tracker.h" |
17 #include "sync/notifier/mock_invalidation_state_tracker.h" | 17 #include "sync/notifier/mock_invalidation_state_tracker.h" |
18 #include "sync/notifier/mock_sync_notifier_observer.h" | 18 #include "sync/notifier/mock_sync_notifier_observer.h" |
19 #include "testing/gmock/include/gmock/gmock.h" | 19 #include "testing/gmock/include/gmock/gmock.h" |
20 #include "testing/gtest/include/gtest/gtest.h" | 20 #include "testing/gtest/include/gtest/gtest.h" |
21 | 21 |
22 namespace syncer { | 22 namespace syncer { |
23 | 23 |
24 namespace { | 24 namespace { |
25 | 25 |
26 using ::testing::InSequence; | 26 using ::testing::InSequence; |
27 using ::testing::StrictMock; | 27 using ::testing::StrictMock; |
28 | 28 |
29 const char kHandlerName[] = "MockObserver"; | |
30 | |
29 class InvalidationNotifierTest : public testing::Test { | 31 class InvalidationNotifierTest : public testing::Test { |
30 protected: | 32 protected: |
31 virtual void TearDown() { | 33 virtual void TearDown() { |
32 if (invalidation_notifier_.get()) | 34 if (invalidation_notifier_.get()) |
33 ResetNotifier(); | 35 ResetNotifier(); |
34 } | 36 } |
35 | 37 |
36 // Constructs an InvalidationNotifier, places it in | 38 // Constructs an InvalidationNotifier and places it in |
msw
2012/08/03 23:30:46
nit: line breaks :)
akalin
2012/08/07 07:25:19
Done.
| |
37 // |invalidation_notifier_|, and adds |mock_observer_| as an observer. This | 39 // |invalidation_notifier_|. This remains in place until either |
38 // remains in place until either TearDown (automatic) or ResetNotifier | 40 // TearDown (automatic) or ResetNotifier (manual) is called. |
39 // (manual) is called. | 41 void CreateNotifier(const std::string& initial_invalidation_state) { |
40 void CreateAndObserveNotifier( | |
41 const std::string& initial_invalidation_state) { | |
42 notifier::NotifierOptions notifier_options; | 42 notifier::NotifierOptions notifier_options; |
43 // Note: URLRequestContextGetters are ref-counted. | 43 // Note: URLRequestContextGetters are ref-counted. |
44 notifier_options.request_context_getter = | 44 notifier_options.request_context_getter = |
45 new TestURLRequestContextGetter(message_loop_.message_loop_proxy()); | 45 new TestURLRequestContextGetter(message_loop_.message_loop_proxy()); |
46 invalidation_notifier_.reset( | 46 invalidation_notifier_.reset( |
47 new InvalidationNotifier( | 47 new InvalidationNotifier( |
48 scoped_ptr<notifier::PushClient>(new notifier::FakePushClient()), | 48 scoped_ptr<notifier::PushClient>(new notifier::FakePushClient()), |
49 InvalidationVersionMap(), | 49 InvalidationVersionMap(), |
50 initial_invalidation_state, | 50 initial_invalidation_state, |
51 MakeWeakHandle(mock_tracker_.AsWeakPtr()), | 51 MakeWeakHandle(mock_tracker_.AsWeakPtr()), |
52 "fake_client_info")); | 52 "fake_client_info")); |
53 } | 53 } |
54 | 54 |
55 void ResetNotifier() { | 55 void ResetNotifier() { |
56 invalidation_notifier_->UpdateRegisteredIds(&mock_observer_, ObjectIdSet()); | 56 invalidation_notifier_->SetHandler(kHandlerName, NULL); |
57 // Stopping the invalidation notifier stops its scheduler, which deletes any | 57 // Stopping the invalidation notifier stops its scheduler, which deletes any |
58 // pending tasks without running them. Some tasks "run and delete" another | 58 // pending tasks without running them. Some tasks "run and delete" another |
59 // task, so they must be run in order to avoid leaking the inner task. | 59 // task, so they must be run in order to avoid leaking the inner task. |
60 // Stopping does not schedule any tasks, so it's both necessary and | 60 // Stopping does not schedule any tasks, so it's both necessary and |
61 // sufficient to drain the task queue before stopping the notifier. | 61 // sufficient to drain the task queue before stopping the notifier. |
62 message_loop_.RunAllPending(); | 62 message_loop_.RunAllPending(); |
63 invalidation_notifier_.reset(); | 63 invalidation_notifier_.reset(); |
64 } | 64 } |
65 | 65 |
66 MessageLoopForIO message_loop_; | 66 MessageLoopForIO message_loop_; |
67 scoped_ptr<InvalidationNotifier> invalidation_notifier_; | 67 scoped_ptr<InvalidationNotifier> invalidation_notifier_; |
68 StrictMock<MockInvalidationStateTracker> mock_tracker_; | 68 StrictMock<MockInvalidationStateTracker> mock_tracker_; |
69 StrictMock<MockSyncNotifierObserver> mock_observer_; | 69 StrictMock<MockSyncNotifierObserver> mock_observer_; |
70 notifier::FakeBaseTask fake_base_task_; | 70 notifier::FakeBaseTask fake_base_task_; |
71 }; | 71 }; |
72 | 72 |
73 TEST_F(InvalidationNotifierTest, Basic) { | 73 TEST_F(InvalidationNotifierTest, Basic) { |
74 CreateAndObserveNotifier("fake_state"); | |
75 InSequence dummy; | 74 InSequence dummy; |
76 | 75 |
77 ModelTypeSet models(PREFERENCES, BOOKMARKS, AUTOFILL); | 76 CreateNotifier("fake_state"); |
78 invalidation_notifier_->UpdateRegisteredIds( | |
79 &mock_observer_, ModelTypeSetToObjectIdSet(models)); | |
80 | 77 |
78 const ModelTypeSet models(PREFERENCES, BOOKMARKS, AUTOFILL); | |
81 const ModelTypePayloadMap& type_payloads = | 79 const ModelTypePayloadMap& type_payloads = |
82 ModelTypePayloadMapFromEnumSet(models, "payload"); | 80 ModelTypePayloadMapFromEnumSet(models, "payload"); |
83 EXPECT_CALL(mock_observer_, OnNotificationsEnabled()); | 81 EXPECT_CALL(mock_observer_, OnNotificationsEnabled()); |
84 EXPECT_CALL(mock_observer_, OnIncomingNotification( | 82 EXPECT_CALL(mock_observer_, OnIncomingNotification( |
85 ModelTypePayloadMapToObjectIdPayloadMap(type_payloads), | 83 ModelTypePayloadMapToObjectIdPayloadMap(type_payloads), |
86 REMOTE_NOTIFICATION)); | 84 REMOTE_NOTIFICATION)); |
87 EXPECT_CALL(mock_observer_, | 85 EXPECT_CALL(mock_observer_, |
88 OnNotificationsDisabled(TRANSIENT_NOTIFICATION_ERROR)); | 86 OnNotificationsDisabled(TRANSIENT_NOTIFICATION_ERROR)); |
89 EXPECT_CALL(mock_observer_, | 87 EXPECT_CALL(mock_observer_, |
90 OnNotificationsDisabled(NOTIFICATION_CREDENTIALS_REJECTED)); | 88 OnNotificationsDisabled(NOTIFICATION_CREDENTIALS_REJECTED)); |
89 | |
90 invalidation_notifier_->SetHandler(kHandlerName, &mock_observer_); | |
91 | |
92 invalidation_notifier_->UpdateRegisteredIds( | |
93 kHandlerName, ModelTypeSetToObjectIdSet(models)); | |
94 | |
91 // Note no expectation on mock_tracker_, as we initialized with | 95 // Note no expectation on mock_tracker_, as we initialized with |
92 // non-empty initial_invalidation_state above. | 96 // non-empty initial_invalidation_state above. |
93 | 97 |
94 // TODO(tim): This call should be a no-op, Remove once bug 124140 and | 98 // TODO(tim): This call should be a no-op, Remove once bug 124140 and |
95 // associated issues are fixed. | 99 // associated issues are fixed. |
96 invalidation_notifier_->SetStateDeprecated("fake_state"); | 100 invalidation_notifier_->SetStateDeprecated("fake_state"); |
97 invalidation_notifier_->SetUniqueId("fake_id"); | 101 invalidation_notifier_->SetUniqueId("fake_id"); |
98 invalidation_notifier_->UpdateCredentials("foo@bar.com", "fake_token"); | 102 invalidation_notifier_->UpdateCredentials("foo@bar.com", "fake_token"); |
99 | 103 |
100 invalidation_notifier_->OnNotificationsEnabled(); | 104 invalidation_notifier_->OnNotificationsEnabled(); |
101 | 105 |
102 invalidation_notifier_->OnInvalidate( | 106 invalidation_notifier_->OnInvalidate( |
103 ModelTypePayloadMapToObjectIdPayloadMap(type_payloads)); | 107 ModelTypePayloadMapToObjectIdPayloadMap(type_payloads)); |
104 | 108 |
105 invalidation_notifier_->OnNotificationsDisabled( | 109 invalidation_notifier_->OnNotificationsDisabled( |
106 TRANSIENT_NOTIFICATION_ERROR); | 110 TRANSIENT_NOTIFICATION_ERROR); |
107 invalidation_notifier_->OnNotificationsDisabled( | 111 invalidation_notifier_->OnNotificationsDisabled( |
108 NOTIFICATION_CREDENTIALS_REJECTED); | 112 NOTIFICATION_CREDENTIALS_REJECTED); |
109 } | 113 } |
110 | 114 |
111 TEST_F(InvalidationNotifierTest, MigrateState) { | 115 TEST_F(InvalidationNotifierTest, MigrateState) { |
112 CreateAndObserveNotifier(std::string()); | |
113 InSequence dummy; | 116 InSequence dummy; |
114 | 117 |
118 CreateNotifier(std::string()); | |
119 | |
115 EXPECT_CALL(mock_tracker_, SetInvalidationState("fake_state")); | 120 EXPECT_CALL(mock_tracker_, SetInvalidationState("fake_state")); |
116 invalidation_notifier_->SetStateDeprecated("fake_state"); | 121 invalidation_notifier_->SetStateDeprecated("fake_state"); |
117 | 122 |
118 // Should do nothing. | 123 // Should do nothing. |
119 invalidation_notifier_->SetStateDeprecated("spurious_fake_state"); | 124 invalidation_notifier_->SetStateDeprecated("spurious_fake_state"); |
120 | 125 |
121 // Pretend Chrome shut down. | 126 // Pretend Chrome shut down. |
122 ResetNotifier(); | 127 ResetNotifier(); |
123 | 128 |
124 CreateAndObserveNotifier("fake_state"); | 129 CreateNotifier("fake_state"); |
125 // Should do nothing. | 130 // Should do nothing. |
126 invalidation_notifier_->SetStateDeprecated("more_spurious_fake_state"); | 131 invalidation_notifier_->SetStateDeprecated("more_spurious_fake_state"); |
127 } | 132 } |
128 | 133 |
129 } // namespace | 134 } // namespace |
130 | 135 |
131 } // namespace syncer | 136 } // namespace syncer |
OLD | NEW |