[GCM] Adding handling of heartbeat intervals to the 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 592 matching lines @@
   iter = load_result->account_mappings.begin();
   EXPECT_EQ(account_mapping2.account_id, iter->account_id);
   EXPECT_EQ(account_mapping2.email, iter->email);
   EXPECT_TRUE(iter->access_token.empty());
   EXPECT_EQ(AccountMapping::REMOVING, iter->status);
   EXPECT_EQ(account_mapping2.status_change_timestamp,
             iter->status_change_timestamp);
   EXPECT_EQ(account_mapping2.last_message_id, iter->last_message_id);
 }
 
+TEST_F(GCMStoreImplTest, HeartbeatInterval) {
+  scoped_ptr<GCMStore> gcm_store(BuildGCMStore());
+  scoped_ptr<GCMStore::LoadResult> load_result;
+  gcm_store->Load(base::Bind(
+      &GCMStoreImplTest::LoadCallback, base::Unretained(this), &load_result));
+
+  std::string scope1 = "scope1";
+  std::string scope2 = "scope2";
+  int heartbeat1 = 120 * 1000;
+  int heartbeat2 = 360 * 1000;
+
+  gcm_store->AddHeartbeatInterval(
+      scope1,
+      heartbeat1,
+      base::Bind(&GCMStoreImplTest::UpdateCallback, base::Unretained(this)));
+  PumpLoop();
+  gcm_store->AddHeartbeatInterval(
+      scope2,
+      heartbeat2,
+      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(2UL, load_result->heartbeat_intervals.size());
+  ASSERT_TRUE(load_result->heartbeat_intervals.find(scope1) !=
+              load_result->heartbeat_intervals.end());
+  EXPECT_EQ(heartbeat1, load_result->heartbeat_intervals[scope1]);
+  ASSERT_TRUE(load_result->heartbeat_intervals.find(scope2) !=
+              load_result->heartbeat_intervals.end());
+  EXPECT_EQ(heartbeat2, load_result->heartbeat_intervals[scope2]);
+
+  gcm_store->RemoveHeartbeatInterval(
+      scope2,
+      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(1UL, load_result->heartbeat_intervals.size());
+  ASSERT_TRUE(load_result->heartbeat_intervals.find(scope1) !=
+              load_result->heartbeat_intervals.end());
+  EXPECT_EQ(heartbeat1, load_result->heartbeat_intervals[scope1]);
+}
+
 // When the database is destroyed, all database updates should fail. At the
 // same time, they per-app message counts should not go up, as failures should
 // result in decrementing the counts.
 TEST_F(GCMStoreImplTest, AddMessageAfterDestroy) {
   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();
@@ skipping 63 matching lines @@
   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);
 }
 
 }  // namespace
 
 }  // namespace gcm