Implement cloud policy invalidations using the invalidation service framework.

chrome/browser/policy/cloud/cloud_policy_manager_unittest.cc

 // Copyright (c) 2012 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/policy/cloud/cloud_policy_manager.h"
 
 #include "base/basictypes.h"
 #include "base/callback.h"
 #include "base/compiler_specific.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop/message_loop.h"
+#include "chrome/browser/invalidation/fake_invalidation_service.h"
 #include "chrome/browser/policy/cloud/cloud_policy_constants.h"
 #include "chrome/browser/policy/cloud/mock_cloud_policy_client.h"
 #include "chrome/browser/policy/cloud/mock_cloud_policy_store.h"
 #include "chrome/browser/policy/cloud/policy_builder.h"
 #include "chrome/browser/policy/configuration_policy_provider_test.h"
 #include "chrome/browser/policy/external_data_fetcher.h"
 #include "chrome/browser/policy/mock_configuration_policy_provider.h"
+#include "sync/notifier/invalidation_util.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::Mock;
 using testing::_;
 
 namespace em = enterprise_management;
 
 namespace policy {
 namespace {
@@ skipping 107 matching lines @@
                                dm_protocol::kChromeUserPolicyType,
                                std::string()),
                            store) {}
   virtual ~TestCloudPolicyManager() {}
 
   // Publish the protected members for testing.
   using CloudPolicyManager::client;
   using CloudPolicyManager::store;
   using CloudPolicyManager::service;
   using CloudPolicyManager::CheckAndPublishPolicy;
+  using CloudPolicyManager::CreateInvalidator;
+  using CloudPolicyManager::UnregisterInvalidator;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(TestCloudPolicyManager);
 };
 
 MATCHER_P(ProtoMatches, proto, "") {
   return arg.SerializePartialAsString() == proto.SerializePartialAsString();
 }
 
 class CloudPolicyManagerTest : public testing::Test {
@@ skipping 174 matching lines @@
 TEST_F(CloudPolicyManagerTest, SignalOnError) {
   // Simulate a failed load and verify that it triggers OnUpdatePolicy().
   store_.policy_.reset(new em::PolicyData(policy_.policy_data()));
   EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get()));
   store_.NotifyStoreError();
   Mock::VerifyAndClearExpectations(&observer_);
 
   EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME));
 }
 
+TEST_F(CloudPolicyManagerTest, CreateAndUnregisterInvalidator) {
+  // Add the invalidation registration info to the policy data.
+  em::PolicyData* policy_data = new em::PolicyData(policy_.policy_data());
+  policy_data->set_invalidation_source(12345);
+  policy_data->set_invalidation_name("12345");
+  store_.policy_.reset(policy_data);
+
+  // Expect registration to occur when the invalidator is created.
+  invalidation::FakeInvalidationService invalidation_service;
+  MockCloudPolicyClient* client = new MockCloudPolicyClient();
+  EXPECT_CALL(*client, SetupRegistration(_, _));
+  manager_->core()->Connect(scoped_ptr<CloudPolicyClient>(client));
+  manager_->core()->StartRefreshScheduler();
+  manager_->CreateInvalidator(&invalidation_service, true /* ignore_switch */);
+  syncer::ObjectIdSet object_ids =
+      invalidation_service.invalidator_registrar().GetAllRegisteredIds();
+  ASSERT_EQ(1U, object_ids.size());
+  EXPECT_EQ(12345, object_ids.begin()->source());
+  EXPECT_EQ("12345", object_ids.begin()->name());
+
+  // Expect unregistration to occur.
+  manager_->UnregisterInvalidator();
+  object_ids =
+      invalidation_service.invalidator_registrar().GetAllRegisteredIds();
+  EXPECT_TRUE(object_ids.empty());
+}
+
 }  // namespace
 }  // namespace policy