Persist Instance ID data to GCM store.

components/gcm_driver/instance_id/instance_id_driver_unittest.cc

 // Copyright 2015 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/instance_id/instance_id_driver.h"
 
 #include <cmath>
 #include "base/bind.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
@@ skipping 16 matching lines @@
     return false;
 
   // Checks if it is URL-safe base64 encoded.
   for (auto ch : str) {
     if (!IsAsciiAlpha(ch) && !IsAsciiDigit(ch) && ch != '_' && ch != '-')
       return false;
   }
   return true;
 }
 
+class FakeGCMDriverForInstanceID : public gcm::FakeGCMDriver,
+                                   public gcm::InstanceIDStore {
+ public:
+  FakeGCMDriverForInstanceID();
+  ~FakeGCMDriverForInstanceID() override;
+
+  // FakeGCMDriver overrides:
+  gcm::InstanceIDStore* GetInstanceIDStore() override;
+
+  // InstanceIDStore overrides:
+  void AddInstanceIDData(const std::string& app_id,
+                         const std::string& instance_id_data) override;
+  void RemoveInstanceIDData(const std::string& app_id) override;
+  void GetInstanceIDData(
+      const std::string& app_id,
+      const gcm::InstanceIDStore::GetInstanceIDDataCallback& callback) override;
+
+ private:
+  std::map<std::string, std::string> instance_id_data_;
+
+  DISALLOW_COPY_AND_ASSIGN(FakeGCMDriverForInstanceID);
+};
+
+FakeGCMDriverForInstanceID::FakeGCMDriverForInstanceID() {
+}
+
+FakeGCMDriverForInstanceID::~FakeGCMDriverForInstanceID() {
+}
+
+gcm::InstanceIDStore* FakeGCMDriverForInstanceID::GetInstanceIDStore() {
+  return this;
+}
+
+void FakeGCMDriverForInstanceID::AddInstanceIDData(
+    const std::string& app_id,
+    const std::string& instance_id_data) {
+  instance_id_data_[app_id] = instance_id_data;
+}
+
+void FakeGCMDriverForInstanceID::RemoveInstanceIDData(
+    const std::string& app_id) {
+  instance_id_data_.erase(app_id);
+}
+
+void FakeGCMDriverForInstanceID::GetInstanceIDData(
+    const std::string& app_id,
+    const gcm::InstanceIDStore::GetInstanceIDDataCallback& callback) {
+  std::string data;
+  auto iter = instance_id_data_.find(app_id);
+  if (iter != instance_id_data_.cend())
+    data = iter->second;
+  base::MessageLoop::current()->PostTask(
+      FROM_HERE,
+      base::Bind(callback, data));
+}
+
 }  // namespace
 
 class InstanceIDDriverTest : public testing::Test {
  public:
   InstanceIDDriverTest();
   ~InstanceIDDriverTest() override;
 
   // testing::Test:
   void SetUp() override;
 
   void WaitForAsyncOperation();
 
+  // Recreates InstanceIDDriver to simulate restart.
+  void RecreateInstanceIDDriver();
+
+  // Sync wrappers for async version.
+  std::string GetID(InstanceID* instance_id);
+  base::Time GetCreationTime(InstanceID* instance_id);
+  InstanceID::Result DeleteID(InstanceID* instance_id);
+
+  InstanceIDDriver* driver() const { return driver_.get(); }
+
+ private:
+  void GetIDCompleted(const std::string& id);
+  void GetCreationTimeCompleted(const base::Time& creation_time);
   void DeleteIDCompleted(InstanceID::Result result);
 
-  InstanceIDDriver* driver() const { return driver_.get(); }
-  InstanceID::Result delete_id_result() const { return delete_id_result_; }
+  base::MessageLoopForUI message_loop_;
+  scoped_ptr<FakeGCMDriverForInstanceID> gcm_driver_;
+  scoped_ptr<InstanceIDDriver> driver_;
 
- private:
-  base::MessageLoopForUI message_loop_;
-  scoped_ptr<gcm::FakeGCMDriver> gcm_driver_;
-  scoped_ptr<InstanceIDDriver> driver_;
-  InstanceID::Result delete_id_result_;
+  std::string id_;
+  base::Time creation_time_;
+  InstanceID::Result result_;
+
+  bool async_operation_completed_;
   base::Closure async_operation_completed_callback_;
 
   DISALLOW_COPY_AND_ASSIGN(InstanceIDDriverTest);
 };
 
 InstanceIDDriverTest::InstanceIDDriverTest()
