src/chrome/browser/chromeos/login/supervised/supervised_user_test_base.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/chromeos/login/supervised/supervised_user_test_base.h"
   6
   7 #include <string>
   8
   9 #include "base/compiler_specific.h"
  10 #include "base/run_loop.h"
  11 #include "base/strings/string_number_conversions.h"
  12 #include "base/strings/stringprintf.h"
  13 #include "base/strings/utf_string_conversions.h"
  14 #include "base/threading/sequenced_worker_pool.h"
  15 #include "chrome/browser/chrome_notification_types.h"
  16 #include "chrome/browser/chromeos/login/login_manager_test.h"
  17 #include "chrome/browser/chromeos/login/startup_utils.h"
  18 #include "chrome/browser/chromeos/login/supervised/supervised_user_authentication.h"
  19 #include "chrome/browser/chromeos/login/ui/login_display_host_impl.h"
  20 #include "chrome/browser/chromeos/login/ui/webui_login_view.h"
  21 #include "chrome/browser/chromeos/login/users/supervised_user_manager.h"
  22 #include "chrome/browser/chromeos/net/network_portal_detector_test_impl.h"
  23 #include "chrome/browser/chromeos/profiles/profile_helper.h"
  24 #include "chrome/browser/chromeos/settings/stub_cros_settings_provider.h"
  25 #include "chrome/browser/profiles/profile_impl.h"
  26 #include "chrome/browser/supervised_user/supervised_user_constants.h"
  27 #include "chrome/browser/supervised_user/supervised_user_registration_utility.h"
  28 #include "chrome/browser/supervised_user/supervised_user_registration_utility_stub.h"
  29 #include "chrome/browser/supervised_user/supervised_user_shared_settings_service.h"
  30 #include "chrome/browser/supervised_user/supervised_user_shared_settings_service_factory.h"
  31 #include "chrome/browser/supervised_user/supervised_user_sync_service.h"
  32 #include "chrome/browser/supervised_user/supervised_user_sync_service_factory.h"
  33 #include "chromeos/cryptohome/mock_async_method_caller.h"
  34 #include "chromeos/cryptohome/mock_homedir_methods.h"
  35 #include "chromeos/login/auth/key.h"
  36 #include "chromeos/login/auth/user_context.h"
  37 #include "content/public/browser/notification_service.h"
  38 #include "content/public/test/browser_test_utils.h"
  39 #include "content/public/test/test_utils.h"
  40 #include "sync/api/fake_sync_change_processor.h"
  41 #include "sync/api/sync_change.h"
  42 #include "sync/api/sync_error_factory_mock.h"
  43 #include "sync/internal_api/public/attachments/attachment_service_proxy_for_test.h"
  44 #include "sync/protocol/sync.pb.h"
  45
  46 using testing::_;
  47 using base::StringPrintf;
  48
  49 namespace chromeos {
  50
  51 namespace {
  52
  53 const char kCurrentPage[] = "$('supervised-user-creation').currentPage_";
  54
  55 const char kStubEthernetGuid[] = "eth0";
  56
  57 }
  58
  59 SupervisedUsersSyncTestAdapter::SupervisedUsersSyncTestAdapter(Profile* profile)
  60     : processor_(), next_sync_data_id_(0) {
  61   service_ = SupervisedUserSyncServiceFactory::GetForProfile(profile);
  62   processor_ = new syncer::FakeSyncChangeProcessor();
  63   service_->MergeDataAndStartSyncing(
  64       syncer::SUPERVISED_USERS,
  65       syncer::SyncDataList(),
  66       scoped_ptr<syncer::SyncChangeProcessor>(processor_),
  67       scoped_ptr<syncer::SyncErrorFactory>(new syncer::SyncErrorFactoryMock));
  68 }
  69
  70 scoped_ptr< ::sync_pb::ManagedUserSpecifics>
  71 SupervisedUsersSyncTestAdapter::GetFirstChange() {
  72   scoped_ptr< ::sync_pb::ManagedUserSpecifics> result(
  73       new ::sync_pb::ManagedUserSpecifics);
  74   CHECK(HasChanges())
  75       << "GetFirstChange() should only be callled if HasChanges() is true";
  76   const syncer::SyncData& data = processor_->changes().front().sync_data();
  77   EXPECT_EQ(syncer::SUPERVISED_USERS, data.GetDataType());
  78   result->CopyFrom(data.GetSpecifics().managed_user());
  79   return result.Pass();
  80 }
  81
  82 void SupervisedUsersSyncTestAdapter::AddChange(
  83     const ::sync_pb::ManagedUserSpecifics& proto,
  84     bool update) {
  85   sync_pb::EntitySpecifics specifics;
  86
  87   specifics.mutable_managed_user()->CopyFrom(proto);
  88
  89   syncer::SyncData change_data = syncer::SyncData::CreateRemoteData(
  90       ++next_sync_data_id_,
  91       specifics,
  92       base::Time(),
  93       syncer::AttachmentIdList(),
  94       syncer::AttachmentServiceProxyForTest::Create());
  95   syncer::SyncChange change(FROM_HERE,
  96                             update ? syncer::SyncChange::ACTION_UPDATE
  97                                    : syncer::SyncChange::ACTION_ADD,
  98                             change_data);
  99
 100   syncer::SyncChangeList change_list;
 101   change_list.push_back(change);
 102
 103   service_->ProcessSyncChanges(FROM_HERE, change_list);
 104 }
 105
 106 SupervisedUsersSharedSettingsSyncTestAdapter::
 107     SupervisedUsersSharedSettingsSyncTestAdapter(Profile* profile)
 108     : processor_(), next_sync_data_id_(0) {
 109   service_ =
 110       SupervisedUserSharedSettingsServiceFactory::GetForBrowserContext(profile);
 111   processor_ = new syncer::FakeSyncChangeProcessor();
 112   service_->MergeDataAndStartSyncing(
 113       syncer::SUPERVISED_USER_SHARED_SETTINGS,
 114       syncer::SyncDataList(),
 115       scoped_ptr<syncer::SyncChangeProcessor>(processor_),
 116       scoped_ptr<syncer::SyncErrorFactory>(new syncer::SyncErrorFactoryMock));
 117 }
 118
 119 scoped_ptr< ::sync_pb::ManagedUserSharedSettingSpecifics>
 120 SupervisedUsersSharedSettingsSyncTestAdapter::GetFirstChange() {
 121   scoped_ptr< ::sync_pb::ManagedUserSharedSettingSpecifics> result(
 122       new ::sync_pb::ManagedUserSharedSettingSpecifics);
 123   CHECK(HasChanges())
 124       << "GetFirstChange() should only be callled if HasChanges() is true";
 125   const syncer::SyncData& data = processor_->changes().front().sync_data();
 126   EXPECT_EQ(syncer::SUPERVISED_USER_SHARED_SETTINGS, data.GetDataType());
 127   result->CopyFrom(data.GetSpecifics().managed_user_shared_setting());
 128   return result.Pass();
 129 }
 130
 131 void SupervisedUsersSharedSettingsSyncTestAdapter::AddChange(
 132     const ::sync_pb::ManagedUserSharedSettingSpecifics& proto,
 133     bool update) {
 134   sync_pb::EntitySpecifics specifics;
 135
 136   specifics.mutable_managed_user_shared_setting()->CopyFrom(proto);
 137
 138   syncer::SyncData change_data = syncer::SyncData::CreateRemoteData(
 139       ++next_sync_data_id_,
 140       specifics,
 141       base::Time(),
 142       syncer::AttachmentIdList(),
 143       syncer::AttachmentServiceProxyForTest::Create());
 144   syncer::SyncChange change(FROM_HERE,
 145                             update ? syncer::SyncChange::ACTION_UPDATE
 146                                    : syncer::SyncChange::ACTION_ADD,
 147                             change_data);
 148
 149   syncer::SyncChangeList change_list;
 150   change_list.push_back(change);
 151
 152   service_->ProcessSyncChanges(FROM_HERE, change_list);
 153 }
 154
 155 void SupervisedUsersSharedSettingsSyncTestAdapter::AddChange(
 156     const std::string& mu_id,
 157     const std::string& key,
 158     const base::Value& value,
 159     bool acknowledged,
 160     bool update) {
 161   syncer::SyncData data =
 162       SupervisedUserSharedSettingsService::CreateSyncDataForSetting(
 163           mu_id, key, value, acknowledged);
 164   AddChange(data.GetSpecifics().managed_user_shared_setting(), update);
 165 }
 166
 167 SupervisedUserTestBase::SupervisedUserTestBase()
 168     : LoginManagerTest(true),
 169       mock_async_method_caller_(NULL),
 170       mock_homedir_methods_(NULL),
 171       network_portal_detector_(NULL),
 172       registration_utility_stub_(NULL) {
 173 }
 174
 175 SupervisedUserTestBase::~SupervisedUserTestBase() {
 176 }
 177
 178 void SupervisedUserTestBase::SetUpInProcessBrowserTestFixture() {
 179   LoginManagerTest::SetUpInProcessBrowserTestFixture();
 180   mock_async_method_caller_ = new cryptohome::MockAsyncMethodCaller;
 181   mock_async_method_caller_->SetUp(true, cryptohome::MOUNT_ERROR_NONE);
 182   cryptohome::AsyncMethodCaller::InitializeForTesting(
 183       mock_async_method_caller_);
 184
 185   mock_homedir_methods_ = new cryptohome::MockHomedirMethods;
 186   mock_homedir_methods_->SetUp(true, cryptohome::MOUNT_ERROR_NONE);
 187   cryptohome::HomedirMethods::InitializeForTesting(mock_homedir_methods_);
 188
 189   registration_utility_stub_ = new SupervisedUserRegistrationUtilityStub();
 190   scoped_utility_.reset(new ScopedTestingSupervisedUserRegistrationUtility(
 191       registration_utility_stub_));
 192
 193   // Setup network portal detector to return online state for both
 194   // ethernet and wifi networks. Ethernet is an active network by
 195   // default.
 196   network_portal_detector_ = new NetworkPortalDetectorTestImpl();
 197   NetworkPortalDetector::InitializeForTesting(network_portal_detector_);
 198   NetworkPortalDetector::CaptivePortalState online_state;
 199   online_state.status = NetworkPortalDetector::CAPTIVE_PORTAL_STATUS_ONLINE;
 200   online_state.response_code = 204;
 201   network_portal_detector_->SetDefaultNetworkForTesting(kStubEthernetGuid);
 202   network_portal_detector_->SetDetectionResultsForTesting(kStubEthernetGuid,
 203                                                           online_state);
 204 }
 205
 206 void SupervisedUserTestBase::TearDown() {
 207   cryptohome::AsyncMethodCaller::Shutdown();
 208   cryptohome::HomedirMethods::Shutdown();
 209   mock_homedir_methods_ = NULL;
 210   mock_async_method_caller_ = NULL;
 211   LoginManagerTest::TearDown();
 212 }
 213
 214 void SupervisedUserTestBase::TearDownInProcessBrowserTestFixture() {
 215   NetworkPortalDetector::Shutdown();
 216 }
 217
 218 void SupervisedUserTestBase::JSEval(const std::string& script) {
 219   EXPECT_TRUE(content::ExecuteScript(web_contents(), script)) << script;
 220 }
 221
 222 void SupervisedUserTestBase::JSExpectAsync(const std::string& function) {
 223   bool result;
 224   EXPECT_TRUE(content::ExecuteScriptAndExtractBool(
 225       web_contents(),
 226       StringPrintf(
 227           "(%s)(function() { window.domAutomationController.send(true); });",
 228           function.c_str()),
 229       &result)) << function;
 230   EXPECT_TRUE(result);
 231 }
 232
 233 void SupervisedUserTestBase::JSSetTextField(const std::string& element_selector,
 234                                          const std::string& value) {
 235   std::string function =
 236       StringPrintf("document.querySelector('%s').value = '%s'",
 237                    element_selector.c_str(),
 238                    value.c_str());
 239   JSEval(function);
 240 }
 241
 242 void SupervisedUserTestBase::PrepareUsers() {
 243   RegisterUser(kTestManager);
 244   RegisterUser(kTestOtherUser);
 245   chromeos::StartupUtils::MarkOobeCompleted();
 246 }
 247
 248 void SupervisedUserTestBase::StartFlowLoginAsManager() {
 249   // Navigate to supervised user creation screen.
 250   JSEval("chrome.send('showSupervisedUserCreationScreen')");
 251
 252   // Read intro and proceed.
 253   JSExpect(StringPrintf("%s == 'intro'", kCurrentPage));
 254
 255   JSEval("$('supervised-user-creation-start-button').click()");
 256
 257   // Check that both users appear as managers, and test-manager@gmail.com is
 258   // the first one.
 259   JSExpect(StringPrintf("%s == 'manager'", kCurrentPage));
 260
 261   std::string manager_pods =
 262       "document.querySelectorAll('#supervised-user-creation-managers-pane "
 263       ".manager-pod')";
 264   std::string selected_manager_pods =
 265       "document.querySelectorAll('#supervised-user-creation-managers-pane "
 266       ".manager-pod.focused')";
 267
 268   int managers_on_device = 2;
 269
 270   JSExpect(StringPrintf("%s.length == 1", selected_manager_pods.c_str()));
 271
 272   JSExpect(StringPrintf(
 273       "$('supervised-user-creation').managerList_.pods.length == %d",
 274       managers_on_device));
 275   JSExpect(StringPrintf(
 276       "%s.length == %d", manager_pods.c_str(), managers_on_device));
 277   JSExpect(StringPrintf("%s[%d].user.emailAddress == '%s'",
 278                         manager_pods.c_str(),
 279                         0,
 280                         kTestManager));
 281
 282   // Select the first user as manager, and enter password.
 283   JSExpect("$('supervised-user-creation-next-button').disabled");
 284   JSSetTextField("#supervised-user-creation .manager-pod.focused input",
 285                  kTestManagerPassword);
 286
 287   JSEval("$('supervised-user-creation').updateNextButtonForManager_()");
 288
 289   // Next button is now enabled.
 290   JSExpect("!$('supervised-user-creation-next-button').disabled");
 291   UserContext user_context(kTestManager);
 292   user_context.SetKey(Key(kTestManagerPassword));
 293   SetExpectedCredentials(user_context);
 294   content::WindowedNotificationObserver login_observer(
 295       chrome::NOTIFICATION_LOGIN_USER_PROFILE_PREPARED,
 296       content::NotificationService::AllSources());
 297
 298   // Log in as manager.
 299   JSEval("$('supervised-user-creation-next-button').click()");
 300   login_observer.Wait();
 301
 302   // OAuth token is valid.
 303   user_manager::UserManager::Get()->SaveUserOAuthStatus(
 304       kTestManager, user_manager::User::OAUTH2_TOKEN_STATUS_VALID);
 305   base::RunLoop().RunUntilIdle();
 306
 307   // Check the page have changed.
 308   JSExpect(StringPrintf("%s == 'username'", kCurrentPage));
 309 }
 310
 311 void SupervisedUserTestBase::FillNewUserData(const std::string& display_name) {
 312   JSExpect("$('supervised-user-creation-next-button').disabled");
 313   JSSetTextField("#supervised-user-creation-name", display_name);
 314   JSEval("$('supervised-user-creation').checkUserName_()");
 315
 316   base::RunLoop().RunUntilIdle();
 317
 318   JSSetTextField("#supervised-user-creation-password",
 319                  kTestSupervisedUserPassword);
 320   JSSetTextField("#supervised-user-creation-password-confirm",
 321                  kTestSupervisedUserPassword);
 322
 323   JSEval("$('supervised-user-creation').updateNextButtonForUser_()");
 324   JSExpect("!$('supervised-user-creation-next-button').disabled");
 325 }
 326
 327 void SupervisedUserTestBase::StartUserCreation(
 328     const std::string& button_id,
 329     const std::string& expected_display_name) {
 330   EXPECT_CALL(*mock_homedir_methods_, MountEx(_, _, _, _)).Times(1);
 331   EXPECT_CALL(*mock_homedir_methods_, AddKeyEx(_, _, _, _, _)).Times(1);
 332
 333   JSEval(std::string("$('").append(button_id).append("').click()"));
 334
 335   ::testing::Mock::VerifyAndClearExpectations(mock_homedir_methods_);
 336
 337   EXPECT_TRUE(registration_utility_stub_->register_was_called());
 338   EXPECT_EQ(registration_utility_stub_->display_name(),
 339             base::UTF8ToUTF16(expected_display_name));
 340
 341   registration_utility_stub_->RunSuccessCallback("token");
 342
 343   // Token writing moves control to BlockingPool and back.
 344   content::RunAllBlockingPoolTasksUntilIdle();
 345
 346   JSExpect(StringPrintf("%s == 'created'", kCurrentPage));
 347   JSEval("$('supervised-user-creation-gotit-button').click()");
 348 }
 349
 350 void SupervisedUserTestBase::SigninAsSupervisedUser(
 351     bool check_homedir_calls,
 352     int user_index,
 353     const std::string& expected_display_name) {
 354   if (check_homedir_calls)
 355     EXPECT_CALL(*mock_homedir_methods_, MountEx(_, _, _, _)).Times(1);
 356
 357   // Log in as supervised user, make sure that everything works.
 358   ASSERT_EQ(3UL, user_manager::UserManager::Get()->GetUsers().size());
 359
 360   // Created supervised user have to be first in a list.
 361   const user_manager::User* user =
 362       user_manager::UserManager::Get()->GetUsers().at(user_index);
 363   ASSERT_EQ(base::UTF8ToUTF16(expected_display_name), user->display_name());
 364   LoginUser(user->email());
 365   if (check_homedir_calls)
 366     ::testing::Mock::VerifyAndClearExpectations(mock_homedir_methods_);
 367   Profile* profile = ProfileHelper::Get()->GetProfileByUserUnsafe(user);
 368   shared_settings_adapter_.reset(
 369       new SupervisedUsersSharedSettingsSyncTestAdapter(profile));
 370
 371   // Check ChromeOS preference is initialized.
 372   EXPECT_TRUE(
 373       static_cast<ProfileImpl*>(profile)->chromeos_preferences_);
 374 }
 375
 376 void SupervisedUserTestBase::SigninAsManager(int user_index) {
 377   // Log in as supervised user, make sure that everything works.
 378   ASSERT_EQ(3UL, user_manager::UserManager::Get()->GetUsers().size());
 379
 380   // Created supervised user have to be first in a list.
 381   const user_manager::User* user =
 382       user_manager::UserManager::Get()->GetUsers().at(user_index);
 383   LoginUser(user->email());
 384   Profile* profile = ProfileHelper::Get()->GetProfileByUserUnsafe(user);
 385   shared_settings_adapter_.reset(
 386       new SupervisedUsersSharedSettingsSyncTestAdapter(profile));
 387   supervised_users_adapter_.reset(new SupervisedUsersSyncTestAdapter(profile));
 388 }
 389
 390 void SupervisedUserTestBase::RemoveSupervisedUser(
 391     size_t original_user_count,
 392     int user_index,
 393     const std::string& expected_display_name) {
 394   // Remove supervised user.
 395   ASSERT_EQ(original_user_count,
 396             user_manager::UserManager::Get()->GetUsers().size());
 397
 398   // Created supervised user have to be first in a list.
 399   const user_manager::User* user =
 400       user_manager::UserManager::Get()->GetUsers().at(user_index);
 401   ASSERT_EQ(base::UTF8ToUTF16(expected_display_name), user->display_name());
 402
 403   // Open pod menu.
 404   JSExpect(
 405       StringPrintf("!$('pod-row').pods[%d].isActionBoxMenuActive", user_index));
 406   JSEval(StringPrintf(
 407       "$('pod-row').pods[%d].querySelector('.action-box-button').click()",
 408       user_index));
 409   JSExpect(
 410       StringPrintf("$('pod-row').pods[%d].isActionBoxMenuActive", user_index));
 411
 412   // Select "Remove user" element.
 413   JSExpect(StringPrintf(
 414       "$('pod-row').pods[%d].actionBoxRemoveUserWarningElement.hidden",
 415       user_index));
 416   JSEval(StringPrintf(
 417       "$('pod-row').pods[%d].querySelector('.action-box-menu-remove').click()",
 418       user_index));
 419   JSExpect(StringPrintf(
 420       "!$('pod-row').pods[%d].actionBoxRemoveUserWarningElement.hidden",
 421       user_index));
 422
 423   EXPECT_CALL(*mock_async_method_caller_, AsyncRemove(_, _)).Times(1);
 424
 425   // Confirm deletion.
 426   JSEval(StringPrintf(
 427       "$('pod-row').pods[%d].querySelector('.remove-warning-button').click()",
 428       user_index));
 429
 430   // Make sure there is no supervised user in list.
 431   ASSERT_EQ(original_user_count - 1,
 432             user_manager::UserManager::Get()->GetUsers().size());
 433 }
 434
 435 }  // namespace chromeos