| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "components/gcm_driver/gcm_account_mapper.h" |
| |
| #include "base/test/simple_test_clock.h" |
| #include "base/time/time.h" |
| #include "components/gcm_driver/fake_gcm_driver.h" |
| #include "google_apis/gcm/engine/account_mapping.h" |
| #include "google_apis/gcm/engine/gcm_store.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace gcm { |
| |
| namespace { |
| |
| const char kGCMAccountMapperSenderId[] = "745476177629"; |
| const char kGCMAccountMapperAppId[] = "com.google.android.gms"; |
| const char kRegistrationId[] = "reg_id"; |
| |
| AccountMapping MakeAccountMapping(const std::string& account_id, |
| AccountMapping::MappingStatus status, |
| const base::Time& status_change_timestamp, |
| const std::string& last_message_id) { |
| AccountMapping account_mapping; |
| account_mapping.account_id = account_id; |
| account_mapping.email = account_id + "@gmail.com"; |
| // account_mapping.access_token intentionally left empty. |
| account_mapping.status = status; |
| account_mapping.status_change_timestamp = status_change_timestamp; |
| account_mapping.last_message_id = last_message_id; |
| return account_mapping; |
| } |
| |
| GCMClient::AccountTokenInfo MakeAccountTokenInfo( |
| const std::string& account_id) { |
| GCMClient::AccountTokenInfo account_token; |
| account_token.account_id = account_id; |
| account_token.email = account_id + "@gmail.com"; |
| account_token.access_token = account_id + "_token"; |
| return account_token; |
| } |
| |
| class CustomFakeGCMDriver : public FakeGCMDriver { |
| public: |
| CustomFakeGCMDriver(); |
| virtual ~CustomFakeGCMDriver(); |
| |
| virtual void UpdateAccountMapping( |
| const AccountMapping& account_mapping) OVERRIDE; |
| virtual void RemoveAccountMapping(const std::string& account_id) OVERRIDE; |
| virtual void AddAppHandler(const std::string& app_id, |
| GCMAppHandler* handler) OVERRIDE; |
| virtual void RemoveAppHandler(const std::string& app_id) OVERRIDE; |
| |
| void CompleteSend(const std::string& message_id, GCMClient::Result result); |
| void SendAcknowledged(const std::string& message_id); |
| void MessageSendError(const std::string& message_id); |
| |
| const AccountMapping& last_account_mapping() const { |
| return account_mapping_; |
| } |
| const std::string& last_message_id() const { return last_message_id_; } |
| const std::string& last_removed_account_id() const { |
| return last_removed_account_id_; |
| } |
| |
| protected: |
| virtual void SendImpl(const std::string& app_id, |
| const std::string& receiver_id, |
| const GCMClient::OutgoingMessage& message) OVERRIDE; |
| |
| private: |
| AccountMapping account_mapping_; |
| std::string last_message_id_; |
| std::string last_removed_account_id_; |
| }; |
| |
| CustomFakeGCMDriver::CustomFakeGCMDriver() { |
| } |
| |
| CustomFakeGCMDriver::~CustomFakeGCMDriver() { |
| } |
| |
| void CustomFakeGCMDriver::UpdateAccountMapping( |
| const AccountMapping& account_mapping) { |
| account_mapping_.email = account_mapping.email; |
| account_mapping_.account_id = account_mapping.account_id; |
| account_mapping_.access_token = account_mapping.access_token; |
| account_mapping_.status = account_mapping.status; |
| account_mapping_.status_change_timestamp = |
| account_mapping.status_change_timestamp; |
| account_mapping_.last_message_id = account_mapping.last_message_id; |
| } |
| |
| void CustomFakeGCMDriver::RemoveAccountMapping(const std::string& account_id) { |
| last_removed_account_id_ = account_id; |
| } |
| |
| void CustomFakeGCMDriver::AddAppHandler(const std::string& app_id, |
| GCMAppHandler* handler) { |
| GCMDriver::AddAppHandler(app_id, handler); |
| } |
| |
| void CustomFakeGCMDriver::RemoveAppHandler(const std::string& app_id) { |
| GCMDriver::RemoveAppHandler(app_id); |
| } |
| |
| void CustomFakeGCMDriver::CompleteSend(const std::string& message_id, |
| GCMClient::Result result) { |
| SendFinished(kGCMAccountMapperAppId, message_id, result); |
| } |
| |
| void CustomFakeGCMDriver::SendAcknowledged(const std::string& message_id) { |
| GetAppHandler(kGCMAccountMapperAppId) |
| ->OnSendAcknowledged(kGCMAccountMapperAppId, message_id); |
| } |
| |
| void CustomFakeGCMDriver::MessageSendError(const std::string& message_id) { |
| GCMClient::SendErrorDetails send_error; |
| send_error.message_id = message_id; |
| send_error.result = GCMClient::TTL_EXCEEDED; |
| GetAppHandler(kGCMAccountMapperAppId) |
| ->OnSendError(kGCMAccountMapperAppId, send_error); |
| } |
| |
| void CustomFakeGCMDriver::SendImpl(const std::string& app_id, |
| const std::string& receiver_id, |
| const GCMClient::OutgoingMessage& message) { |
| DCHECK_EQ(kGCMAccountMapperAppId, app_id); |
| DCHECK_EQ(kGCMAccountMapperSenderId, receiver_id); |
| |
| last_message_id_ = message.id; |
| } |
| |
| } // namespace |
| |
| class GCMAccountMapperTest : public testing::Test { |
| public: |
| GCMAccountMapperTest(); |
| virtual ~GCMAccountMapperTest(); |
| |
| void Restart(); |
| |
| const std::vector<AccountMapping>& GetAccounts() const { |
| return account_mapper_->accounts_; |
| } |
| |
| GCMAccountMapper* mapper() { return account_mapper_.get(); } |
| |
| CustomFakeGCMDriver& gcm_driver() { return gcm_driver_; } |
| |
| base::SimpleTestClock* clock() { return clock_; } |
| |
| private: |
| CustomFakeGCMDriver gcm_driver_; |
| scoped_ptr<GCMAccountMapper> account_mapper_; |
| base::SimpleTestClock* clock_; |
| }; |
| |
| GCMAccountMapperTest::GCMAccountMapperTest() { |
| Restart(); |
| } |
| |
| GCMAccountMapperTest::~GCMAccountMapperTest() { |
| } |
| |
| void GCMAccountMapperTest::Restart() { |
| if (account_mapper_) |
| account_mapper_->ShutdownHandler(); |
| account_mapper_.reset(new GCMAccountMapper(&gcm_driver_)); |
| scoped_ptr<base::SimpleTestClock> clock(new base::SimpleTestClock); |
| clock_ = clock.get(); |
| account_mapper_->SetClockForTesting(clock.PassAs<base::Clock>()); |
| } |
| |
| // Tests the initialization of account mappings (from the store) when empty. |
| TEST_F(GCMAccountMapperTest, InitializeAccountMappingsEmpty) { |
| std::vector<AccountMapping> account_mappings; |
| mapper()->Initialize(account_mappings, ""); |
| EXPECT_TRUE(GetAccounts().empty()); |
| } |
| |
| // Tests the initialization of account mappings (from the store). |
| TEST_F(GCMAccountMapperTest, InitializeAccountMappings) { |
| std::vector<AccountMapping> account_mappings; |
| AccountMapping account_mapping1 = MakeAccountMapping("acc_id1", |
| AccountMapping::MAPPED, |
| base::Time::Now(), |
| std::string()); |
| AccountMapping account_mapping2 = MakeAccountMapping("acc_id2", |
| AccountMapping::ADDING, |
| base::Time::Now(), |
| "add_message_1"); |
| account_mappings.push_back(account_mapping1); |
| account_mappings.push_back(account_mapping2); |
| |
| mapper()->Initialize(account_mappings, ""); |
| |
| std::vector<AccountMapping> mappings = GetAccounts(); |
| EXPECT_EQ(2UL, mappings.size()); |
| std::vector<AccountMapping>::const_iterator iter = mappings.begin(); |
| |
| EXPECT_EQ(account_mapping1.account_id, iter->account_id); |
| EXPECT_EQ(account_mapping1.email, iter->email); |
| EXPECT_TRUE(account_mapping1.access_token.empty()); |
| EXPECT_EQ(account_mapping1.status, iter->status); |
| EXPECT_EQ(account_mapping1.status_change_timestamp, |
| iter->status_change_timestamp); |
| EXPECT_TRUE(account_mapping1.last_message_id.empty()); |
| |
| ++iter; |
| EXPECT_EQ(account_mapping2.account_id, iter->account_id); |
| EXPECT_EQ(account_mapping2.email, iter->email); |
| EXPECT_TRUE(account_mapping2.access_token.empty()); |
| EXPECT_EQ(account_mapping2.status, iter->status); |
| EXPECT_EQ(account_mapping2.status_change_timestamp, |
| iter->status_change_timestamp); |
| EXPECT_EQ(account_mapping2.last_message_id, iter->last_message_id); |
| } |
| |
| // Tests the part where a new account is added with a token, to the point when |
| // GCM message is sent. |
| TEST_F(GCMAccountMapperTest, AddMappingToMessageSent) { |
| mapper()->Initialize(GCMAccountMapper::AccountMappings(), kRegistrationId); |
| |
| std::vector<GCMClient::AccountTokenInfo> account_tokens; |
| GCMClient::AccountTokenInfo account_token = MakeAccountTokenInfo("acc_id"); |
| account_tokens.push_back(account_token); |
| mapper()->SetAccountTokens(account_tokens); |
| |
| std::vector<AccountMapping> mappings = GetAccounts(); |
| EXPECT_EQ(1UL, mappings.size()); |
| std::vector<AccountMapping>::const_iterator iter = mappings.begin(); |
| EXPECT_EQ("acc_id", iter->account_id); |
| EXPECT_EQ("acc_id@gmail.com", iter->email); |
| EXPECT_EQ("acc_id_token", iter->access_token); |
| EXPECT_EQ(AccountMapping::NEW, iter->status); |
| EXPECT_EQ(base::Time(), iter->status_change_timestamp); |
| |
| EXPECT_TRUE(!gcm_driver().last_message_id().empty()); |
| } |
| |
| // Tests the part where GCM message is successfully queued. |
| TEST_F(GCMAccountMapperTest, AddMappingMessageQueued) { |
| mapper()->Initialize(GCMAccountMapper::AccountMappings(), kRegistrationId); |
| |
| std::vector<GCMClient::AccountTokenInfo> account_tokens; |
| GCMClient::AccountTokenInfo account_token = MakeAccountTokenInfo("acc_id"); |
| account_tokens.push_back(account_token); |
| mapper()->SetAccountTokens(account_tokens); |
| |
| clock()->SetNow(base::Time::Now()); |
| gcm_driver().CompleteSend(gcm_driver().last_message_id(), GCMClient::SUCCESS); |
| |
| EXPECT_EQ(account_token.email, gcm_driver().last_account_mapping().email); |
| EXPECT_EQ(account_token.account_id, |
| gcm_driver().last_account_mapping().account_id); |
| EXPECT_EQ(account_token.access_token, |
| gcm_driver().last_account_mapping().access_token); |
| EXPECT_EQ(AccountMapping::ADDING, gcm_driver().last_account_mapping().status); |
| EXPECT_EQ(clock()->Now(), |
| gcm_driver().last_account_mapping().status_change_timestamp); |
| EXPECT_EQ(gcm_driver().last_message_id(), |
| gcm_driver().last_account_mapping().last_message_id); |
| |
| std::vector<AccountMapping> mappings = GetAccounts(); |
| std::vector<AccountMapping>::const_iterator iter = mappings.begin(); |
| EXPECT_EQ(account_token.email, iter->email); |
| EXPECT_EQ(account_token.account_id, iter->account_id); |
| EXPECT_EQ(account_token.access_token, iter->access_token); |
| EXPECT_EQ(AccountMapping::ADDING, iter->status); |
| EXPECT_EQ(clock()->Now(), iter->status_change_timestamp); |
| EXPECT_EQ(gcm_driver().last_message_id(), iter->last_message_id); |
| } |
| |
| // Tests status change from ADDING to MAPPED (Message is acknowledged). |
| TEST_F(GCMAccountMapperTest, AddMappingMessageAcknowledged) { |
| mapper()->Initialize(GCMAccountMapper::AccountMappings(), kRegistrationId); |
| |
| std::vector<GCMClient::AccountTokenInfo> account_tokens; |
| GCMClient::AccountTokenInfo account_token = MakeAccountTokenInfo("acc_id"); |
| account_tokens.push_back(account_token); |
| mapper()->SetAccountTokens(account_tokens); |
| |
| clock()->SetNow(base::Time::Now()); |
| gcm_driver().CompleteSend(gcm_driver().last_message_id(), GCMClient::SUCCESS); |
| clock()->SetNow(base::Time::Now()); |
| gcm_driver().SendAcknowledged(gcm_driver().last_message_id()); |
| |
| EXPECT_EQ(account_token.email, gcm_driver().last_account_mapping().email); |
| EXPECT_EQ(account_token.account_id, |
| gcm_driver().last_account_mapping().account_id); |
| EXPECT_EQ(account_token.access_token, |
| gcm_driver().last_account_mapping().access_token); |
| EXPECT_EQ(AccountMapping::MAPPED, gcm_driver().last_account_mapping().status); |
| EXPECT_EQ(clock()->Now(), |
| gcm_driver().last_account_mapping().status_change_timestamp); |
| EXPECT_TRUE(gcm_driver().last_account_mapping().last_message_id.empty()); |
| |
| std::vector<AccountMapping> mappings = GetAccounts(); |
| std::vector<AccountMapping>::const_iterator iter = mappings.begin(); |
| EXPECT_EQ(account_token.email, iter->email); |
| EXPECT_EQ(account_token.account_id, iter->account_id); |
| EXPECT_EQ(account_token.access_token, iter->access_token); |
| EXPECT_EQ(AccountMapping::MAPPED, iter->status); |
| EXPECT_EQ(clock()->Now(), iter->status_change_timestamp); |
| EXPECT_TRUE(iter->last_message_id.empty()); |
| } |
| |
| // Tests status change form ADDING to MAPPED (When message was acknowledged, |
| // after Chrome was restarted). |
| TEST_F(GCMAccountMapperTest, AddMappingMessageAckedAfterRestart) { |
| mapper()->Initialize(GCMAccountMapper::AccountMappings(), kRegistrationId); |
| |
| std::vector<GCMClient::AccountTokenInfo> account_tokens; |
| GCMClient::AccountTokenInfo account_token = MakeAccountTokenInfo("acc_id"); |
| account_tokens.push_back(account_token); |
| mapper()->SetAccountTokens(account_tokens); |
| |
| clock()->SetNow(base::Time::Now()); |
| gcm_driver().CompleteSend(gcm_driver().last_message_id(), GCMClient::SUCCESS); |
| |
| Restart(); |
| std::vector<AccountMapping> stored_mappings; |
| stored_mappings.push_back(gcm_driver().last_account_mapping()); |
| mapper()->Initialize(stored_mappings, kRegistrationId); |
| |
| clock()->SetNow(base::Time::Now()); |
| gcm_driver().SendAcknowledged(gcm_driver().last_message_id()); |
| |
| EXPECT_EQ(account_token.email, gcm_driver().last_account_mapping().email); |
| EXPECT_EQ(account_token.account_id, |
| gcm_driver().last_account_mapping().account_id); |
| EXPECT_EQ(account_token.access_token, |
| gcm_driver().last_account_mapping().access_token); |
| EXPECT_EQ(AccountMapping::MAPPED, gcm_driver().last_account_mapping().status); |
| EXPECT_EQ(clock()->Now(), |
| gcm_driver().last_account_mapping().status_change_timestamp); |
| EXPECT_TRUE(gcm_driver().last_account_mapping().last_message_id.empty()); |
| |
| std::vector<AccountMapping> mappings = GetAccounts(); |
| std::vector<AccountMapping>::const_iterator iter = mappings.begin(); |
| EXPECT_EQ(account_token.email, iter->email); |
| EXPECT_EQ(account_token.account_id, iter->account_id); |
| EXPECT_EQ(account_token.access_token, iter->access_token); |
| EXPECT_EQ(AccountMapping::MAPPED, iter->status); |
| EXPECT_EQ(clock()->Now(), iter->status_change_timestamp); |
| EXPECT_TRUE(iter->last_message_id.empty()); |
| } |
| |
| // Tests a case when ADD message times out for a new account. |
| TEST_F(GCMAccountMapperTest, AddMappingMessageSendErrorForNewAccount) { |
| mapper()->Initialize(GCMAccountMapper::AccountMappings(), kRegistrationId); |
| |
| std::vector<GCMClient::AccountTokenInfo> account_tokens; |
| GCMClient::AccountTokenInfo account_token = MakeAccountTokenInfo("acc_id"); |
| account_tokens.push_back(account_token); |
| mapper()->SetAccountTokens(account_tokens); |
| |
| clock()->SetNow(base::Time::Now()); |
| gcm_driver().CompleteSend(gcm_driver().last_message_id(), GCMClient::SUCCESS); |
| |
| clock()->SetNow(base::Time::Now()); |
| std::string old_message_id = gcm_driver().last_message_id(); |
| gcm_driver().MessageSendError(old_message_id); |
| |
| // No new message is sent because of the send error, as the token is stale. |
| // Because the account was new, the entry should be deleted. |
| EXPECT_EQ(old_message_id, gcm_driver().last_message_id()); |
| EXPECT_EQ(account_token.account_id, gcm_driver().last_removed_account_id()); |
| EXPECT_TRUE(GetAccounts().empty()); |
| } |
| |
| /// Tests a case when ADD message times out for a MAPPED account. |
| TEST_F(GCMAccountMapperTest, AddMappingMessageSendErrorForMappedAccount) { |
| // Start with one account that is mapped. |
| base::Time status_change_timestamp = base::Time::Now(); |
| AccountMapping mapping = MakeAccountMapping("acc_id", |
| AccountMapping::MAPPED, |
| status_change_timestamp, |
| "add_message_id"); |
| |
| GCMAccountMapper::AccountMappings stored_mappings; |
| stored_mappings.push_back(mapping); |
| mapper()->Initialize(stored_mappings, kRegistrationId); |
| |
| clock()->SetNow(base::Time::Now()); |
| gcm_driver().MessageSendError("add_message_id"); |
| |
| // No new message is sent because of the send error, as the token is stale. |
| // Because the account was new, the entry should be deleted. |
| EXPECT_TRUE(gcm_driver().last_message_id().empty()); |
| |
| std::vector<AccountMapping> mappings = GetAccounts(); |
| std::vector<AccountMapping>::const_iterator iter = mappings.begin(); |
| EXPECT_EQ(mapping.email, iter->email); |
| EXPECT_EQ(mapping.account_id, iter->account_id); |
| EXPECT_EQ(mapping.access_token, iter->access_token); |
| EXPECT_EQ(AccountMapping::MAPPED, iter->status); |
| EXPECT_EQ(status_change_timestamp, iter->status_change_timestamp); |
| EXPECT_TRUE(iter->last_message_id.empty()); |
| } |
| |
| // Tests that a missing token for an account will trigger removing of that |
| // account. This test goes only until the message is passed to GCM. |
| TEST_F(GCMAccountMapperTest, RemoveMappingToMessageSent) { |
| // Start with one account that is mapped. |
| AccountMapping mapping = MakeAccountMapping("acc_id", |
| AccountMapping::MAPPED, |
| base::Time::Now(), |
| std::string()); |
| |
| GCMAccountMapper::AccountMappings stored_mappings; |
| stored_mappings.push_back(mapping); |
| mapper()->Initialize(stored_mappings, kRegistrationId); |
| clock()->SetNow(base::Time::Now()); |
| |
| mapper()->SetAccountTokens(std::vector<GCMClient::AccountTokenInfo>()); |
| |
| EXPECT_EQ(mapping.account_id, gcm_driver().last_account_mapping().account_id); |
| EXPECT_EQ(mapping.email, gcm_driver().last_account_mapping().email); |
| EXPECT_EQ(AccountMapping::REMOVING, |
| gcm_driver().last_account_mapping().status); |
| EXPECT_EQ(clock()->Now(), |
| gcm_driver().last_account_mapping().status_change_timestamp); |
| EXPECT_TRUE(gcm_driver().last_account_mapping().last_message_id.empty()); |
| |
| std::vector<AccountMapping> mappings = GetAccounts(); |
| std::vector<AccountMapping>::const_iterator iter = mappings.begin(); |
| EXPECT_EQ(mapping.email, iter->email); |
| EXPECT_EQ(mapping.account_id, iter->account_id); |
| EXPECT_EQ(mapping.access_token, iter->access_token); |
| EXPECT_EQ(AccountMapping::REMOVING, iter->status); |
| EXPECT_EQ(clock()->Now(), iter->status_change_timestamp); |
| EXPECT_TRUE(iter->last_message_id.empty()); |
| } |
| |
| // Tests that a missing token for an account will trigger removing of that |
| // account. This test goes until the message is queued by GCM. |
| TEST_F(GCMAccountMapperTest, RemoveMappingMessageQueued) { |
| // Start with one account that is mapped. |
| AccountMapping mapping = MakeAccountMapping("acc_id", |
| AccountMapping::MAPPED, |
| base::Time::Now(), |
| std::string()); |
| |
| GCMAccountMapper::AccountMappings stored_mappings; |
| stored_mappings.push_back(mapping); |
| mapper()->Initialize(stored_mappings, kRegistrationId); |
| clock()->SetNow(base::Time::Now()); |
| base::Time status_change_timestamp = clock()->Now(); |
| |
| mapper()->SetAccountTokens(std::vector<GCMClient::AccountTokenInfo>()); |
| clock()->SetNow(base::Time::Now()); |
| gcm_driver().CompleteSend(gcm_driver().last_message_id(), GCMClient::SUCCESS); |
| |
| EXPECT_EQ(mapping.account_id, gcm_driver().last_account_mapping().account_id); |
| EXPECT_EQ(mapping.email, gcm_driver().last_account_mapping().email); |
| EXPECT_EQ(AccountMapping::REMOVING, |
| gcm_driver().last_account_mapping().status); |
| EXPECT_EQ(status_change_timestamp, |
| gcm_driver().last_account_mapping().status_change_timestamp); |
| EXPECT_TRUE(!gcm_driver().last_account_mapping().last_message_id.empty()); |
| |
| std::vector<AccountMapping> mappings = GetAccounts(); |
| std::vector<AccountMapping>::const_iterator iter = mappings.begin(); |
| EXPECT_EQ(mapping.email, iter->email); |
| EXPECT_EQ(mapping.account_id, iter->account_id); |
| EXPECT_EQ(mapping.access_token, iter->access_token); |
| EXPECT_EQ(AccountMapping::REMOVING, iter->status); |
| EXPECT_EQ(status_change_timestamp, iter->status_change_timestamp); |
| EXPECT_EQ(gcm_driver().last_account_mapping().last_message_id, |
| iter->last_message_id); |
| } |
| |
| // Tests that a missing token for an account will trigger removing of that |
| // account. This test goes until the message is acknowledged by GCM. |
| // This is a complete success scenario for account removal, and it end with |
| // account mapping being completely gone. |
| TEST_F(GCMAccountMapperTest, RemoveMappingMessageAcknowledged) { |
| // Start with one account that is mapped. |
| AccountMapping mapping = MakeAccountMapping("acc_id", |
| AccountMapping::MAPPED, |
| base::Time::Now(), |
| std::string()); |
| |
| GCMAccountMapper::AccountMappings stored_mappings; |
| stored_mappings.push_back(mapping); |
| mapper()->Initialize(stored_mappings, kRegistrationId); |
| clock()->SetNow(base::Time::Now()); |
| |
| mapper()->SetAccountTokens(std::vector<GCMClient::AccountTokenInfo>()); |
| gcm_driver().CompleteSend(gcm_driver().last_message_id(), GCMClient::SUCCESS); |
| gcm_driver().SendAcknowledged(gcm_driver().last_message_id()); |
| |
| EXPECT_EQ(mapping.account_id, gcm_driver().last_removed_account_id()); |
| |
| std::vector<AccountMapping> mappings = GetAccounts(); |
| EXPECT_TRUE(mappings.empty()); |
| } |
| |
| // Tests that account removing proceeds, when a removing message is acked after |
| // Chrome was restarted. |
| TEST_F(GCMAccountMapperTest, RemoveMappingMessageAckedAfterRestart) { |
| // Start with one account that is mapped. |
| AccountMapping mapping = MakeAccountMapping("acc_id", |
| AccountMapping::REMOVING, |
| base::Time::Now(), |
| "remove_message_id"); |
| |
| GCMAccountMapper::AccountMappings stored_mappings; |
| stored_mappings.push_back(mapping); |
| mapper()->Initialize(stored_mappings, kRegistrationId); |
| |
| gcm_driver().SendAcknowledged("remove_message_id"); |
| |
| EXPECT_EQ(mapping.account_id, gcm_driver().last_removed_account_id()); |
| |
| std::vector<AccountMapping> mappings = GetAccounts(); |
| EXPECT_TRUE(mappings.empty()); |
| } |
| |
| // Tests that account removing proceeds, when a removing message is acked after |
| // Chrome was restarted. |
| TEST_F(GCMAccountMapperTest, RemoveMappingMessageSendError) { |
| // Start with one account that is mapped. |
| base::Time status_change_timestamp = base::Time::Now(); |
| AccountMapping mapping = MakeAccountMapping("acc_id", |
| AccountMapping::REMOVING, |
| status_change_timestamp, |
| "remove_message_id"); |
| |
| GCMAccountMapper::AccountMappings stored_mappings; |
| stored_mappings.push_back(mapping); |
| mapper()->Initialize(stored_mappings, kRegistrationId); |
| |
| clock()->SetNow(base::Time::Now()); |
| gcm_driver().MessageSendError("remove_message_id"); |
| |
| EXPECT_TRUE(gcm_driver().last_removed_account_id().empty()); |
| |
| EXPECT_EQ(mapping.account_id, gcm_driver().last_account_mapping().account_id); |
| EXPECT_EQ(mapping.email, gcm_driver().last_account_mapping().email); |
| EXPECT_EQ(AccountMapping::REMOVING, |
| gcm_driver().last_account_mapping().status); |
| EXPECT_EQ(status_change_timestamp, |
| gcm_driver().last_account_mapping().status_change_timestamp); |
| // Message is not persisted, until send is completed. |
| EXPECT_TRUE(gcm_driver().last_account_mapping().last_message_id.empty()); |
| |
| std::vector<AccountMapping> mappings = GetAccounts(); |
| std::vector<AccountMapping>::const_iterator iter = mappings.begin(); |
| EXPECT_EQ(mapping.email, iter->email); |
| EXPECT_EQ(mapping.account_id, iter->account_id); |
| EXPECT_TRUE(iter->access_token.empty()); |
| EXPECT_EQ(AccountMapping::REMOVING, iter->status); |
| EXPECT_EQ(status_change_timestamp, iter->status_change_timestamp); |
| EXPECT_TRUE(iter->last_message_id.empty()); |
| } |
| |
| } // namespace gcm |