Send GCM id to DMServer

chrome/browser/chromeos/policy/heartbeat_scheduler_unittest.cc

 // Copyright (c) 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 "chrome/browser/chromeos/policy/heartbeat_scheduler.h"
 
 #include "base/strings/string_number_conversions.h"
 #include "base/test/test_simple_task_runner.h"
 #include "chrome/browser/chromeos/settings/scoped_cros_settings_test_helper.h"
 #include "chromeos/settings/cros_settings_names.h"
 #include "components/gcm_driver/fake_gcm_driver.h"
+#include "components/policy/core/common/cloud/cloud_policy_client.h"
+#include "components/policy/core/common/cloud/mock_cloud_policy_client.h"
 #include "content/public/test/test_utils.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using ::testing::_;
 using ::testing::SaveArg;
 
 namespace {
 const char* const kFakeEnrollmentDomain = "example.com";
 const char* const kFakeDeviceId = "fake_device_id";
 const char* const kHeartbeatGCMAppID = "com.google.chromeos.monitoring";
+const char* const kRegistrationId = "registration_id";
+const char* const kDMToken = "fake_dm_token";
 
 class MockGCMDriver : public testing::StrictMock<gcm::FakeGCMDriver> {
  public:
   MockGCMDriver() {
   }
 
   ~MockGCMDriver() override {
   }
 
   MOCK_METHOD2(RegisterImpl,
                void(const std::string&, const std::vector<std::string>&));
   MOCK_METHOD3(SendImpl,
                void(const std::string&,
                     const std::string&,
                     const gcm::OutgoingMessage& message));
 
   // Helper function to complete a registration previously started by
   // Register().
   void CompleteRegistration(const std::string& app_id,
                             gcm::GCMClient::Result result) {
-    RegisterFinished(app_id, "registration_id", result);
+    RegisterFinished(app_id, kRegistrationId, result);
   }
 
   // Helper function to complete a send operation previously started by
   // Send().
   void CompleteSend(const std::string& app_id,
                     const std::string& message_id,
                     gcm::GCMClient::Result result) {
     SendFinished(app_id, message_id, result);
   }
 };
 
 class HeartbeatSchedulerTest : public testing::Test {
  public:
-  HeartbeatSchedulerTest()
-      : task_runner_(new base::TestSimpleTaskRunner()),
-        scheduler_(
-            &gcm_driver_, kFakeEnrollmentDomain, kFakeDeviceId, task_runner_) {
-  }
+  HeartbeatSchedulerTest() : task_runner_(new base::TestSimpleTaskRunner()) {}
 
   void SetUp() override {
+    InitializeScheduler(nullptr);

[Andrew T Wilson (Slow), 2015/08/06 11:17:06]

why not set a mock cloud policy client here, then you don't have to reset it
down in SendGcmIdMappingUpdate()?

Better to have consistent setup for all tests.

[binjin, 2015/08/06 13:33:01]

Done.

     settings_helper_.ReplaceProvider(chromeos::kHeartbeatEnabled);
   }
 
   void TearDown() override {
     content::RunAllBlockingPoolTasksUntilIdle();
   }
 
+  void InitializeScheduler(policy::CloudPolicyClient* cloud_policy_client) {
+    scheduler_.reset(new policy::HeartbeatScheduler(
+        &gcm_driver_, cloud_policy_client, kFakeEnrollmentDomain, kFakeDeviceId,
+        task_runner_));
+  }
+
   void CheckPendingTaskDelay(base::Time last_heartbeat,
                              base::TimeDelta expected_delay) {
     EXPECT_FALSE(last_heartbeat.is_null());
     base::Time now = base::Time::NowFromSystemTime();
     EXPECT_GE(now, last_heartbeat);
     base::TimeDelta actual_delay = task_runner_->NextPendingTaskDelay();
 
     // NextPendingTaskDelay() returns the exact original delay value the task
     // was posted with. The heartbeat task would have been calculated to fire at
     // |last_heartbeat| + |expected_delay|, but we don't know the exact time
@@ skipping 10 matching lines @@
   }
 
   base::MessageLoop loop_;
   MockGCMDriver gcm_driver_;
   chromeos::ScopedCrosSettingsTestHelper settings_helper_;
 
   // TaskRunner used to run individual tests.
   scoped_refptr<base::TestSimpleTaskRunner> task_runner_;
 
   // The HeartbeatScheduler instance under test.
-  policy::HeartbeatScheduler scheduler_;
+  scoped_ptr<policy::HeartbeatScheduler> scheduler_;
 };
 
 TEST_F(HeartbeatSchedulerTest, Basic) {
-  // Just makes sure we can spin up and shutdown the scheduler with
   // heartbeats disabled.
   settings_helper_.SetBoolean(chromeos::kHeartbeatEnabled, false);
   ASSERT_TRUE(task_runner_->GetPendingTasks().empty());
 }
 
 TEST_F(HeartbeatSchedulerTest, PermanentlyFailedGCMRegistration) {
   // If heartbeats are enabled, we should register with GCMDriver.
   EXPECT_CALL(gcm_driver_, RegisterImpl(kHeartbeatGCMAppID, _));
   settings_helper_.SetBoolean(chromeos::kHeartbeatEnabled, true);
   gcm_driver_.CompleteRegistration(
@@ skipping 41 matching lines @@
   EXPECT_CALL(gcm_driver_, SendImpl(kHeartbeatGCMAppID, _, _))
       .WillOnce(SaveArg<2>(&message));
   task_runner_->RunPendingTasks();
   EXPECT_TRUE(task_runner_->GetPendingTasks().empty());
 
   // Complete sending a message - we should queue up the next heartbeat
   // even if the previous attempt failed.
   gcm_driver_.CompleteSend(
       kHeartbeatGCMAppID, message.id, gcm::GCMClient::SERVER_ERROR);
   EXPECT_EQ(1U, task_runner_->GetPendingTasks().size());
-  CheckPendingTaskDelay(scheduler_.last_heartbeat(),
+  CheckPendingTaskDelay(scheduler_->last_heartbeat(),
                         base::TimeDelta::FromMilliseconds(new_delay));
 }
 
 TEST_F(HeartbeatSchedulerTest, DisableHeartbeats) {
   // Makes sure that we can disable heartbeats on the fly.
   EXPECT_CALL(gcm_driver_, RegisterImpl(kHeartbeatGCMAppID, _));
   settings_helper_.SetBoolean(chromeos::kHeartbeatEnabled, true);
   gcm::OutgoingMessage message;
   EXPECT_CALL(gcm_driver_, SendImpl(kHeartbeatGCMAppID, _, _))
       .WillOnce(SaveArg<2>(&message));
   gcm_driver_.CompleteRegistration(
       kHeartbeatGCMAppID, gcm::GCMClient::SUCCESS);
   // Should have a heartbeat task posted.
   EXPECT_EQ(1U, task_runner_->GetPendingTasks().size());
   task_runner_->RunPendingTasks();
 
   // Complete sending a message - we should queue up the next heartbeat.
   gcm_driver_.CompleteSend(
       kHeartbeatGCMAppID, message.id, gcm::GCMClient::SUCCESS);
 
   // Should have a new heartbeat task posted.
   ASSERT_EQ(1U, task_runner_->GetPendingTasks().size());
   CheckPendingTaskDelay(
-      scheduler_.last_heartbeat(),
+      scheduler_->last_heartbeat(),
       base::TimeDelta::FromMilliseconds(
           policy::HeartbeatScheduler::kDefaultHeartbeatIntervalMs));
   testing::Mock::VerifyAndClearExpectations(&gcm_driver_);
 
   // Now disable heartbeats. Should get no more heartbeats sent.
   settings_helper_.SetBoolean(chromeos::kHeartbeatEnabled, false);
   task_runner_->RunPendingTasks();
   EXPECT_TRUE(task_runner_->GetPendingTasks().empty());
 }
 
@@ skipping 13 matching lines @@
             message.time_to_live);
 
   // Check the values in the message payload.
   EXPECT_EQ("hb", message.data["type"]);
   int64 timestamp;
   EXPECT_TRUE(base::StringToInt64(message.data["timestamp"], &timestamp));
   EXPECT_EQ(kFakeEnrollmentDomain, message.data["domain_name"]);
   EXPECT_EQ(kFakeDeviceId, message.data["device_id"]);
 }
 
+TEST_F(HeartbeatSchedulerTest, SendGcmIdMappingUpdate) {
+  // Verifies that GCM id mapping update was sent after GCM registration.
+  policy::MockCloudPolicyClient cloud_policy_client;
+  cloud_policy_client.SetDMToken(kDMToken);
+  policy::CloudPolicyClient::StatusCallback callback;
+  EXPECT_CALL(cloud_policy_client, UpdateGcmIdMapping(kRegistrationId, _))
+      .WillOnce(SaveArg<1>(&callback));
+
+  InitializeScheduler(&cloud_policy_client);
+
+  // Enable heartbests.
+  EXPECT_CALL(gcm_driver_, RegisterImpl(kHeartbeatGCMAppID, _));
+  EXPECT_CALL(gcm_driver_, SendImpl(kHeartbeatGCMAppID, _, _));
+  settings_helper_.SetBoolean(chromeos::kHeartbeatEnabled, true);
+  gcm_driver_.CompleteRegistration(kHeartbeatGCMAppID, gcm::GCMClient::SUCCESS);
+  task_runner_->RunPendingTasks();
+
+  // Verifies that CloudPolicyClient got the update request, with a valid
+  // callback.
+  testing::Mock::VerifyAndClearExpectations(&cloud_policy_client);
+  EXPECT_FALSE(callback.is_null());
+  callback.Run(true);
+}
+
 }  // namespace