1 // Copyright 2014 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.
5 #include "components/gcm_driver/gcm_client_impl.h"
7 #include "base/command_line.h"
8 #include "base/files/file_path.h"
9 #include "base/files/file_util.h"
10 #include "base/files/scoped_temp_dir.h"
11 #include "base/memory/scoped_ptr.h"
12 #include "base/strings/string_number_conversions.h"
13 #include "base/test/test_mock_time_task_runner.h"
14 #include "base/thread_task_runner_handle.h"
15 #include "base/time/clock.h"
16 #include "base/timer/timer.h"
17 #include "google_apis/gcm/base/fake_encryptor.h"
18 #include "google_apis/gcm/base/mcs_message.h"
19 #include "google_apis/gcm/base/mcs_util.h"
20 #include "google_apis/gcm/engine/fake_connection_factory.h"
21 #include "google_apis/gcm/engine/fake_connection_handler.h"
22 #include "google_apis/gcm/engine/gservices_settings.h"
23 #include "google_apis/gcm/monitoring/gcm_stats_recorder.h"
24 #include "google_apis/gcm/protocol/android_checkin.pb.h"
25 #include "google_apis/gcm/protocol/checkin.pb.h"
26 #include "google_apis/gcm/protocol/mcs.pb.h"
27 #include "net/url_request/test_url_fetcher_factory.h"
28 #include "net/url_request/url_fetcher_delegate.h"
29 #include "net/url_request/url_request_test_util.h"
30 #include "testing/gtest/include/gtest/gtest.h"
39 REGISTRATION_COMPLETED
,
40 UNREGISTRATION_COMPLETED
,
47 const char kChromeVersion
[] = "45.0.0.1";
48 const uint64 kDeviceAndroidId
= 54321;
49 const uint64 kDeviceSecurityToken
= 12345;
50 const uint64 kDeviceAndroidId2
= 11111;
51 const uint64 kDeviceSecurityToken2
= 2222;
52 const int64 kSettingsCheckinInterval
= 16 * 60 * 60;
53 const char kAppId
[] = "app_id";
54 const char kSender
[] = "project_id";
55 const char kSender2
[] = "project_id2";
56 const char kSender3
[] = "project_id3";
57 const char kRegistrationResponsePrefix
[] = "token=";
58 const char kUnregistrationResponsePrefix
[] = "deleted=";
60 const char kInstanceID
[] = "iid_1";
61 const char kScope
[] = "GCM";
62 const char kDeleteTokenResponse
[] = "token=foo";
64 // Helper for building arbitrary data messages.
65 MCSMessage
BuildDownstreamMessage(
66 const std::string
& project_id
,
67 const std::string
& app_id
,
68 const std::map
<std::string
, std::string
>& data
) {
69 mcs_proto::DataMessageStanza data_message
;
70 data_message
.set_from(project_id
);
71 data_message
.set_category(app_id
);
72 for (std::map
<std::string
, std::string
>::const_iterator iter
= data
.begin();
75 mcs_proto::AppData
* app_data
= data_message
.add_app_data();
76 app_data
->set_key(iter
->first
);
77 app_data
->set_value(iter
->second
);
79 return MCSMessage(kDataMessageStanzaTag
, data_message
);
82 GCMClient::AccountTokenInfo
MakeAccountToken(const std::string
& email
,
83 const std::string
& token
) {
84 GCMClient::AccountTokenInfo account_token
;
85 account_token
.email
= email
;
86 account_token
.access_token
= token
;
90 std::map
<std::string
, std::string
> MakeEmailToTokenMap(
91 const std::vector
<GCMClient::AccountTokenInfo
>& account_tokens
) {
92 std::map
<std::string
, std::string
> email_token_map
;
93 for (std::vector
<GCMClient::AccountTokenInfo
>::const_iterator iter
=
94 account_tokens
.begin(); iter
!= account_tokens
.end(); ++iter
) {
95 email_token_map
[iter
->email
] = iter
->access_token
;
97 return email_token_map
;
100 class FakeMCSClient
: public MCSClient
{
102 FakeMCSClient(base::Clock
* clock
,
103 ConnectionFactory
* connection_factory
,
105 GCMStatsRecorder
* recorder
);
106 ~FakeMCSClient() override
;
107 void Login(uint64 android_id
, uint64 security_token
) override
;
108 void SendMessage(const MCSMessage
& message
) override
;
110 uint64
last_android_id() const { return last_android_id_
; }
111 uint64
last_security_token() const { return last_security_token_
; }
112 uint8
last_message_tag() const { return last_message_tag_
; }
113 const mcs_proto::DataMessageStanza
& last_data_message_stanza() const {
114 return last_data_message_stanza_
;
118 uint64 last_android_id_
;
119 uint64 last_security_token_
;
120 uint8 last_message_tag_
;
121 mcs_proto::DataMessageStanza last_data_message_stanza_
;
124 FakeMCSClient::FakeMCSClient(base::Clock
* clock
,
125 ConnectionFactory
* connection_factory
,
127 GCMStatsRecorder
* recorder
)
128 : MCSClient("", clock
, connection_factory
, gcm_store
, recorder
),
129 last_android_id_(0u),
130 last_security_token_(0u),
131 last_message_tag_(kNumProtoTypes
) {
134 FakeMCSClient::~FakeMCSClient() {
137 void FakeMCSClient::Login(uint64 android_id
, uint64 security_token
) {
138 last_android_id_
= android_id
;
139 last_security_token_
= security_token
;
142 void FakeMCSClient::SendMessage(const MCSMessage
& message
) {
143 last_message_tag_
= message
.tag();
144 if (last_message_tag_
== kDataMessageStanzaTag
) {
145 last_data_message_stanza_
.CopyFrom(
146 reinterpret_cast<const mcs_proto::DataMessageStanza
&>(
147 message
.GetProtobuf()));
151 class AutoAdvancingTestClock
: public base::Clock
{
153 explicit AutoAdvancingTestClock(base::TimeDelta auto_increment_time_delta
);
154 ~AutoAdvancingTestClock() override
;
156 base::Time
Now() override
;
157 void Advance(TimeDelta delta
);
158 int call_count() const { return call_count_
; }
162 base::TimeDelta auto_increment_time_delta_
;
165 DISALLOW_COPY_AND_ASSIGN(AutoAdvancingTestClock
);
168 AutoAdvancingTestClock::AutoAdvancingTestClock(
169 base::TimeDelta auto_increment_time_delta
)
170 : call_count_(0), auto_increment_time_delta_(auto_increment_time_delta
) {
173 AutoAdvancingTestClock::~AutoAdvancingTestClock() {
176 base::Time
AutoAdvancingTestClock::Now() {
178 now_
+= auto_increment_time_delta_
;
182 void AutoAdvancingTestClock::Advance(base::TimeDelta delta
) {
186 class FakeGCMInternalsBuilder
: public GCMInternalsBuilder
{
188 FakeGCMInternalsBuilder(base::TimeDelta clock_step
);
189 ~FakeGCMInternalsBuilder() override
;
191 scoped_ptr
<base::Clock
> BuildClock() override
;
192 scoped_ptr
<MCSClient
> BuildMCSClient(const std::string
& version
,
194 ConnectionFactory
* connection_factory
,
196 GCMStatsRecorder
* recorder
) override
;
197 scoped_ptr
<ConnectionFactory
> BuildConnectionFactory(
198 const std::vector
<GURL
>& endpoints
,
199 const net::BackoffEntry::Policy
& backoff_policy
,
200 const scoped_refptr
<net::HttpNetworkSession
>& gcm_network_session
,
201 const scoped_refptr
<net::HttpNetworkSession
>& http_network_session
,
202 net::NetLog
* net_log
,
203 GCMStatsRecorder
* recorder
) override
;
206 base::TimeDelta clock_step_
;
209 FakeGCMInternalsBuilder::FakeGCMInternalsBuilder(base::TimeDelta clock_step
)
210 : clock_step_(clock_step
) {
213 FakeGCMInternalsBuilder::~FakeGCMInternalsBuilder() {}
215 scoped_ptr
<base::Clock
> FakeGCMInternalsBuilder::BuildClock() {
216 return make_scoped_ptr
<base::Clock
>(new AutoAdvancingTestClock(clock_step_
));
219 scoped_ptr
<MCSClient
> FakeGCMInternalsBuilder::BuildMCSClient(
220 const std::string
& version
,
222 ConnectionFactory
* connection_factory
,
224 GCMStatsRecorder
* recorder
) {
225 return make_scoped_ptr
<MCSClient
>(new FakeMCSClient(clock
,
231 scoped_ptr
<ConnectionFactory
> FakeGCMInternalsBuilder::BuildConnectionFactory(
232 const std::vector
<GURL
>& endpoints
,
233 const net::BackoffEntry::Policy
& backoff_policy
,
234 const scoped_refptr
<net::HttpNetworkSession
>& gcm_network_session
,
235 const scoped_refptr
<net::HttpNetworkSession
>& http_network_session
,
236 net::NetLog
* net_log
,
237 GCMStatsRecorder
* recorder
) {
238 return make_scoped_ptr
<ConnectionFactory
>(new FakeConnectionFactory());
243 class GCMClientImplTest
: public testing::Test
,
244 public GCMClient::Delegate
{
247 ~GCMClientImplTest() override
;
249 void SetUp() override
;
251 void SetUpUrlFetcherFactory();
253 void BuildGCMClient(base::TimeDelta clock_step
);
254 void InitializeGCMClient();
255 void StartGCMClient();
256 void Register(const std::string
& app_id
,
257 const std::vector
<std::string
>& senders
);
258 void Unregister(const std::string
& app_id
);
259 void ReceiveMessageFromMCS(const MCSMessage
& message
);
260 void ReceiveOnMessageSentToMCS(
261 const std::string
& app_id
,
262 const std::string
& message_id
,
263 const MCSClient::MessageSendStatus status
);
264 void CompleteCheckin(uint64 android_id
,
265 uint64 security_token
,
266 const std::string
& digest
,
267 const std::map
<std::string
, std::string
>& settings
);
268 void CompleteRegistration(const std::string
& registration_id
);
269 void CompleteUnregistration(const std::string
& app_id
);
270 void VerifyPendingRequestFetcherDeleted();
272 bool ExistsRegistration(const std::string
& app_id
) const;
273 void AddRegistration(const std::string
& app_id
,
274 const std::vector
<std::string
>& sender_ids
,
275 const std::string
& registration_id
);
277 // GCMClient::Delegate overrides (for verification).
278 void OnRegisterFinished(const linked_ptr
<RegistrationInfo
>& registration_info
,
279 const std::string
& registration_id
,
280 GCMClient::Result result
) override
;
281 void OnUnregisterFinished(
282 const linked_ptr
<RegistrationInfo
>& registration_info
,
283 GCMClient::Result result
) override
;
284 void OnSendFinished(const std::string
& app_id
,
285 const std::string
& message_id
,
286 GCMClient::Result result
) override
{}
287 void OnMessageReceived(const std::string
& registration_id
,
288 const GCMClient::IncomingMessage
& message
) override
;
289 void OnMessagesDeleted(const std::string
& app_id
) override
;
290 void OnMessageSendError(
291 const std::string
& app_id
,
292 const gcm::GCMClient::SendErrorDetails
& send_error_details
) override
;
293 void OnSendAcknowledged(const std::string
& app_id
,
294 const std::string
& message_id
) override
;
295 void OnGCMReady(const std::vector
<AccountMapping
>& account_mappings
,
296 const base::Time
& last_token_fetch_time
) override
;
297 void OnActivityRecorded() override
{}
298 void OnConnected(const net::IPEndPoint
& ip_endpoint
) override
{}
299 void OnDisconnected() override
{}
301 GCMClientImpl
* gcm_client() const { return gcm_client_
.get(); }
302 GCMClientImpl::State
gcm_client_state() const {
303 return gcm_client_
->state_
;
305 FakeMCSClient
* mcs_client() const {
306 return reinterpret_cast<FakeMCSClient
*>(gcm_client_
->mcs_client_
.get());
308 ConnectionFactory
* connection_factory() const {
309 return gcm_client_
->connection_factory_
.get();
312 const GCMClientImpl::CheckinInfo
& device_checkin_info() const {
313 return gcm_client_
->device_checkin_info_
;
316 void reset_last_event() {
318 last_app_id_
.clear();
319 last_registration_id_
.clear();
320 last_message_id_
.clear();
321 last_result_
= GCMClient::UNKNOWN_ERROR
;
322 last_account_mappings_
.clear();
323 last_token_fetch_time_
= base::Time();
326 LastEvent
last_event() const { return last_event_
; }
327 const std::string
& last_app_id() const { return last_app_id_
; }
328 const std::string
& last_registration_id() const {
329 return last_registration_id_
;
331 const std::string
& last_message_id() const { return last_message_id_
; }
332 GCMClient::Result
last_result() const { return last_result_
; }
333 const GCMClient::IncomingMessage
& last_message() const {
334 return last_message_
;
336 const GCMClient::SendErrorDetails
& last_error_details() const {
337 return last_error_details_
;
339 const base::Time
& last_token_fetch_time() const {
340 return last_token_fetch_time_
;
342 const std::vector
<AccountMapping
>& last_account_mappings() {
343 return last_account_mappings_
;
346 const GServicesSettings
& gservices_settings() const {
347 return gcm_client_
->gservices_settings_
;
350 const base::FilePath
& temp_directory_path() const {
351 return temp_directory_
.path();
354 base::FilePath
gcm_store_path() const {
355 // Pass an non-existent directory as store path to match the exact
356 // behavior in the production code. Currently GCMStoreImpl checks if
357 // the directory exist or not to determine the store existence.
358 return temp_directory_
.path().Append(FILE_PATH_LITERAL("GCM Store"));
364 void PumpLoopUntilIdle();
365 bool CreateUniqueTempDir();
366 AutoAdvancingTestClock
* clock() const {
367 return reinterpret_cast<AutoAdvancingTestClock
*>(gcm_client_
->clock_
.get());
369 net::TestURLFetcherFactory
* url_fetcher_factory() {
370 return &url_fetcher_factory_
;
372 base::TestMockTimeTaskRunner
* task_runner() {
373 return task_runner_
.get();
377 // Variables used for verification.
378 LastEvent last_event_
;
379 std::string last_app_id_
;
380 std::string last_registration_id_
;
381 std::string last_message_id_
;
382 GCMClient::Result last_result_
;
383 GCMClient::IncomingMessage last_message_
;
384 GCMClient::SendErrorDetails last_error_details_
;
385 base::Time last_token_fetch_time_
;
386 std::vector
<AccountMapping
> last_account_mappings_
;
388 scoped_ptr
<GCMClientImpl
> gcm_client_
;
390 net::TestURLFetcherFactory url_fetcher_factory_
;
392 scoped_refptr
<base::TestMockTimeTaskRunner
> task_runner_
;
393 base::ThreadTaskRunnerHandle task_runner_handle_
;
395 // Injected to GCM client:
396 base::ScopedTempDir temp_directory_
;
397 scoped_refptr
<net::TestURLRequestContextGetter
> url_request_context_getter_
;
400 GCMClientImplTest::GCMClientImplTest()
402 last_result_(GCMClient::UNKNOWN_ERROR
),
403 task_runner_(new base::TestMockTimeTaskRunner
),
404 task_runner_handle_(task_runner_
),
405 url_request_context_getter_(
406 new net::TestURLRequestContextGetter(task_runner_
)) {
409 GCMClientImplTest::~GCMClientImplTest() {}
411 void GCMClientImplTest::SetUp() {
412 testing::Test::SetUp();
413 ASSERT_TRUE(CreateUniqueTempDir());
414 BuildGCMClient(base::TimeDelta());
415 InitializeGCMClient();
417 SetUpUrlFetcherFactory();
418 CompleteCheckin(kDeviceAndroidId
,
419 kDeviceSecurityToken
,
421 std::map
<std::string
, std::string
>());
424 void GCMClientImplTest::SetUpUrlFetcherFactory() {
425 url_fetcher_factory_
.set_remove_fetcher_on_delete(true);
428 void GCMClientImplTest::PumpLoopUntilIdle() {
429 task_runner_
->RunUntilIdle();
432 bool GCMClientImplTest::CreateUniqueTempDir() {
433 return temp_directory_
.CreateUniqueTempDir();
436 void GCMClientImplTest::BuildGCMClient(base::TimeDelta clock_step
) {
437 gcm_client_
.reset(new GCMClientImpl(make_scoped_ptr
<GCMInternalsBuilder
>(
438 new FakeGCMInternalsBuilder(clock_step
))));
441 void GCMClientImplTest::CompleteCheckin(
443 uint64 security_token
,
444 const std::string
& digest
,
445 const std::map
<std::string
, std::string
>& settings
) {
446 checkin_proto::AndroidCheckinResponse response
;
447 response
.set_stats_ok(true);
448 response
.set_android_id(android_id
);
449 response
.set_security_token(security_token
);
451 // For testing G-services settings.
452 if (!digest
.empty()) {
453 response
.set_digest(digest
);
454 for (std::map
<std::string
, std::string
>::const_iterator it
=
456 it
!= settings
.end();
458 checkin_proto::GservicesSetting
* setting
= response
.add_setting();
459 setting
->set_name(it
->first
);
460 setting
->set_value(it
->second
);
462 response
.set_settings_diff(false);
465 std::string response_string
;
466 response
.SerializeToString(&response_string
);
468 net::TestURLFetcher
* fetcher
= url_fetcher_factory_
.GetFetcherByID(0);
469 ASSERT_TRUE(fetcher
);
470 fetcher
->set_response_code(net::HTTP_OK
);
471 fetcher
->SetResponseString(response_string
);
472 fetcher
->delegate()->OnURLFetchComplete(fetcher
);
473 // Give a chance for GCMStoreImpl::Backend to finish persisting data.
477 void GCMClientImplTest::CompleteRegistration(
478 const std::string
& registration_id
) {
479 std::string
response(kRegistrationResponsePrefix
);
480 response
.append(registration_id
);
481 net::TestURLFetcher
* fetcher
= url_fetcher_factory_
.GetFetcherByID(0);
482 ASSERT_TRUE(fetcher
);
483 fetcher
->set_response_code(net::HTTP_OK
);
484 fetcher
->SetResponseString(response
);
485 fetcher
->delegate()->OnURLFetchComplete(fetcher
);
486 // Give a chance for GCMStoreImpl::Backend to finish persisting data.
490 void GCMClientImplTest::CompleteUnregistration(
491 const std::string
& app_id
) {
492 std::string
response(kUnregistrationResponsePrefix
);
493 response
.append(app_id
);
494 net::TestURLFetcher
* fetcher
= url_fetcher_factory_
.GetFetcherByID(0);
495 ASSERT_TRUE(fetcher
);
496 fetcher
->set_response_code(net::HTTP_OK
);
497 fetcher
->SetResponseString(response
);
498 fetcher
->delegate()->OnURLFetchComplete(fetcher
);
499 // Give a chance for GCMStoreImpl::Backend to finish persisting data.
503 void GCMClientImplTest::VerifyPendingRequestFetcherDeleted() {
504 net::TestURLFetcher
* fetcher
= url_fetcher_factory_
.GetFetcherByID(0);
505 EXPECT_FALSE(fetcher
);
508 bool GCMClientImplTest::ExistsRegistration(const std::string
& app_id
) const {
509 return ExistsGCMRegistrationInMap(gcm_client_
->registrations_
, app_id
);
512 void GCMClientImplTest::AddRegistration(
513 const std::string
& app_id
,
514 const std::vector
<std::string
>& sender_ids
,
515 const std::string
& registration_id
) {
516 linked_ptr
<GCMRegistrationInfo
> registration(new GCMRegistrationInfo
);
517 registration
->app_id
= app_id
;
518 registration
->sender_ids
= sender_ids
;
519 gcm_client_
->registrations_
[registration
] = registration_id
;
522 void GCMClientImplTest::InitializeGCMClient() {
523 clock()->Advance(base::TimeDelta::FromMilliseconds(1));
525 // Actual initialization.
526 GCMClient::ChromeBuildInfo chrome_build_info
;
527 chrome_build_info
.version
= kChromeVersion
;
528 gcm_client_
->Initialize(
532 url_request_context_getter_
,
533 make_scoped_ptr
<Encryptor
>(new FakeEncryptor
),
537 void GCMClientImplTest::StartGCMClient() {
538 // Start loading and check-in.
539 gcm_client_
->Start(GCMClient::IMMEDIATE_START
);
544 void GCMClientImplTest::Register(const std::string
& app_id
,
545 const std::vector
<std::string
>& senders
) {
546 scoped_ptr
<GCMRegistrationInfo
> gcm_info(new GCMRegistrationInfo
);
547 gcm_info
->app_id
= app_id
;
548 gcm_info
->sender_ids
= senders
;
549 gcm_client()->Register(make_linked_ptr
<RegistrationInfo
>(gcm_info
.release()));
552 void GCMClientImplTest::Unregister(const std::string
& app_id
) {
553 scoped_ptr
<GCMRegistrationInfo
> gcm_info(new GCMRegistrationInfo
);
554 gcm_info
->app_id
= app_id
;
555 gcm_client()->Unregister(
556 make_linked_ptr
<RegistrationInfo
>(gcm_info
.release()));
559 void GCMClientImplTest::ReceiveMessageFromMCS(const MCSMessage
& message
) {
560 gcm_client_
->recorder_
.RecordConnectionInitiated(std::string());
561 gcm_client_
->recorder_
.RecordConnectionSuccess();
562 gcm_client_
->OnMessageReceivedFromMCS(message
);
565 void GCMClientImplTest::ReceiveOnMessageSentToMCS(
566 const std::string
& app_id
,
567 const std::string
& message_id
,
568 const MCSClient::MessageSendStatus status
) {
569 gcm_client_
->OnMessageSentToMCS(0LL, app_id
, message_id
, status
);
572 void GCMClientImplTest::OnGCMReady(
573 const std::vector
<AccountMapping
>& account_mappings
,
574 const base::Time
& last_token_fetch_time
) {
575 last_event_
= LOADING_COMPLETED
;
576 last_account_mappings_
= account_mappings
;
577 last_token_fetch_time_
= last_token_fetch_time
;
580 void GCMClientImplTest::OnMessageReceived(
581 const std::string
& registration_id
,
582 const GCMClient::IncomingMessage
& message
) {
583 last_event_
= MESSAGE_RECEIVED
;
584 last_app_id_
= registration_id
;
585 last_message_
= message
;
588 void GCMClientImplTest::OnRegisterFinished(
589 const linked_ptr
<RegistrationInfo
>& registration_info
,
590 const std::string
& registration_id
,
591 GCMClient::Result result
) {
592 last_event_
= REGISTRATION_COMPLETED
;
593 last_app_id_
= registration_info
->app_id
;
594 last_registration_id_
= registration_id
;
595 last_result_
= result
;
598 void GCMClientImplTest::OnUnregisterFinished(
599 const linked_ptr
<RegistrationInfo
>& registration_info
,
600 GCMClient::Result result
) {
601 last_event_
= UNREGISTRATION_COMPLETED
;
602 last_app_id_
= registration_info
->app_id
;
603 last_result_
= result
;
606 void GCMClientImplTest::OnMessagesDeleted(const std::string
& app_id
) {
607 last_event_
= MESSAGES_DELETED
;
608 last_app_id_
= app_id
;
611 void GCMClientImplTest::OnMessageSendError(
612 const std::string
& app_id
,
613 const gcm::GCMClient::SendErrorDetails
& send_error_details
) {
614 last_event_
= MESSAGE_SEND_ERROR
;
615 last_app_id_
= app_id
;
616 last_error_details_
= send_error_details
;
619 void GCMClientImplTest::OnSendAcknowledged(const std::string
& app_id
,
620 const std::string
& message_id
) {
621 last_event_
= MESSAGE_SEND_ACK
;
622 last_app_id_
= app_id
;
623 last_message_id_
= message_id
;
626 int64
GCMClientImplTest::CurrentTime() {
627 return clock()->Now().ToInternalValue() / base::Time::kMicrosecondsPerSecond
;
630 TEST_F(GCMClientImplTest
, LoadingCompleted
) {
631 EXPECT_EQ(LOADING_COMPLETED
, last_event());
632 EXPECT_EQ(kDeviceAndroidId
, mcs_client()->last_android_id());
633 EXPECT_EQ(kDeviceSecurityToken
, mcs_client()->last_security_token());
635 // Checking freshly loaded CheckinInfo.
636 EXPECT_EQ(kDeviceAndroidId
, device_checkin_info().android_id
);
637 EXPECT_EQ(kDeviceSecurityToken
, device_checkin_info().secret
);
638 EXPECT_TRUE(device_checkin_info().last_checkin_accounts
.empty());
639 EXPECT_TRUE(device_checkin_info().accounts_set
);
640 EXPECT_TRUE(device_checkin_info().account_tokens
.empty());
643 TEST_F(GCMClientImplTest
, LoadingBusted
) {
644 // Close the GCM store.
645 gcm_client()->Stop();
648 // Mess up the store.
649 base::FilePath store_file_path
=
650 gcm_store_path().Append(FILE_PATH_LITERAL("CURRENT"));
651 ASSERT_TRUE(base::AppendToFile(store_file_path
, "A", 1));
653 // Restart GCM client. The store should be reset and the loading should
654 // complete successfully.
656 BuildGCMClient(base::TimeDelta());
657 InitializeGCMClient();
659 CompleteCheckin(kDeviceAndroidId2
,
660 kDeviceSecurityToken2
,
662 std::map
<std::string
, std::string
>());
664 EXPECT_EQ(LOADING_COMPLETED
, last_event());
665 EXPECT_EQ(kDeviceAndroidId2
, mcs_client()->last_android_id());
666 EXPECT_EQ(kDeviceSecurityToken2
, mcs_client()->last_security_token());
669 TEST_F(GCMClientImplTest
, DestroyStoreWhenNotNeeded
) {
670 // Close the GCM store.
671 gcm_client()->Stop();
674 // Restart GCM client. The store is loaded successfully.
676 BuildGCMClient(base::TimeDelta());
677 InitializeGCMClient();
678 gcm_client()->Start(GCMClient::DELAYED_START
);
681 EXPECT_EQ(GCMClientImpl::LOADED
, gcm_client_state());
682 EXPECT_TRUE(device_checkin_info().android_id
);
683 EXPECT_TRUE(device_checkin_info().secret
);
685 // Fast forward the clock to trigger the store destroying logic.
686 task_runner()->FastForwardBy(base::TimeDelta::FromMilliseconds(300000));
689 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
690 EXPECT_FALSE(device_checkin_info().android_id
);
691 EXPECT_FALSE(device_checkin_info().secret
);
694 TEST_F(GCMClientImplTest
, RegisterApp
) {
695 EXPECT_FALSE(ExistsRegistration(kAppId
));
697 std::vector
<std::string
> senders
;
698 senders
.push_back("sender");
699 Register(kAppId
, senders
);
700 CompleteRegistration("reg_id");
702 EXPECT_EQ(REGISTRATION_COMPLETED
, last_event());
703 EXPECT_EQ(kAppId
, last_app_id());
704 EXPECT_EQ("reg_id", last_registration_id());
705 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
706 EXPECT_TRUE(ExistsRegistration(kAppId
));
709 TEST_F(GCMClientImplTest
, DISABLED_RegisterAppFromCache
) {
710 EXPECT_FALSE(ExistsRegistration(kAppId
));
712 std::vector
<std::string
> senders
;
713 senders
.push_back("sender");
714 Register(kAppId
, senders
);
715 CompleteRegistration("reg_id");
716 EXPECT_TRUE(ExistsRegistration(kAppId
));
718 EXPECT_EQ(kAppId
, last_app_id());
719 EXPECT_EQ("reg_id", last_registration_id());
720 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
721 EXPECT_EQ(REGISTRATION_COMPLETED
, last_event());
723 // Recreate GCMClient in order to load from the persistent store.
724 BuildGCMClient(base::TimeDelta());
725 InitializeGCMClient();
728 EXPECT_TRUE(ExistsRegistration(kAppId
));
731 TEST_F(GCMClientImplTest
, RegisterPreviousSenderAgain
) {
732 EXPECT_FALSE(ExistsRegistration(kAppId
));
734 // Register a sender.
735 std::vector
<std::string
> senders
;
736 senders
.push_back("sender");
737 Register(kAppId
, senders
);
738 CompleteRegistration("reg_id");
740 EXPECT_EQ(REGISTRATION_COMPLETED
, last_event());
741 EXPECT_EQ(kAppId
, last_app_id());
742 EXPECT_EQ("reg_id", last_registration_id());
743 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
744 EXPECT_TRUE(ExistsRegistration(kAppId
));
748 // Register a different sender. Different registration ID from previous one
749 // should be returned.
750 std::vector
<std::string
> senders2
;
751 senders2
.push_back("sender2");
752 Register(kAppId
, senders2
);
753 CompleteRegistration("reg_id2");
755 EXPECT_EQ(REGISTRATION_COMPLETED
, last_event());
756 EXPECT_EQ(kAppId
, last_app_id());
757 EXPECT_EQ("reg_id2", last_registration_id());
758 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
759 EXPECT_TRUE(ExistsRegistration(kAppId
));
763 // Register the 1st sender again. Different registration ID from previous one
764 // should be returned.
765 std::vector
<std::string
> senders3
;
766 senders3
.push_back("sender");
767 Register(kAppId
, senders3
);
768 CompleteRegistration("reg_id");
770 EXPECT_EQ(REGISTRATION_COMPLETED
, last_event());
771 EXPECT_EQ(kAppId
, last_app_id());
772 EXPECT_EQ("reg_id", last_registration_id());
773 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
774 EXPECT_TRUE(ExistsRegistration(kAppId
));
777 TEST_F(GCMClientImplTest
, UnregisterApp
) {
778 EXPECT_FALSE(ExistsRegistration(kAppId
));
780 std::vector
<std::string
> senders
;
781 senders
.push_back("sender");
782 Register(kAppId
, senders
);
783 CompleteRegistration("reg_id");
784 EXPECT_TRUE(ExistsRegistration(kAppId
));
787 CompleteUnregistration(kAppId
);
789 EXPECT_EQ(UNREGISTRATION_COMPLETED
, last_event());
790 EXPECT_EQ(kAppId
, last_app_id());
791 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
792 EXPECT_FALSE(ExistsRegistration(kAppId
));
795 // Tests that stopping the GCMClient also deletes pending registration requests.
796 // This is tested by checking that url fetcher contained in the request was
798 TEST_F(GCMClientImplTest
, DeletePendingRequestsWhenStopping
) {
799 std::vector
<std::string
> senders
;
800 senders
.push_back("sender");
801 Register(kAppId
, senders
);
803 gcm_client()->Stop();
805 VerifyPendingRequestFetcherDeleted();
808 TEST_F(GCMClientImplTest
, DispatchDownstreamMessage
) {
809 // Register to receive messages from kSender and kSender2 only.
810 std::vector
<std::string
> senders
;
811 senders
.push_back(kSender
);
812 senders
.push_back(kSender2
);
813 AddRegistration(kAppId
, senders
, "reg_id");
815 std::map
<std::string
, std::string
> expected_data
;
816 expected_data
["message_type"] = "gcm";
817 expected_data
["key"] = "value";
818 expected_data
["key2"] = "value2";
820 // Message for kSender will be received.
821 MCSMessage
message(BuildDownstreamMessage(kSender
, kAppId
, expected_data
));
822 EXPECT_TRUE(message
.IsValid());
823 ReceiveMessageFromMCS(message
);
825 expected_data
.erase(expected_data
.find("message_type"));
826 EXPECT_EQ(MESSAGE_RECEIVED
, last_event());
827 EXPECT_EQ(kAppId
, last_app_id());
828 EXPECT_EQ(expected_data
.size(), last_message().data
.size());
829 EXPECT_EQ(expected_data
, last_message().data
);
830 EXPECT_EQ(kSender
, last_message().sender_id
);
834 // Message for kSender2 will be received.
835 MCSMessage
message2(BuildDownstreamMessage(kSender2
, kAppId
, expected_data
));
836 EXPECT_TRUE(message2
.IsValid());
837 ReceiveMessageFromMCS(message2
);
839 EXPECT_EQ(MESSAGE_RECEIVED
, last_event());
840 EXPECT_EQ(kAppId
, last_app_id());
841 EXPECT_EQ(expected_data
.size(), last_message().data
.size());
842 EXPECT_EQ(expected_data
, last_message().data
);
843 EXPECT_EQ(kSender2
, last_message().sender_id
);
847 // Message from kSender3 will be dropped.
848 MCSMessage
message3(BuildDownstreamMessage(kSender3
, kAppId
, expected_data
));
849 EXPECT_TRUE(message3
.IsValid());
850 ReceiveMessageFromMCS(message3
);
852 EXPECT_NE(MESSAGE_RECEIVED
, last_event());
853 EXPECT_NE(kAppId
, last_app_id());
856 TEST_F(GCMClientImplTest
, DispatchDownstreamMessageSendError
) {
857 std::map
<std::string
, std::string
> expected_data
;
858 expected_data
["message_type"] = "send_error";
859 expected_data
["google.message_id"] = "007";
860 expected_data
["error_details"] = "some details";
861 MCSMessage
message(BuildDownstreamMessage(
862 kSender
, kAppId
, expected_data
));
863 EXPECT_TRUE(message
.IsValid());
864 ReceiveMessageFromMCS(message
);
866 EXPECT_EQ(MESSAGE_SEND_ERROR
, last_event());
867 EXPECT_EQ(kAppId
, last_app_id());
868 EXPECT_EQ("007", last_error_details().message_id
);
869 EXPECT_EQ(1UL, last_error_details().additional_data
.size());
870 GCMClient::MessageData::const_iterator iter
=
871 last_error_details().additional_data
.find("error_details");
872 EXPECT_TRUE(iter
!= last_error_details().additional_data
.end());
873 EXPECT_EQ("some details", iter
->second
);
876 TEST_F(GCMClientImplTest
, DispatchDownstreamMessgaesDeleted
) {
877 std::map
<std::string
, std::string
> expected_data
;
878 expected_data
["message_type"] = "deleted_messages";
879 MCSMessage
message(BuildDownstreamMessage(
880 kSender
, kAppId
, expected_data
));
881 EXPECT_TRUE(message
.IsValid());
882 ReceiveMessageFromMCS(message
);
884 EXPECT_EQ(MESSAGES_DELETED
, last_event());
885 EXPECT_EQ(kAppId
, last_app_id());
888 TEST_F(GCMClientImplTest
, SendMessage
) {
889 GCMClient::OutgoingMessage message
;
891 message
.time_to_live
= 500;
892 message
.data
["key"] = "value";
893 gcm_client()->Send(kAppId
, kSender
, message
);
895 EXPECT_EQ(kDataMessageStanzaTag
, mcs_client()->last_message_tag());
896 EXPECT_EQ(kAppId
, mcs_client()->last_data_message_stanza().category());
897 EXPECT_EQ(kSender
, mcs_client()->last_data_message_stanza().to());
898 EXPECT_EQ(500, mcs_client()->last_data_message_stanza().ttl());
899 EXPECT_EQ(CurrentTime(), mcs_client()->last_data_message_stanza().sent());
900 EXPECT_EQ("007", mcs_client()->last_data_message_stanza().id());
901 EXPECT_EQ("gcm@chrome.com", mcs_client()->last_data_message_stanza().from());
902 EXPECT_EQ(kSender
, mcs_client()->last_data_message_stanza().to());
903 EXPECT_EQ("key", mcs_client()->last_data_message_stanza().app_data(0).key());
905 mcs_client()->last_data_message_stanza().app_data(0).value());
908 TEST_F(GCMClientImplTest
, SendMessageAcknowledged
) {
909 ReceiveOnMessageSentToMCS(kAppId
, "007", MCSClient::SENT
);
910 EXPECT_EQ(MESSAGE_SEND_ACK
, last_event());
911 EXPECT_EQ(kAppId
, last_app_id());
912 EXPECT_EQ("007", last_message_id());
915 class GCMClientImplCheckinTest
: public GCMClientImplTest
{
917 GCMClientImplCheckinTest();
918 ~GCMClientImplCheckinTest() override
;
920 void SetUp() override
;
923 GCMClientImplCheckinTest::GCMClientImplCheckinTest() {
926 GCMClientImplCheckinTest::~GCMClientImplCheckinTest() {
929 void GCMClientImplCheckinTest::SetUp() {
930 testing::Test::SetUp();
931 // Creating unique temp directory that will be used by GCMStore shared between
932 // GCM Client and G-services settings.
933 ASSERT_TRUE(CreateUniqueTempDir());
934 // Time will be advancing one hour every time it is checked.
935 BuildGCMClient(base::TimeDelta::FromSeconds(kSettingsCheckinInterval
));
936 InitializeGCMClient();
940 TEST_F(GCMClientImplCheckinTest
, GServicesSettingsAfterInitialCheckin
) {
941 std::map
<std::string
, std::string
> settings
;
942 settings
["checkin_interval"] = base::Int64ToString(kSettingsCheckinInterval
);
943 settings
["checkin_url"] = "http://alternative.url/checkin";
944 settings
["gcm_hostname"] = "alternative.gcm.host";
945 settings
["gcm_secure_port"] = "7777";
946 settings
["gcm_registration_url"] = "http://alternative.url/registration";
947 CompleteCheckin(kDeviceAndroidId
,
948 kDeviceSecurityToken
,
949 GServicesSettings::CalculateDigest(settings
),
951 EXPECT_EQ(base::TimeDelta::FromSeconds(kSettingsCheckinInterval
),
952 gservices_settings().GetCheckinInterval());
953 EXPECT_EQ(GURL("http://alternative.url/checkin"),
954 gservices_settings().GetCheckinURL());
955 EXPECT_EQ(GURL("http://alternative.url/registration"),
956 gservices_settings().GetRegistrationURL());
957 EXPECT_EQ(GURL("https://alternative.gcm.host:7777"),
958 gservices_settings().GetMCSMainEndpoint());
959 EXPECT_EQ(GURL("https://alternative.gcm.host:443"),
960 gservices_settings().GetMCSFallbackEndpoint());
963 // This test only checks that periodic checkin happens.
964 TEST_F(GCMClientImplCheckinTest
, PeriodicCheckin
) {
965 std::map
<std::string
, std::string
> settings
;
966 settings
["checkin_interval"] = base::IntToString(kSettingsCheckinInterval
);
967 settings
["checkin_url"] = "http://alternative.url/checkin";
968 settings
["gcm_hostname"] = "alternative.gcm.host";
969 settings
["gcm_secure_port"] = "7777";
970 settings
["gcm_registration_url"] = "http://alternative.url/registration";
971 CompleteCheckin(kDeviceAndroidId
,
972 kDeviceSecurityToken
,
973 GServicesSettings::CalculateDigest(settings
),
976 EXPECT_EQ(2, clock()->call_count());
979 CompleteCheckin(kDeviceAndroidId
,
980 kDeviceSecurityToken
,
981 GServicesSettings::CalculateDigest(settings
),
985 TEST_F(GCMClientImplCheckinTest
, LoadGSettingsFromStore
) {
986 std::map
<std::string
, std::string
> settings
;
987 settings
["checkin_interval"] = base::IntToString(kSettingsCheckinInterval
);
988 settings
["checkin_url"] = "http://alternative.url/checkin";
989 settings
["gcm_hostname"] = "alternative.gcm.host";
990 settings
["gcm_secure_port"] = "7777";
991 settings
["gcm_registration_url"] = "http://alternative.url/registration";
992 CompleteCheckin(kDeviceAndroidId
,
993 kDeviceSecurityToken
,
994 GServicesSettings::CalculateDigest(settings
),
997 BuildGCMClient(base::TimeDelta());
998 InitializeGCMClient();
1001 EXPECT_EQ(base::TimeDelta::FromSeconds(kSettingsCheckinInterval
),
1002 gservices_settings().GetCheckinInterval());
1003 EXPECT_EQ(GURL("http://alternative.url/checkin"),
1004 gservices_settings().GetCheckinURL());
1005 EXPECT_EQ(GURL("http://alternative.url/registration"),
1006 gservices_settings().GetRegistrationURL());
1007 EXPECT_EQ(GURL("https://alternative.gcm.host:7777"),
1008 gservices_settings().GetMCSMainEndpoint());
1009 EXPECT_EQ(GURL("https://alternative.gcm.host:443"),
1010 gservices_settings().GetMCSFallbackEndpoint());
1013 // This test only checks that periodic checkin happens.
1014 TEST_F(GCMClientImplCheckinTest
, CheckinWithAccounts
) {
1015 std::map
<std::string
, std::string
> settings
;
1016 settings
["checkin_interval"] = base::IntToString(kSettingsCheckinInterval
);
1017 settings
["checkin_url"] = "http://alternative.url/checkin";
1018 settings
["gcm_hostname"] = "alternative.gcm.host";
1019 settings
["gcm_secure_port"] = "7777";
1020 settings
["gcm_registration_url"] = "http://alternative.url/registration";
1021 CompleteCheckin(kDeviceAndroidId
,
1022 kDeviceSecurityToken
,
1023 GServicesSettings::CalculateDigest(settings
),
1026 std::vector
<GCMClient::AccountTokenInfo
> account_tokens
;
1027 account_tokens
.push_back(MakeAccountToken("test_user1@gmail.com", "token1"));
1028 account_tokens
.push_back(MakeAccountToken("test_user2@gmail.com", "token2"));
1029 gcm_client()->SetAccountTokens(account_tokens
);
1031 EXPECT_TRUE(device_checkin_info().last_checkin_accounts
.empty());
1032 EXPECT_TRUE(device_checkin_info().accounts_set
);
1033 EXPECT_EQ(MakeEmailToTokenMap(account_tokens
),
1034 device_checkin_info().account_tokens
);
1036 PumpLoopUntilIdle();
1037 CompleteCheckin(kDeviceAndroidId
,
1038 kDeviceSecurityToken
,
1039 GServicesSettings::CalculateDigest(settings
),
1042 std::set
<std::string
> accounts
;
1043 accounts
.insert("test_user1@gmail.com");
1044 accounts
.insert("test_user2@gmail.com");
1045 EXPECT_EQ(accounts
, device_checkin_info().last_checkin_accounts
);
1046 EXPECT_TRUE(device_checkin_info().accounts_set
);
1047 EXPECT_EQ(MakeEmailToTokenMap(account_tokens
),
1048 device_checkin_info().account_tokens
);
1051 // This test only checks that periodic checkin happens.
1052 TEST_F(GCMClientImplCheckinTest
, CheckinWhenAccountRemoved
) {
1053 std::map
<std::string
, std::string
> settings
;
1054 settings
["checkin_interval"] = base::IntToString(kSettingsCheckinInterval
);
1055 settings
["checkin_url"] = "http://alternative.url/checkin";
1056 settings
["gcm_hostname"] = "alternative.gcm.host";
1057 settings
["gcm_secure_port"] = "7777";
1058 settings
["gcm_registration_url"] = "http://alternative.url/registration";
1059 CompleteCheckin(kDeviceAndroidId
,
1060 kDeviceSecurityToken
,
1061 GServicesSettings::CalculateDigest(settings
),
1064 std::vector
<GCMClient::AccountTokenInfo
> account_tokens
;
1065 account_tokens
.push_back(MakeAccountToken("test_user1@gmail.com", "token1"));
1066 account_tokens
.push_back(MakeAccountToken("test_user2@gmail.com", "token2"));
1067 gcm_client()->SetAccountTokens(account_tokens
);
1068 PumpLoopUntilIdle();
1069 CompleteCheckin(kDeviceAndroidId
,
1070 kDeviceSecurityToken
,
1071 GServicesSettings::CalculateDigest(settings
),
1074 EXPECT_EQ(2UL, device_checkin_info().last_checkin_accounts
.size());
1075 EXPECT_TRUE(device_checkin_info().accounts_set
);
1076 EXPECT_EQ(MakeEmailToTokenMap(account_tokens
),
1077 device_checkin_info().account_tokens
);
1079 account_tokens
.erase(account_tokens
.begin() + 1);
1080 gcm_client()->SetAccountTokens(account_tokens
);
1082 PumpLoopUntilIdle();
1083 CompleteCheckin(kDeviceAndroidId
,
1084 kDeviceSecurityToken
,
1085 GServicesSettings::CalculateDigest(settings
),
1088 std::set
<std::string
> accounts
;
1089 accounts
.insert("test_user1@gmail.com");
1090 EXPECT_EQ(accounts
, device_checkin_info().last_checkin_accounts
);
1091 EXPECT_TRUE(device_checkin_info().accounts_set
);
1092 EXPECT_EQ(MakeEmailToTokenMap(account_tokens
),
1093 device_checkin_info().account_tokens
);
1096 // This test only checks that periodic checkin happens.
1097 TEST_F(GCMClientImplCheckinTest
, CheckinWhenAccountReplaced
) {
1098 std::map
<std::string
, std::string
> settings
;
1099 settings
["checkin_interval"] = base::IntToString(kSettingsCheckinInterval
);
1100 settings
["checkin_url"] = "http://alternative.url/checkin";
1101 settings
["gcm_hostname"] = "alternative.gcm.host";
1102 settings
["gcm_secure_port"] = "7777";
1103 settings
["gcm_registration_url"] = "http://alternative.url/registration";
1104 CompleteCheckin(kDeviceAndroidId
,
1105 kDeviceSecurityToken
,
1106 GServicesSettings::CalculateDigest(settings
),
1109 std::vector
<GCMClient::AccountTokenInfo
> account_tokens
;
1110 account_tokens
.push_back(MakeAccountToken("test_user1@gmail.com", "token1"));
1111 gcm_client()->SetAccountTokens(account_tokens
);
1113 PumpLoopUntilIdle();
1114 CompleteCheckin(kDeviceAndroidId
,
1115 kDeviceSecurityToken
,
1116 GServicesSettings::CalculateDigest(settings
),
1119 std::set
<std::string
> accounts
;
1120 accounts
.insert("test_user1@gmail.com");
1121 EXPECT_EQ(accounts
, device_checkin_info().last_checkin_accounts
);
1123 // This should trigger another checkin, because the list of accounts is
1125 account_tokens
.clear();
1126 account_tokens
.push_back(MakeAccountToken("test_user2@gmail.com", "token2"));
1127 gcm_client()->SetAccountTokens(account_tokens
);
1129 PumpLoopUntilIdle();
1130 CompleteCheckin(kDeviceAndroidId
,
1131 kDeviceSecurityToken
,
1132 GServicesSettings::CalculateDigest(settings
),
1136 accounts
.insert("test_user2@gmail.com");
1137 EXPECT_EQ(accounts
, device_checkin_info().last_checkin_accounts
);
1138 EXPECT_TRUE(device_checkin_info().accounts_set
);
1139 EXPECT_EQ(MakeEmailToTokenMap(account_tokens
),
1140 device_checkin_info().account_tokens
);
1143 class GCMClientImplStartAndStopTest
: public GCMClientImplTest
{
1145 GCMClientImplStartAndStopTest();
1146 ~GCMClientImplStartAndStopTest() override
;
1148 void SetUp() override
;
1150 void DefaultCompleteCheckin();
1153 GCMClientImplStartAndStopTest::GCMClientImplStartAndStopTest() {
1156 GCMClientImplStartAndStopTest::~GCMClientImplStartAndStopTest() {
1159 void GCMClientImplStartAndStopTest::SetUp() {
1160 testing::Test::SetUp();
1161 ASSERT_TRUE(CreateUniqueTempDir());
1162 BuildGCMClient(base::TimeDelta());
1163 InitializeGCMClient();
1166 void GCMClientImplStartAndStopTest::DefaultCompleteCheckin() {
1167 SetUpUrlFetcherFactory();
1168 CompleteCheckin(kDeviceAndroidId
,
1169 kDeviceSecurityToken
,
1171 std::map
<std::string
, std::string
>());
1172 PumpLoopUntilIdle();
1175 TEST_F(GCMClientImplStartAndStopTest
, StartStopAndRestart
) {
1176 // GCMClientImpl should be in INITIALIZED state at first.
1177 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1179 // Delay start the GCM.
1180 gcm_client()->Start(GCMClient::DELAYED_START
);
1181 PumpLoopUntilIdle();
1182 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1185 gcm_client()->Stop();
1186 PumpLoopUntilIdle();
1187 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1189 // Restart the GCM without delay.
1190 gcm_client()->Start(GCMClient::IMMEDIATE_START
);
1191 PumpLoopUntilIdle();
1192 EXPECT_EQ(GCMClientImpl::INITIAL_DEVICE_CHECKIN
, gcm_client_state());
1195 TEST_F(GCMClientImplStartAndStopTest
, DelayedStartAndStopImmediately
) {
1196 // GCMClientImpl should be in INITIALIZED state at first.
1197 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1199 // Delay start the GCM and then stop it immediately.
1200 gcm_client()->Start(GCMClient::DELAYED_START
);
1201 gcm_client()->Stop();
1202 PumpLoopUntilIdle();
1203 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1206 TEST_F(GCMClientImplStartAndStopTest
, ImmediateStartAndStopImmediately
) {
1207 // GCMClientImpl should be in INITIALIZED state at first.
1208 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1210 // Start the GCM and then stop it immediately.
1211 gcm_client()->Start(GCMClient::IMMEDIATE_START
);
1212 gcm_client()->Stop();
1213 PumpLoopUntilIdle();
1214 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1217 TEST_F(GCMClientImplStartAndStopTest
, DelayedStartStopAndRestart
) {
1218 // GCMClientImpl should be in INITIALIZED state at first.
1219 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1221 // Delay start the GCM and then stop and restart it immediately.
1222 gcm_client()->Start(GCMClient::DELAYED_START
);
1223 gcm_client()->Stop();
1224 gcm_client()->Start(GCMClient::DELAYED_START
);
1225 PumpLoopUntilIdle();
1226 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1229 TEST_F(GCMClientImplStartAndStopTest
, ImmediateStartStopAndRestart
) {
1230 // GCMClientImpl should be in INITIALIZED state at first.
1231 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1233 // Start the GCM and then stop and restart it immediately.
1234 gcm_client()->Start(GCMClient::IMMEDIATE_START
);
1235 gcm_client()->Stop();
1236 gcm_client()->Start(GCMClient::IMMEDIATE_START
);
1237 PumpLoopUntilIdle();
1238 EXPECT_EQ(GCMClientImpl::INITIAL_DEVICE_CHECKIN
, gcm_client_state());
1241 TEST_F(GCMClientImplStartAndStopTest
, ImmediateStartAndThenImmediateStart
) {
1242 // GCMClientImpl should be in INITIALIZED state at first.
1243 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1245 // Start the GCM immediately and complete the checkin.
1246 gcm_client()->Start(GCMClient::IMMEDIATE_START
);
1247 PumpLoopUntilIdle();
1248 EXPECT_EQ(GCMClientImpl::INITIAL_DEVICE_CHECKIN
, gcm_client_state());
1249 DefaultCompleteCheckin();
1250 EXPECT_EQ(GCMClientImpl::READY
, gcm_client_state());
1253 gcm_client()->Stop();
1254 PumpLoopUntilIdle();
1255 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1257 // Start the GCM immediately. GCMClientImpl should be in READY state.
1258 BuildGCMClient(base::TimeDelta());
1259 InitializeGCMClient();
1260 gcm_client()->Start(GCMClient::IMMEDIATE_START
);
1261 PumpLoopUntilIdle();
1262 EXPECT_EQ(GCMClientImpl::READY
, gcm_client_state());
1265 TEST_F(GCMClientImplStartAndStopTest
, ImmediateStartAndThenDelayStart
) {
1266 // GCMClientImpl should be in INITIALIZED state at first.
1267 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1269 // Start the GCM immediately and complete the checkin.
1270 gcm_client()->Start(GCMClient::IMMEDIATE_START
);
1271 PumpLoopUntilIdle();
1272 EXPECT_EQ(GCMClientImpl::INITIAL_DEVICE_CHECKIN
, gcm_client_state());
1273 DefaultCompleteCheckin();
1274 EXPECT_EQ(GCMClientImpl::READY
, gcm_client_state());
1277 gcm_client()->Stop();
1278 PumpLoopUntilIdle();
1279 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1281 // Delay start the GCM. GCMClientImpl should be in LOADED state.
1282 BuildGCMClient(base::TimeDelta());
1283 InitializeGCMClient();
1284 gcm_client()->Start(GCMClient::DELAYED_START
);
1285 PumpLoopUntilIdle();
1286 EXPECT_EQ(GCMClientImpl::LOADED
, gcm_client_state());
1289 TEST_F(GCMClientImplStartAndStopTest
, DelayedStart
) {
1290 // GCMClientImpl should be in INITIALIZED state at first.
1291 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1293 // Delay start the GCM. The store will not be loaded and GCMClientImpl should
1294 // still be in INITIALIZED state.
1295 gcm_client()->Start(GCMClient::DELAYED_START
);
1296 PumpLoopUntilIdle();
1297 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1299 // Start the GCM immediately and complete the checkin.
1300 gcm_client()->Start(GCMClient::IMMEDIATE_START
);
1301 PumpLoopUntilIdle();
1302 EXPECT_EQ(GCMClientImpl::INITIAL_DEVICE_CHECKIN
, gcm_client_state());
1303 DefaultCompleteCheckin();
1304 EXPECT_EQ(GCMClientImpl::READY
, gcm_client_state());
1307 std::vector
<std::string
> senders
;
1308 senders
.push_back("sender");
1309 Register(kAppId
, senders
);
1310 CompleteRegistration("reg_id");
1311 EXPECT_EQ(GCMClientImpl::READY
, gcm_client_state());
1314 gcm_client()->Stop();
1315 PumpLoopUntilIdle();
1316 EXPECT_EQ(GCMClientImpl::INITIALIZED
, gcm_client_state());
1318 // Delay start the GCM. GCM is indeed started without delay because the
1319 // registration record has been found.
1320 BuildGCMClient(base::TimeDelta());
1321 InitializeGCMClient();
1322 gcm_client()->Start(GCMClient::DELAYED_START
);
1323 PumpLoopUntilIdle();
1324 EXPECT_EQ(GCMClientImpl::READY
, gcm_client_state());
1327 // Test for known account mappings and last token fetching time being passed
1329 TEST_F(GCMClientImplStartAndStopTest
, OnGCMReadyAccountsAndTokenFetchingTime
) {
1330 // Start the GCM and wait until it is ready.
1331 gcm_client()->Start(GCMClient::IMMEDIATE_START
);
1332 PumpLoopUntilIdle();
1333 DefaultCompleteCheckin();
1335 base::Time expected_time
= base::Time::Now();
1336 gcm_client()->SetLastTokenFetchTime(expected_time
);
1337 AccountMapping expected_mapping
;
1338 expected_mapping
.account_id
= "accId";
1339 expected_mapping
.email
= "email@gmail.com";
1340 expected_mapping
.status
= AccountMapping::MAPPED
;
1341 expected_mapping
.status_change_timestamp
= expected_time
;
1342 gcm_client()->UpdateAccountMapping(expected_mapping
);
1343 PumpLoopUntilIdle();
1346 gcm_client()->Stop();
1347 PumpLoopUntilIdle();
1350 gcm_client()->Start(GCMClient::IMMEDIATE_START
);
1351 PumpLoopUntilIdle();
1353 EXPECT_EQ(LOADING_COMPLETED
, last_event());
1354 EXPECT_EQ(expected_time
, last_token_fetch_time());
1355 ASSERT_EQ(1UL, last_account_mappings().size());
1356 const AccountMapping
& actual_mapping
= last_account_mappings()[0];
1357 EXPECT_EQ(expected_mapping
.account_id
, actual_mapping
.account_id
);
1358 EXPECT_EQ(expected_mapping
.email
, actual_mapping
.email
);
1359 EXPECT_EQ(expected_mapping
.status
, actual_mapping
.status
);
1360 EXPECT_EQ(expected_mapping
.status_change_timestamp
,
1361 actual_mapping
.status_change_timestamp
);
1365 class GCMClientInstanceIDTest
: public GCMClientImplTest
{
1367 GCMClientInstanceIDTest();
1368 ~GCMClientInstanceIDTest() override
;
1370 void AddInstanceID(const std::string
& app_id
,
1371 const std::string
& instance_id
);
1372 void RemoveInstanceID(const std::string
& app_id
);
1373 void GetToken(const std::string
& app_id
,
1374 const std::string
& authorized_entity
,
1375 const std::string
& scope
);
1376 void DeleteToken(const std::string
& app_id
,
1377 const std::string
& authorized_entity
,
1378 const std::string
& scope
);
1379 void CompleteDeleteToken();
1380 bool ExistsToken(const std::string
& app_id
,
1381 const std::string
& authorized_entity
,
1382 const std::string
& scope
) const;
1385 GCMClientInstanceIDTest::GCMClientInstanceIDTest() {
1388 GCMClientInstanceIDTest::~GCMClientInstanceIDTest() {
1391 void GCMClientInstanceIDTest::AddInstanceID(const std::string
& app_id
,
1392 const std::string
& instance_id
) {
1393 gcm_client()->AddInstanceIDData(app_id
, instance_id
, "123");
1396 void GCMClientInstanceIDTest::RemoveInstanceID(const std::string
& app_id
) {
1397 gcm_client()->RemoveInstanceIDData(app_id
);
1400 void GCMClientInstanceIDTest::GetToken(const std::string
& app_id
,
1401 const std::string
& authorized_entity
,
1402 const std::string
& scope
) {
1403 scoped_ptr
<InstanceIDTokenInfo
> instance_id_info(new InstanceIDTokenInfo
);
1404 instance_id_info
->app_id
= app_id
;
1405 instance_id_info
->authorized_entity
= authorized_entity
;
1406 instance_id_info
->scope
= scope
;
1407 gcm_client()->Register(
1408 make_linked_ptr
<RegistrationInfo
>(instance_id_info
.release()));
1411 void GCMClientInstanceIDTest::DeleteToken(const std::string
& app_id
,
1412 const std::string
& authorized_entity
,
1413 const std::string
& scope
) {
1414 scoped_ptr
<InstanceIDTokenInfo
> instance_id_info(new InstanceIDTokenInfo
);
1415 instance_id_info
->app_id
= app_id
;
1416 instance_id_info
->authorized_entity
= authorized_entity
;
1417 instance_id_info
->scope
= scope
;
1418 gcm_client()->Unregister(
1419 make_linked_ptr
<RegistrationInfo
>(instance_id_info
.release()));
1422 void GCMClientInstanceIDTest::CompleteDeleteToken() {
1423 std::string
response(kDeleteTokenResponse
);
1424 net::TestURLFetcher
* fetcher
= url_fetcher_factory()->GetFetcherByID(0);
1425 ASSERT_TRUE(fetcher
);
1426 fetcher
->set_response_code(net::HTTP_OK
);
1427 fetcher
->SetResponseString(response
);
1428 fetcher
->delegate()->OnURLFetchComplete(fetcher
);
1429 // Give a chance for GCMStoreImpl::Backend to finish persisting data.
1430 PumpLoopUntilIdle();
1433 bool GCMClientInstanceIDTest::ExistsToken(const std::string
& app_id
,
1434 const std::string
& authorized_entity
,
1435 const std::string
& scope
) const {
1436 scoped_ptr
<InstanceIDTokenInfo
> instance_id_info(new InstanceIDTokenInfo
);
1437 instance_id_info
->app_id
= app_id
;
1438 instance_id_info
->authorized_entity
= authorized_entity
;
1439 instance_id_info
->scope
= scope
;
1440 return gcm_client()->registrations_
.count(
1441 make_linked_ptr
<RegistrationInfo
>(instance_id_info
.release())) > 0;
1444 TEST_F(GCMClientInstanceIDTest
, GetToken
) {
1445 AddInstanceID(kAppId
, kInstanceID
);
1448 EXPECT_FALSE(ExistsToken(kAppId
, kSender
, kScope
));
1449 GetToken(kAppId
, kSender
, kScope
);
1450 CompleteRegistration("token1");
1452 EXPECT_EQ(REGISTRATION_COMPLETED
, last_event());
1453 EXPECT_EQ(kAppId
, last_app_id());
1454 EXPECT_EQ("token1", last_registration_id());
1455 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
1456 EXPECT_TRUE(ExistsToken(kAppId
, kSender
, kScope
));
1458 // Get another token.
1459 EXPECT_FALSE(ExistsToken(kAppId
, kSender2
, kScope
));
1460 GetToken(kAppId
, kSender2
, kScope
);
1461 CompleteRegistration("token2");
1463 EXPECT_EQ(REGISTRATION_COMPLETED
, last_event());
1464 EXPECT_EQ(kAppId
, last_app_id());
1465 EXPECT_EQ("token2", last_registration_id());
1466 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
1467 EXPECT_TRUE(ExistsToken(kAppId
, kSender2
, kScope
));
1468 // The 1st token still exists.
1469 EXPECT_TRUE(ExistsToken(kAppId
, kSender
, kScope
));
1472 TEST_F(GCMClientInstanceIDTest
, DeleteInvalidToken
) {
1473 AddInstanceID(kAppId
, kInstanceID
);
1475 // Delete an invalid token.
1476 DeleteToken(kAppId
, "Foo@#$", kScope
);
1477 PumpLoopUntilIdle();
1479 EXPECT_EQ(UNREGISTRATION_COMPLETED
, last_event());
1480 EXPECT_EQ(kAppId
, last_app_id());
1481 EXPECT_EQ(GCMClient::INVALID_PARAMETER
, last_result());
1485 // Delete a non-existing token.
1486 DeleteToken(kAppId
, kSender
, kScope
);
1487 PumpLoopUntilIdle();
1489 EXPECT_EQ(UNREGISTRATION_COMPLETED
, last_event());
1490 EXPECT_EQ(kAppId
, last_app_id());
1491 EXPECT_EQ(GCMClient::INVALID_PARAMETER
, last_result());
1494 TEST_F(GCMClientInstanceIDTest
, DeleteSingleToken
) {
1495 AddInstanceID(kAppId
, kInstanceID
);
1498 EXPECT_FALSE(ExistsToken(kAppId
, kSender
, kScope
));
1499 GetToken(kAppId
, kSender
, kScope
);
1500 CompleteRegistration("token1");
1502 EXPECT_EQ(REGISTRATION_COMPLETED
, last_event());
1503 EXPECT_EQ(kAppId
, last_app_id());
1504 EXPECT_EQ("token1", last_registration_id());
1505 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
1506 EXPECT_TRUE(ExistsToken(kAppId
, kSender
, kScope
));
1510 // Get another token.
1511 EXPECT_FALSE(ExistsToken(kAppId
, kSender2
, kScope
));
1512 GetToken(kAppId
, kSender2
, kScope
);
1513 CompleteRegistration("token2");
1515 EXPECT_EQ(REGISTRATION_COMPLETED
, last_event());
1516 EXPECT_EQ(kAppId
, last_app_id());
1517 EXPECT_EQ("token2", last_registration_id());
1518 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
1519 EXPECT_TRUE(ExistsToken(kAppId
, kSender2
, kScope
));
1520 // The 1st token still exists.
1521 EXPECT_TRUE(ExistsToken(kAppId
, kSender
, kScope
));
1525 // Delete the 2nd token.
1526 DeleteToken(kAppId
, kSender2
, kScope
);
1527 CompleteDeleteToken();
1529 EXPECT_EQ(UNREGISTRATION_COMPLETED
, last_event());
1530 EXPECT_EQ(kAppId
, last_app_id());
1531 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
1532 // The 2nd token is gone while the 1st token still exists.
1533 EXPECT_TRUE(ExistsToken(kAppId
, kSender
, kScope
));
1534 EXPECT_FALSE(ExistsToken(kAppId
, kSender2
, kScope
));
1538 // Delete the 1st token.
1539 DeleteToken(kAppId
, kSender
, kScope
);
1540 CompleteDeleteToken();
1542 EXPECT_EQ(UNREGISTRATION_COMPLETED
, last_event());
1543 EXPECT_EQ(kAppId
, last_app_id());
1544 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
1545 // Both tokens are gone now.
1546 EXPECT_FALSE(ExistsToken(kAppId
, kSender
, kScope
));
1547 EXPECT_FALSE(ExistsToken(kAppId
, kSender
, kScope
));
1551 // Trying to delete the token again will get an error.
1552 DeleteToken(kAppId
, kSender
, kScope
);
1553 PumpLoopUntilIdle();
1555 EXPECT_EQ(UNREGISTRATION_COMPLETED
, last_event());
1556 EXPECT_EQ(kAppId
, last_app_id());
1557 EXPECT_EQ(GCMClient::INVALID_PARAMETER
, last_result());
1560 TEST_F(GCMClientInstanceIDTest
, DeleteAllTokens
) {
1561 AddInstanceID(kAppId
, kInstanceID
);
1564 EXPECT_FALSE(ExistsToken(kAppId
, kSender
, kScope
));
1565 GetToken(kAppId
, kSender
, kScope
);
1566 CompleteRegistration("token1");
1568 EXPECT_EQ(REGISTRATION_COMPLETED
, last_event());
1569 EXPECT_EQ(kAppId
, last_app_id());
1570 EXPECT_EQ("token1", last_registration_id());
1571 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
1572 EXPECT_TRUE(ExistsToken(kAppId
, kSender
, kScope
));
1576 // Get another token.
1577 EXPECT_FALSE(ExistsToken(kAppId
, kSender2
, kScope
));
1578 GetToken(kAppId
, kSender2
, kScope
);
1579 CompleteRegistration("token2");
1581 EXPECT_EQ(REGISTRATION_COMPLETED
, last_event());
1582 EXPECT_EQ(kAppId
, last_app_id());
1583 EXPECT_EQ("token2", last_registration_id());
1584 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
1585 EXPECT_TRUE(ExistsToken(kAppId
, kSender2
, kScope
));
1586 // The 1st token still exists.
1587 EXPECT_TRUE(ExistsToken(kAppId
, kSender
, kScope
));
1591 // Delete all tokens.
1592 DeleteToken(kAppId
, "*", "*");
1593 CompleteDeleteToken();
1595 EXPECT_EQ(UNREGISTRATION_COMPLETED
, last_event());
1596 EXPECT_EQ(kAppId
, last_app_id());
1597 EXPECT_EQ(GCMClient::SUCCESS
, last_result());
1598 // All tokens are gone now.
1599 EXPECT_FALSE(ExistsToken(kAppId
, kSender
, kScope
));
1600 EXPECT_FALSE(ExistsToken(kAppId
, kSender
, kScope
));
1603 TEST_F(GCMClientInstanceIDTest
, DeleteAllTokensBeforeGetAnyToken
) {
1604 AddInstanceID(kAppId
, kInstanceID
);
1606 // Delete all tokens without getting a token first.
1607 DeleteToken(kAppId
, "*", "*");
1608 // No need to call CompleteDeleteToken since unregistration request should
1609 // not be triggered.
1610 PumpLoopUntilIdle();
1612 EXPECT_EQ(UNREGISTRATION_COMPLETED
, last_event());
1613 EXPECT_EQ(kAppId
, last_app_id());
1614 EXPECT_EQ(GCMClient::SUCCESS
, last_result());