Tests for supervised users password sync:

chrome/browser/chromeos/login/managed/managed_user_test_base.cc

Index: chrome/browser/chromeos/login/managed/managed_user_test_base.cc
diff --git a/chrome/browser/chromeos/login/managed/managed_user_test_base.cc b/chrome/browser/chromeos/login/managed/managed_user_test_base.cc
new file mode 100644
index 0000000000000000000000000000000000000000..276df838c429215bffd0079f95775ddaa050ac47
--- /dev/null
+++ b/chrome/browser/chromeos/login/managed/managed_user_test_base.cc
@@ -0,0 +1,428 @@
+// Copyright 2013 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/login/managed/managed_user_test_base.h"
+
+#include <string>
+
+#include "base/compiler_specific.h"
+#include "base/run_loop.h"
+#include "base/strings/string_number_conversions.h"
+#include "base/strings/stringprintf.h"
+#include "base/strings/utf_string_conversions.h"
+#include "base/threading/sequenced_worker_pool.h"
+#include "chrome/browser/chrome_notification_types.h"
+#include "chrome/browser/chromeos/login/login_display_host_impl.h"
+#include "chrome/browser/chromeos/login/login_manager_test.h"
+#include "chrome/browser/chromeos/login/managed/supervised_user_authentication.h"
+#include "chrome/browser/chromeos/login/startup_utils.h"
+#include "chrome/browser/chromeos/login/supervised_user_manager.h"
+#include "chrome/browser/chromeos/login/webui_login_view.h"
+#include "chrome/browser/chromeos/net/network_portal_detector_test_impl.h"
+#include "chrome/browser/chromeos/settings/stub_cros_settings_provider.h"
+#include "chrome/browser/managed_mode/managed_user_constants.h"
+#include "chrome/browser/managed_mode/managed_user_registration_utility.h"
+#include "chrome/browser/managed_mode/managed_user_registration_utility_stub.h"
+#include "chrome/browser/managed_mode/managed_user_shared_settings_service.h"
+#include "chrome/browser/managed_mode/managed_user_shared_settings_service_factory.h"
+#include "chrome/browser/managed_mode/managed_user_sync_service.h"
+#include "chrome/browser/managed_mode/managed_user_sync_service_factory.h"
+#include "chromeos/cryptohome/mock_async_method_caller.h"
+#include "chromeos/cryptohome/mock_homedir_methods.h"
+#include "content/public/browser/notification_service.h"
+#include "content/public/test/browser_test_utils.h"
+#include "content/public/test/test_utils.h"
+#include "sync/api/attachments/attachment_service_proxy_for_test.h"
+#include "sync/api/fake_sync_change_processor.h"
+#include "sync/api/sync_change.h"
+#include "sync/api/sync_error_factory_mock.h"
+#include "sync/protocol/sync.pb.h"
+
+using testing::_;
+using base::StringPrintf;
+
+namespace chromeos {
+
+namespace testing {
+
+namespace {
+
+const char kCurrentPage[] = "$('managed-user-creation').currentPage_";
+}
+
+ManagedUsersSyncTestAdapter::ManagedUsersSyncTestAdapter(Profile* profile)
+    : processor_(), next_sync_data_id_(0) {
+  service_ = ManagedUserSyncServiceFactory::GetForProfile(profile);
+  processor_ = new syncer::FakeSyncChangeProcessor();
+  service_->MergeDataAndStartSyncing(
+      syncer::MANAGED_USERS,
+      syncer::SyncDataList(),
+      scoped_ptr<syncer::SyncChangeProcessor>(processor_),
+      scoped_ptr<syncer::SyncErrorFactory>(new syncer::SyncErrorFactoryMock));
+}
+
+scoped_ptr< ::sync_pb::ManagedUserSpecifics>
+ManagedUsersSyncTestAdapter::GetFirstChange() {
+  scoped_ptr< ::sync_pb::ManagedUserSpecifics> result(
+      new ::sync_pb::ManagedUserSpecifics);
+  CHECK(HasChanges())
+      << "GetFirstChange() should only be callled if HasChanges() is true";
+  const syncer::SyncData& data = processor_->changes().front().sync_data();
+  EXPECT_EQ(syncer::MANAGED_USERS, data.GetDataType());
+  result->CopyFrom(data.GetSpecifics().managed_user());
+  return result.Pass();
+}
+
+void ManagedUsersSyncTestAdapter::AddChange(
+    const ::sync_pb::ManagedUserSpecifics& proto,
+    bool update) {
+  sync_pb::EntitySpecifics specifics;
+
+  specifics.mutable_managed_user()->CopyFrom(proto);
+
+  syncer::SyncData change_data = syncer::SyncData::CreateRemoteData(
+      ++next_sync_data_id_,
+      specifics,
+      base::Time(),
+      syncer::AttachmentIdList(),
+      syncer::AttachmentServiceProxyForTest::Create());
+  syncer::SyncChange change(FROM_HERE,
+                            update ? syncer::SyncChange::ACTION_UPDATE
+                                   : syncer::SyncChange::ACTION_ADD,
+                            change_data);
+
+  syncer::SyncChangeList change_list;
+  change_list.push_back(change);
+
+  service_->ProcessSyncChanges(FROM_HERE, change_list);
+}
+
+ManagedUsersSharedSettingsSyncTestAdapter::
+    ManagedUsersSharedSettingsSyncTestAdapter(Profile* profile)
+    : processor_(), next_sync_data_id_(0) {
+  service_ =
+      ManagedUserSharedSettingsServiceFactory::GetForBrowserContext(profile);
+  processor_ = new syncer::FakeSyncChangeProcessor();
+  service_->MergeDataAndStartSyncing(
+      syncer::MANAGED_USER_SHARED_SETTINGS,
+      syncer::SyncDataList(),
+      scoped_ptr<syncer::SyncChangeProcessor>(processor_),
+      scoped_ptr<syncer::SyncErrorFactory>(new syncer::SyncErrorFactoryMock));
+}
+
+scoped_ptr< ::sync_pb::ManagedUserSharedSettingSpecifics>
+ManagedUsersSharedSettingsSyncTestAdapter::GetFirstChange() {
+  scoped_ptr< ::sync_pb::ManagedUserSharedSettingSpecifics> result(
+      new ::sync_pb::ManagedUserSharedSettingSpecifics);
+  CHECK(HasChanges())
+      << "GetFirstChange() should only be callled if HasChanges() is true";
+  const syncer::SyncData& data = processor_->changes().front().sync_data();
+  EXPECT_EQ(syncer::MANAGED_USER_SHARED_SETTINGS, data.GetDataType());
+  result->CopyFrom(data.GetSpecifics().managed_user_shared_setting());
+  return result.Pass();
+}
+
+void ManagedUsersSharedSettingsSyncTestAdapter::AddChange(
+    const ::sync_pb::ManagedUserSharedSettingSpecifics& proto,
+    bool update) {
+  sync_pb::EntitySpecifics specifics;
+
+  specifics.mutable_managed_user_shared_setting()->CopyFrom(proto);
+
+  syncer::SyncData change_data = syncer::SyncData::CreateRemoteData(
+      ++next_sync_data_id_,
+      specifics,
+      base::Time(),
+      syncer::AttachmentIdList(),
+      syncer::AttachmentServiceProxyForTest::Create());
+  syncer::SyncChange change(FROM_HERE,
+                            update ? syncer::SyncChange::ACTION_UPDATE
+                                   : syncer::SyncChange::ACTION_ADD,
+                            change_data);
+
+  syncer::SyncChangeList change_list;
+  change_list.push_back(change);
+
+  service_->ProcessSyncChanges(FROM_HERE, change_list);
+}
+
+void ManagedUsersSharedSettingsSyncTestAdapter::AddChange(
+    const std::string& mu_id,
+    const std::string& key,
+    const base::Value& value,
+    bool acknowledged,
+    bool update) {
+  syncer::SyncData data =
+      ManagedUserSharedSettingsService::CreateSyncDataForSetting(
+          mu_id, key, value, acknowledged);
+  AddChange(data.GetSpecifics().managed_user_shared_setting(), update);
+}
+
+ManagedUserTestBase::ManagedUserTestBase()
+    : LoginManagerTest(true),
+      mock_async_method_caller_(NULL),
+      mock_homedir_methods_(NULL),
+      network_portal_detector_(NULL),
+      registration_utility_stub_(NULL) {
+}
+
+ManagedUserTestBase::~ManagedUserTestBase() {
+}
+
+void ManagedUserTestBase::SetUpInProcessBrowserTestFixture() {
+  LoginManagerTest::SetUpInProcessBrowserTestFixture();
+  mock_async_method_caller_ = new cryptohome::MockAsyncMethodCaller;
+  mock_async_method_caller_->SetUp(true, cryptohome::MOUNT_ERROR_NONE);
+  cryptohome::AsyncMethodCaller::InitializeForTesting(
+      mock_async_method_caller_);
+
+  mock_homedir_methods_ = new cryptohome::MockHomedirMethods;
+  mock_homedir_methods_->SetUp(true, cryptohome::MOUNT_ERROR_NONE);
+  cryptohome::HomedirMethods::InitializeForTesting(mock_homedir_methods_);
+
+  registration_utility_stub_ = new ManagedUserRegistrationUtilityStub();
+  scoped_utility_.reset(new ScopedTestingManagedUserRegistrationUtility(
+      registration_utility_stub_));
+
+  // Setup network portal detector to return online state for both
+  // ethernet and wifi networks. Ethernet is an active network by
+  // default.
+  network_portal_detector_ = new NetworkPortalDetectorTestImpl();
+  NetworkPortalDetector::InitializeForTesting(network_portal_detector_);
+  NetworkPortalDetector::CaptivePortalState online_state;
+  online_state.status = NetworkPortalDetector::CAPTIVE_PORTAL_STATUS_ONLINE;
+  online_state.response_code = 204;
+  network_portal_detector_->SetDefaultNetworkPathForTesting(
+      kStubEthernetServicePath);
+  network_portal_detector_->SetDetectionResultsForTesting(
+      kStubEthernetServicePath, online_state);
+}
+
+void ManagedUserTestBase::CleanUpOnMainThread() {
+  LoginManagerTest::CleanUpOnMainThread();
+}
+
+void ManagedUserTestBase::TearDown() {
+  cryptohome::AsyncMethodCaller::Shutdown();
+  cryptohome::HomedirMethods::Shutdown();
+  mock_homedir_methods_ = NULL;
+  mock_async_method_caller_ = NULL;
+  LoginManagerTest::TearDown();
+}
+
+void ManagedUserTestBase::TearDownInProcessBrowserTestFixture() {
+  NetworkPortalDetector::Shutdown();
+}
+
+void ManagedUserTestBase::JSEval(const std::string& script) {
+  EXPECT_TRUE(content::ExecuteScript(web_contents(), script));
+}
+
+void ManagedUserTestBase::JSExpectAsync(const std::string& function) {
+  bool result;
+  EXPECT_TRUE(content::ExecuteScriptAndExtractBool(
+      web_contents(),
+      StringPrintf(
+          "(%s)(function() { window.domAutomationController.send(true); });",
+          function.c_str()),
+      &result));
+  EXPECT_TRUE(result);
+}
+
+void ManagedUserTestBase::JSSetTextField(const std::string& element_selector,
+                                         const std::string& value) {
+  std::string function =
+      StringPrintf("document.querySelector('%s').value = '%s'",
+                   element_selector.c_str(),
+                   value.c_str());
+  JSEval(function);
+}
+
+void ManagedUserTestBase::PrepareUsers() {
+  RegisterUser(kTestManager);
+  RegisterUser(kTestOtherUser);
+  chromeos::StartupUtils::MarkOobeCompleted();
+}
+
+void ManagedUserTestBase::StartFlowLoginAsManager() {
+  // Navigate to supervised user creation screen.
+  JSEval("chrome.send('showLocallyManagedUserCreationScreen')");
+
+  // Read intro and proceed.
+  JSExpect(StringPrintf("%s == 'intro'", kCurrentPage));
+
+  JSEval("$('managed-user-creation-start-button').click()");
+
+  // Check that both users appear as managers, and test-manager@gmail.com is
+  // the first one.
+  JSExpect(StringPrintf("%s == 'manager'", kCurrentPage));
+
+  std::string manager_pods =
+      "document.querySelectorAll('#managed-user-creation-managers-pane "
+      ".manager-pod')";
+  std::string selected_manager_pods =
+      "document.querySelectorAll('#managed-user-creation-managers-pane "
+      ".manager-pod.focused')";
+
+  int managers_on_device = 2;
+
+  JSExpect(StringPrintf("%s.length == 1", selected_manager_pods.c_str()));
+
+  JSExpect(
+      StringPrintf("$('managed-user-creation').managerList_.pods.length == %d",
+                   managers_on_device));
+  JSExpect(StringPrintf(
+      "%s.length == %d", manager_pods.c_str(), managers_on_device));
+  JSExpect(StringPrintf("%s[%d].user.emailAddress == '%s'",
+                        manager_pods.c_str(),
+                        0,
+                        kTestManager));
+
+  // Select the first user as manager, and enter password.
+  JSExpect("$('managed-user-creation-next-button').disabled");
+  JSSetTextField("#managed-user-creation .manager-pod.focused input",
+                 kTestManagerPassword);
+
+  JSEval("$('managed-user-creation').updateNextButtonForManager_()");
+
+  // Next button is now enabled.
+  JSExpect("!$('managed-user-creation-next-button').disabled");
+  SetExpectedCredentials(kTestManager, kTestManagerPassword);
+  content::WindowedNotificationObserver login_observer(
+      chrome::NOTIFICATION_LOGIN_USER_PROFILE_PREPARED,
+      content::NotificationService::AllSources());
+
+  // Log in as manager.
+  JSEval("$('managed-user-creation-next-button').click()");
+  login_observer.Wait();
+
+  // OAuth token is valid.
+  UserManager::Get()->SaveUserOAuthStatus(kTestManager,
+                                          User::OAUTH2_TOKEN_STATUS_VALID);
+  base::RunLoop().RunUntilIdle();
+
+  // Check the page have changed.
+  JSExpect(StringPrintf("%s == 'username'", kCurrentPage));
+}
+
+void ManagedUserTestBase::FillNewUserData(const std::string& display_name) {
+  JSExpect("$('managed-user-creation-next-button').disabled");
+  JSSetTextField("#managed-user-creation-name", display_name);
+  JSEval("$('managed-user-creation').checkUserName_()");
+
+  base::RunLoop().RunUntilIdle();
+
+  JSSetTextField("#managed-user-creation-password",
+                 kTestSupervisedUserPassword);
+  JSSetTextField("#managed-user-creation-password-confirm",
+                 kTestSupervisedUserPassword);
+
+  JSEval("$('managed-user-creation').updateNextButtonForUser_()");
+  JSExpect("!$('managed-user-creation-next-button').disabled");
+}
+
+void ManagedUserTestBase::StartUserCreation(
+    const std::string& button_id,
+    const std::string& expected_display_name) {
+  EXPECT_CALL(*mock_homedir_methods_, MountEx(_, _, _, _)).Times(1);
+  EXPECT_CALL(*mock_homedir_methods_, AddKeyEx(_, _, _, _, _)).Times(1);
+
+  JSEval(std::string("$('").append(button_id).append("').click()"));
+
+  ::testing::Mock::VerifyAndClearExpectations(mock_homedir_methods_);
+
+  EXPECT_TRUE(registration_utility_stub_->register_was_called());
+  EXPECT_EQ(registration_utility_stub_->display_name(),
+            base::UTF8ToUTF16(expected_display_name));
+
+  registration_utility_stub_->RunSuccessCallback("token");
+
+  // Token writing moves control to BlockingPool and back.
+  base::RunLoop().RunUntilIdle();
+  content::BrowserThread::GetBlockingPool()->FlushForTesting();
+  base::RunLoop().RunUntilIdle();
+
+  JSExpect(StringPrintf("%s == 'created'", kCurrentPage));
+  JSEval("$('managed-user-creation-gotit-button').click()");
+}
+
+void ManagedUserTestBase::SigninAsSupervisedUser(
+    bool check_homedir_calls,
+    int user_index,
+    const std::string& expected_display_name) {
+  if (check_homedir_calls)
+    EXPECT_CALL(*mock_homedir_methods_, MountEx(_, _, _, _)).Times(1);
+
+  // Log in as supervised user, make sure that everything works.
+  ASSERT_EQ(3UL, UserManager::Get()->GetUsers().size());
+
+  // Created supervised user have to be first in a list.
+  const User* user = UserManager::Get()->GetUsers().at(user_index);
+  ASSERT_EQ(base::UTF8ToUTF16(expected_display_name), user->display_name());
+  LoginUser(user->email());
+  if (check_homedir_calls)
+    ::testing::Mock::VerifyAndClearExpectations(mock_homedir_methods_);
+  Profile* profile = UserManager::Get()->GetProfileByUser(user);
+  shared_settings_adapter_.reset(
+      new ManagedUsersSharedSettingsSyncTestAdapter(profile));
+}
+
+void ManagedUserTestBase::SigninAsManager(int user_index) {
+  // Log in as supervised user, make sure that everything works.
+  ASSERT_EQ(3UL, UserManager::Get()->GetUsers().size());
+
+  // Created supervised user have to be first in a list.
+  const User* user = UserManager::Get()->GetUsers().at(user_index);
+  LoginUser(user->email());
+  Profile* profile = UserManager::Get()->GetProfileByUser(user);
+  shared_settings_adapter_.reset(
+      new ManagedUsersSharedSettingsSyncTestAdapter(profile));
+  managed_users_adapter_.reset(new ManagedUsersSyncTestAdapter(profile));
+}
+
+void ManagedUserTestBase::RemoveSupervisedUser(
+    unsigned long original_user_count,
+    int user_index,
+    const std::string& expected_display_name) {
+  // Remove supervised user.
+  ASSERT_EQ(original_user_count, UserManager::Get()->GetUsers().size());
+
+  // Created supervised user have to be first in a list.
+  const User* user = UserManager::Get()->GetUsers().at(user_index);
+  ASSERT_EQ(base::UTF8ToUTF16(expected_display_name), user->display_name());
+
+  // Open pod menu.
+  JSExpect(
+      StringPrintf("!$('pod-row').pods[%d].isActionBoxMenuActive", user_index));
+  JSEval(StringPrintf(
+      "$('pod-row').pods[%d].querySelector('.action-box-button').click()",
+      user_index));
+  JSExpect(
+      StringPrintf("$('pod-row').pods[%d].isActionBoxMenuActive", user_index));
+
+  // Select "Remove user" element.
+  JSExpect(StringPrintf(
+      "$('pod-row').pods[%d].actionBoxRemoveUserWarningElement.hidden",
+      user_index));
+  JSEval(StringPrintf(
+      "$('pod-row').pods[%d].querySelector('.action-box-menu-remove').click()",
+      user_index));
+  JSExpect(StringPrintf(
+      "!$('pod-row').pods[%d].actionBoxRemoveUserWarningElement.hidden",
+      user_index));
+
+  EXPECT_CALL(*mock_async_method_caller_, AsyncRemove(_, _)).Times(1);
+
+  // Confirm deletion.
+  JSEval(StringPrintf(
+      "$('pod-row').pods[%d].querySelector('.remove-warning-button').click()",
+      user_index));
+
+  // Make sure there is no supervised user in list.
+  ASSERT_EQ(original_user_count - 1, UserManager::Get()->GetUsers().size());
+}
+
+}  // namespace testing
+
+}  // namespace chromeos