-    : delete_id_result_(InstanceID::UNKNOWN_ERROR) {
+    : result_(InstanceID::UNKNOWN_ERROR),
+      async_operation_completed_(false) {
 }
 
 InstanceIDDriverTest::~InstanceIDDriverTest() {
 }
 
 void InstanceIDDriverTest::SetUp() {
   gcm_driver_.reset(new gcm::FakeGCMDriver);
+  RecreateInstanceIDDriver();
+}
+
+void InstanceIDDriverTest::RecreateInstanceIDDriver() {
   driver_.reset(new InstanceIDDriver(gcm_driver_.get()));
 }
 
 void InstanceIDDriverTest::WaitForAsyncOperation() {
+  // No need to wait if async operation is not needed.
+  if (async_operation_completed_)
+    return;
   base::RunLoop run_loop;
   async_operation_completed_callback_ = run_loop.QuitClosure();
   run_loop.Run();
 }
 
-void InstanceIDDriverTest::DeleteIDCompleted(InstanceID::Result result) {
-  delete_id_result_ = result;
+std::string InstanceIDDriverTest::GetID(InstanceID* instance_id) {
+  async_operation_completed_ = false;
+  id_.clear();
+  instance_id->GetID(base::Bind(&InstanceIDDriverTest::GetIDCompleted,
+                     base::Unretained(this)));
+  WaitForAsyncOperation();
+  return id_;
+}
+
+base::Time InstanceIDDriverTest::GetCreationTime(InstanceID* instance_id) {
+  async_operation_completed_ = false;
+  creation_time_ = base::Time();
+  instance_id->GetCreationTime(
+      base::Bind(&InstanceIDDriverTest::GetCreationTimeCompleted,
+                 base::Unretained(this)));
+  WaitForAsyncOperation();
+  return creation_time_;
+}
+
+InstanceID::Result InstanceIDDriverTest::DeleteID(InstanceID* instance_id) {
+  async_operation_completed_ = false;
+  result_ = InstanceID::UNKNOWN_ERROR;;
+  instance_id->DeleteID(base::Bind(&InstanceIDDriverTest::DeleteIDCompleted,
+                        base::Unretained(this)));
+  WaitForAsyncOperation();
+  return result_;
+}
+
+void InstanceIDDriverTest::GetIDCompleted(const std::string& id) {
+  async_operation_completed_ = true;
+  id_ = id;
   if (!async_operation_completed_callback_.is_null())
     async_operation_completed_callback_.Run();
 }
+
+void InstanceIDDriverTest::GetCreationTimeCompleted(
+    const base::Time& creation_time) {
+  async_operation_completed_ = true;
+  creation_time_ = creation_time;
+  if (!async_operation_completed_callback_.is_null())
+    async_operation_completed_callback_.Run();
+}
+
+void InstanceIDDriverTest::DeleteIDCompleted(InstanceID::Result result) {
+  async_operation_completed_ = true;
+  result_ = result;
+  if (!async_operation_completed_callback_.is_null())
+    async_operation_completed_callback_.Run();
+}
 
 TEST_F(InstanceIDDriverTest, NewID) {
   // Creation time should not be set when the ID is not created.
   InstanceID* instance_id1 = driver()->GetInstanceID(kTestAppID1);
-  EXPECT_TRUE(instance_id1->GetCreationTime().is_null());
+  EXPECT_TRUE(GetCreationTime(instance_id1).is_null());
 
   // New ID is generated for the first time.
-  std::string id1 = instance_id1->GetID();
-  EXPECT_FALSE(id1.empty());
+  std::string id1 = GetID(instance_id1);
   EXPECT_TRUE(VerifyInstanceID(id1));
-  base::Time creation_time = instance_id1->GetCreationTime();
+  base::Time creation_time = GetCreationTime(instance_id1);
   EXPECT_FALSE(creation_time.is_null());
 
   // Same ID is returned for the same app.
-  EXPECT_EQ(id1, instance_id1->GetID());
-  EXPECT_EQ(creation_time, instance_id1->GetCreationTime());
+  EXPECT_EQ(id1, GetID(instance_id1));
+  EXPECT_EQ(creation_time, GetCreationTime(instance_id1));
 
   // New ID is generated for another app.
   InstanceID* instance_id2 = driver()->GetInstanceID(kTestAppID2);
-  std::string id2 = instance_id2->GetID();
-  EXPECT_FALSE(id2.empty());
+  std::string id2 = GetID(instance_id2);
   EXPECT_TRUE(VerifyInstanceID(id2));
   EXPECT_NE(id1, id2);
-  EXPECT_FALSE(instance_id2->GetCreationTime().is_null());
+  EXPECT_FALSE(GetCreationTime(instance_id2).is_null());
+}
+
+TEST_F(InstanceIDDriverTest, PersistID) {
+  InstanceID* instance_id = driver()->GetInstanceID(kTestAppID1);
+
+  // Create the ID for the first time. The ID and creation time should be saved
+  // to the store.
+  std::string id = GetID(instance_id);
+  EXPECT_FALSE(id.empty());
+  base::Time creation_time = GetCreationTime(instance_id);
+  EXPECT_FALSE(creation_time.is_null());
+
+  // Simulate restart by recreating InstanceIDDriver. Same ID and creation time
+  // should be expected.
+  RecreateInstanceIDDriver();
+  instance_id = driver()->GetInstanceID(kTestAppID1);
+  EXPECT_EQ(creation_time, GetCreationTime(instance_id));
+  EXPECT_EQ(id, GetID(instance_id));
+
+  // Delete the ID. The ID and creation time should be removed from the store.
+  EXPECT_EQ(InstanceID::SUCCESS, DeleteID(instance_id));
+  EXPECT_TRUE(GetCreationTime(instance_id).is_null());
+
+  // Simulate restart by recreating InstanceIDDriver. Different ID and creation
+  // time should be expected.
+  RecreateInstanceIDDriver();
+  instance_id = driver()->GetInstanceID(kTestAppID1);
+  EXPECT_NE(id, GetID(instance_id));
+  EXPECT_NE(creation_time, GetCreationTime(instance_id));
 }
 
 TEST_F(InstanceIDDriverTest, DeleteID) {
   InstanceID* instance_id = driver()->GetInstanceID(kTestAppID1);
-  std::string id1 = instance_id->GetID();
+  std::string id1 = GetID(instance_id);
   EXPECT_FALSE(id1.empty());
-  EXPECT_FALSE(instance_id->GetCreationTime().is_null());
+  EXPECT_FALSE(GetCreationTime(instance_id).is_null());
 
   // New ID will be generated from GetID after calling DeleteID.
-  instance_id->DeleteID(base::Bind(&InstanceIDDriverTest::DeleteIDCompleted,
-                                   base::Unretained(this)));
-  WaitForAsyncOperation();
-  EXPECT_EQ(InstanceID::SUCCESS, delete_id_result());
-  EXPECT_TRUE(instance_id->GetCreationTime().is_null());
+  EXPECT_EQ(InstanceID::SUCCESS, DeleteID(instance_id));
+  EXPECT_TRUE(GetCreationTime(instance_id).is_null());
 
-  std::string id2 = instance_id->GetID();
+  std::string id2 = GetID(instance_id);
   EXPECT_FALSE(id2.empty());
   EXPECT_NE(id1, id2);
-  EXPECT_FALSE(instance_id->GetCreationTime().is_null());
+  EXPECT_FALSE(GetCreationTime(instance_id).is_null());
 }
 
 }  // instance_id