Persist Instance ID data to GCM store.

google_apis/gcm/engine/gcm_store_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 "google_apis/gcm/engine/gcm_store_impl.h"
 
 #include <string>
 #include <vector>
 
 #include "base/bind.h"
@@ skipping 15 matching lines @@
 namespace {
 
 // Number of persistent ids to use in tests.
 const int kNumPersistentIds = 10;
 
 // Number of per-app messages in tests.
 const int kNumMessagesPerApp = 20;
 
 // App name for testing.
 const char kAppName[] = "my_app";
+const char kAppName2[] = "my_app_2";
 
 // Category name for testing.
 const char kCategoryName[] = "my_category";
 
 const uint64 kDeviceId = 22;
 const uint64 kDeviceToken = 55;
 
 class GCMStoreImplTest : public testing::Test {
  public:
   GCMStoreImplTest();
@@ skipping 688 matching lines @@
       base::Bind(&GCMStoreImplTest::UpdateCallback, base::Unretained(this)));
   PumpLoop();
 
   gcm_store = BuildGCMStore().Pass();
   gcm_store->Load(base::Bind(
       &GCMStoreImplTest::LoadCallback, base::Unretained(this), &load_result));
   PumpLoop();
   EXPECT_EQ(base::Time(), load_result->last_token_fetch_time);
 }
 
+TEST_F(GCMStoreImplTest, InstanceIDData) {
+  scoped_ptr<GCMStore> gcm_store(BuildGCMStore());
+  scoped_ptr<GCMStore::LoadResult> load_result;
+  gcm_store->Load(base::Bind(
+      &GCMStoreImplTest::LoadCallback, base::Unretained(this), &load_result));
+  PumpLoop();
+
+  std::string instance_id_data("Foo");
+  gcm_store->AddInstanceIDData(
+      kAppName,
+      instance_id_data,
+      base::Bind(&GCMStoreImplTest::UpdateCallback, base::Unretained(this)));
+  PumpLoop();
+
+  std::string instance_id_data2("Hello Instance ID");
+  gcm_store->AddInstanceIDData(
+      kAppName2,
+      instance_id_data2,
+      base::Bind(&GCMStoreImplTest::UpdateCallback, base::Unretained(this)));
+  PumpLoop();
+
+  gcm_store = BuildGCMStore().Pass();
+  gcm_store->Load(base::Bind(
+      &GCMStoreImplTest::LoadCallback, base::Unretained(this), &load_result));
+  PumpLoop();
+
+  ASSERT_EQ(2u, load_result->instance_id_data.size());
+  ASSERT_TRUE(load_result->instance_id_data.find(kAppName) !=
+              load_result->instance_id_data.end());
+  ASSERT_TRUE(load_result->instance_id_data.find(kAppName2) !=
+              load_result->instance_id_data.end());
+  EXPECT_EQ(instance_id_data, load_result->instance_id_data[kAppName]);
+  EXPECT_EQ(instance_id_data2, load_result->instance_id_data[kAppName2]);
+
+  gcm_store->RemoveInstanceIDData(
+      kAppName,
+      base::Bind(&GCMStoreImplTest::UpdateCallback, base::Unretained(this)));
+  PumpLoop();
+
+  gcm_store = BuildGCMStore().Pass();
+  gcm_store->Load(base::Bind(
+      &GCMStoreImplTest::LoadCallback, base::Unretained(this), &load_result));
+  PumpLoop();
+
+  ASSERT_EQ(1u, load_result->instance_id_data.size());
+  ASSERT_TRUE(load_result->instance_id_data.find(kAppName2) !=
+              load_result->instance_id_data.end());
+  EXPECT_EQ(instance_id_data2, load_result->instance_id_data[kAppName2]);
+}
+
 }  // namespace
 
 }  // namespace gcm