Split UserCloudPolicyManager implementation.

chrome/browser/policy/cloud_policy_browsertest.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 "base/base_paths.h"
 #include "base/bind_helpers.h"
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/path_service.h"
 #include "base/run_loop.h"
 #include "base/stringprintf.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/policy/browser_policy_connector.h"
 #include "chrome/browser/policy/cloud_policy_client.h"
 #include "chrome/browser/policy/cloud_policy_data_store.h"
 #include "chrome/browser/policy/policy_map.h"
 #include "chrome/browser/policy/policy_service.h"
 #include "chrome/browser/policy/proto/chrome_settings.pb.h"
 #include "chrome/browser/policy/proto/cloud_policy.pb.h"
-#include "chrome/browser/policy/user_cloud_policy_manager.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_source.h"
 #include "content/public/test/test_utils.h"
 #include "googleurl/src/gurl.h"
 #include "net/test/test_server.h"
 #include "policy/policy_constants.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 #if defined(OS_CHROMEOS)
 #include "chrome/browser/chromeos/login/user_manager.h"
+#include "chrome/browser/policy/user_cloud_policy_manager_chromeos.h"
 #else
+#include "chrome/browser/policy/user_cloud_policy_manager.h"
+#include "chrome/browser/policy/user_cloud_policy_manager_factory.h"
 #include "chrome/browser/signin/signin_manager.h"
 #include "chrome/browser/signin/signin_manager_factory.h"
 #endif
 
 using testing::InvokeWithoutArgs;
 using testing::Mock;
 using testing::_;
 
 namespace em = enterprise_management;
 
@@ skipping 67 matching lines @@
 }
 #endif
 
 void SetUpNewStackBeforeCreatingBrowser() {
   CommandLine* command_line = CommandLine::ForCurrentProcess();
   command_line->AppendSwitch(switches::kEnableCloudPolicyService);
   command_line->AppendSwitch(switches::kLoadCloudPolicyOnSignin);
 }
 
 void SetUpNewStackAfterCreatingBrowser(Browser* browser) {
-#if !defined(OS_CHROMEOS)
+  BrowserPolicyConnector* connector =
+      g_browser_process->browser_policy_connector();
+  connector->ScheduleServiceInitialization(0);
+
+#if defined(OS_CHROMEOS)
+  connector->InitializeUserPolicy(GetTestUser(), true);
+  UserCloudPolicyManagerChromeOS* policy_manager =
+      connector->GetUserCloudPolicyManager();
+  ASSERT_TRUE(policy_manager);
+#else
   // Mock a signed-in user. This is used by the UserCloudPolicyStore to pass the
   // username to the UserCloudPolicyValidator.
   SigninManager* signin_manager =
       SigninManagerFactory::GetForProfile(browser->profile());
   ASSERT_TRUE(signin_manager);
   signin_manager->SetAuthenticatedUsername(GetTestUser());
-#endif
-
-  BrowserPolicyConnector* connector =
-      g_browser_process->browser_policy_connector();
-  connector->ScheduleServiceInitialization(0);
 
   UserCloudPolicyManager* policy_manager =
-      browser->profile()->GetUserCloudPolicyManager();
+      UserCloudPolicyManagerFactory::GetForProfile(browser->profile());
   ASSERT_TRUE(policy_manager);
   policy_manager->Initialize(g_browser_process->local_state(),
-                             connector->device_management_service(),
-                             policy::USER_AFFILIATION_MANAGED);
+                             connector->device_management_service());
+#endif  // defined(OS_CHROMEOS)
 
   ASSERT_TRUE(policy_manager->cloud_policy_client());
   base::RunLoop run_loop;
   MockCloudPolicyClientObserver observer;
   EXPECT_CALL(observer, OnRegistrationStateChanged(_)).WillOnce(
       InvokeWithoutArgs(&run_loop, &base::RunLoop::Quit));
   policy_manager->cloud_policy_client()->AddObserver(&observer);
 
   // Give a bogus OAuth token to the |policy_manager|. This should make its
   // CloudPolicyClient fetch the DMToken.
@@ skipping 153 matching lines @@
 
   // They should now serialize to the same bytes.
   std::string chrome_settings_serialized;
   std::string cloud_policy_serialized;
   ASSERT_TRUE(chrome_settings.SerializeToString(&chrome_settings_serialized));
   ASSERT_TRUE(cloud_policy.SerializeToString(&cloud_policy_serialized));
   EXPECT_EQ(chrome_settings_serialized, cloud_policy_serialized);
 }
 
 }  // namespace policy