Reset the GCM store to recover from it when it is busted

components/gcm_driver/gcm_client_impl_unittest.cc

 // 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_client_impl.h"
 
 #include "base/command_line.h"
+#include "base/files/file_path.h"
+#include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/time/clock.h"
 #include "base/timer/timer.h"
 #include "google_apis/gcm/base/fake_encryptor.h"
 #include "google_apis/gcm/base/mcs_message.h"
 #include "google_apis/gcm/base/mcs_util.h"
@@ skipping 19 matching lines @@
   REGISTRATION_COMPLETED,
   UNREGISTRATION_COMPLETED,
   MESSAGE_SEND_ERROR,
   MESSAGE_SEND_ACK,
   MESSAGE_RECEIVED,
   MESSAGES_DELETED,
 };
 
 const uint64 kDeviceAndroidId = 54321;
 const uint64 kDeviceSecurityToken = 12345;
+const uint64 kDeviceAndroidId2 = 11111;
+const uint64 kDeviceSecurityToken2 = 2222;
 const int64 kSettingsCheckinInterval = 16 * 60 * 60;
 const char kAppId[] = "app_id";
 const char kSender[] = "project_id";
 const char kSender2[] = "project_id2";
 const char kSender3[] = "project_id3";
 const char kRegistrationResponsePrefix[] = "token=";
 const char kUnregistrationResponsePrefix[] = "deleted=";
 
 // Helper for building arbitrary data messages.
 MCSMessage BuildDownstreamMessage(
@@ skipping 270 matching lines @@
     return last_token_fetch_time_;
   }
   const std::vector<AccountMapping>& last_account_mappings() {
     return last_account_mappings_;
   }
 
   const GServicesSettings& gservices_settings() const {
     return gcm_client_->gservices_settings_;
   }
 
+  const base::FilePath& temp_directory_path() const {
+    return temp_directory_.path();
+  }
+
   int64 CurrentTime();
 
   // Tooling.
   void PumpLoop();
   void PumpLoopUntilIdle();
   void QuitLoop();
   void InitializeLoop();
   bool CreateUniqueTempDir();
   AutoAdvancingTestClock* clock() const {
     return reinterpret_cast<AutoAdvancingTestClock*>(gcm_client_->clock_.get());
@@ skipping 249 matching lines @@
   EXPECT_EQ(kDeviceSecurityToken, mcs_client()->last_security_token());
 
   // Checking freshly loaded CheckinInfo.
   EXPECT_EQ(kDeviceAndroidId, device_checkin_info().android_id);
   EXPECT_EQ(kDeviceSecurityToken, device_checkin_info().secret);
   EXPECT_TRUE(device_checkin_info().last_checkin_accounts.empty());
   EXPECT_TRUE(device_checkin_info().accounts_set);
   EXPECT_TRUE(device_checkin_info().account_tokens.empty());
 }
 
+TEST_F(GCMClientImplTest, LoadingBusted) {
+  // Close the GCM store.
+  gcm_client()->Stop();
+  PumpLoopUntilIdle();
+
+  // Mess up the store.
+  base::FilePath store_file_path =
+      temp_directory_path().Append(FILE_PATH_LITERAL("CURRENT"));
+  ASSERT_TRUE(base::AppendToFile(store_file_path, "A", 1));
+
+  // Restart GCM client. The store should be resetted and the loading should

[fgorski, 2015/02/10 18:31:31]

reset

[jianli, 2015/02/10 19:45:14]

Done.

+  // complete successfully.
+  reset_last_event();
+  BuildGCMClient(base::TimeDelta());
+  InitializeGCMClient();
+  StartGCMClient();
+  CompleteCheckin(kDeviceAndroidId2,
+                  kDeviceSecurityToken2,
+                  std::string(),
+                  std::map<std::string, std::string>());
+
+  EXPECT_EQ(LOADING_COMPLETED, last_event());
+  EXPECT_EQ(kDeviceAndroidId2, mcs_client()->last_android_id());
+  EXPECT_EQ(kDeviceSecurityToken2, mcs_client()->last_security_token());
+}
+
 TEST_F(GCMClientImplTest, RegisterApp) {
   EXPECT_FALSE(ExistsRegistration(kAppId));
 
   std::vector<std::string> senders;
   senders.push_back("sender");
   gcm_client()->Register(kAppId, senders);
   CompleteRegistration("reg_id");
 
   EXPECT_EQ(REGISTRATION_COMPLETED, last_event());
   EXPECT_EQ(kAppId, last_app_id());
@@ skipping 545 matching lines @@
   ASSERT_EQ(1UL, last_account_mappings().size());
   const AccountMapping& actual_mapping = last_account_mappings()[0];
   EXPECT_EQ(expected_mapping.account_id, actual_mapping.account_id);
   EXPECT_EQ(expected_mapping.email, actual_mapping.email);
   EXPECT_EQ(expected_mapping.status, actual_mapping.status);
   EXPECT_EQ(expected_mapping.status_change_timestamp,
             actual_mapping.status_change_timestamp);
 }
 
 }  // namespace gcm