Chromium Code Reviews| 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..b979e79ea5207d66e19c4d8cdc005663a9ad5605 |
| --- /dev/null |
| +++ b/chrome/browser/chromeos/login/managed/managed_user_test_base.cc |
| @@ -0,0 +1,417 @@ |
| +// 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 { |
| + |
|
Nikita (slow)
2014/04/23 09:50:17
nit: Drop empty line (or insert empty line after c
Denis Kuznetsov (DE-MUC)
2014/04/24 17:44:09
Done.
|
| +std::string current_page = "$('managed-user-creation').currentPage_"; |
|
Nikita (slow)
2014/04/23 09:50:17
const char kCurrentPage[] = "...";
Denis Kuznetsov (DE-MUC)
2014/04/24 17:44:09
Done.
|
| +} |
| + |
| +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()); |
|
Nikita (slow)
2014/04/23 09:50:17
Why not ASSERT or EXPECT? We should be using them
Denis Kuznetsov (DE-MUC)
2014/04/24 17:44:09
These are sanity checks, not things we try to vali
Nikita (slow)
2014/04/25 06:43:59
Still this is just a test and should not ever cras
Denis Kuznetsov (DE-MUC)
2014/04/25 12:52:04
ASSERT can be used only inside tests (that macro e
|
| + const syncer::SyncData& data = processor_->changes().front().sync_data(); |
| + CHECK_EQ(syncer::MANAGED_USERS, data.GetDataType()); |
| + result->CopyFrom(data.GetSpecifics().managed_user()); |
| + return result.Pass(); |
| +} |
| + |
| +void ManagedUsersSyncTestAdapter::AddChange( |
| + const ::sync_pb::ManagedUserSpecifics& proto) { |
| + 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, syncer::SyncChange::ACTION_UPDATE, 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()); |
| + const syncer::SyncData& data = processor_->changes().front().sync_data(); |
| + CHECK_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) { |
| + 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, syncer::SyncChange::ACTION_UPDATE, 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) { |
| + syncer::SyncData data = |
| + ManagedUserSharedSettingsService::CreateSyncDataForSetting( |
| + mu_id, key, value, acknowledged); |
| + AddChange(data.GetSpecifics().managed_user_shared_setting()); |
| +} |
| + |
| +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(); |
| + scoped_utility_.reset(NULL); |
|
Nikita (slow)
2014/04/23 09:50:17
Is this needed? Can you just rely on scoped_ptr de
Denis Kuznetsov (DE-MUC)
2014/04/24 17:44:09
Done.
|
| +} |
| + |
| +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'", current_page.c_str())); |
| + |
| + 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'", current_page.c_str())); |
| + |
| + 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. |
|
Nikita (slow)
2014/04/23 09:50:17
nit: insert empty line.
Denis Kuznetsov (DE-MUC)
2014/04/24 17:44:09
Done.
|
| + JSEval("$('managed-user-creation-next-button').click()"); |
| + login_observer.Wait(); |
| + // OAuth token is valid. |
|
Nikita (slow)
2014/04/23 09:50:17
nit: insert empty line.
Denis Kuznetsov (DE-MUC)
2014/04/24 17:44:09
Done.
|
| + UserManager::Get()->SaveUserOAuthStatus(kTestManager, |
| + User::OAUTH2_TOKEN_STATUS_VALID); |
| + base::RunLoop().RunUntilIdle(); |
| + |
| + // Check the page have changed. |
| + JSExpect(StringPrintf("%s == 'username'", current_page.c_str())); |
| +} |
| + |
| +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'", current_page.c_str())); |
| + 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) { |
|
Nikita (slow)
2014/04/23 09:50:17
nit: drop {}
Denis Kuznetsov (DE-MUC)
2014/04/24 17:44:09
Done.
|
| + EXPECT_CALL(*mock_homedir_methods_, MountEx(_, _, _, _)).Times(1); |
| + } |
| + // Log in as supervised user, make sure that everything works. |
|
Nikita (slow)
2014/04/23 09:50:17
nit: insert empty line.
Denis Kuznetsov (DE-MUC)
2014/04/24 17:44:09
Done.
|
| + ASSERT_EQ(3UL, UserManager::Get()->GetUsers().size()); |
| + // Created supervised user have to be first in a list. |
|
Nikita (slow)
2014/04/23 09:50:17
nit: insert empty line.
Denis Kuznetsov (DE-MUC)
2014/04/24 17:44:09
Done.
|
| + 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) { |
|
Nikita (slow)
2014/04/23 09:50:17
nit: drop {}
Denis Kuznetsov (DE-MUC)
2014/04/24 17:44:09
Done.
|
| + ::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. |
|
Nikita (slow)
2014/04/23 09:50:17
nit: insert empty line.
Denis Kuznetsov (DE-MUC)
2014/04/24 17:44:09
Done.
|
| + 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. |
| + |
|
Nikita (slow)
2014/04/23 09:50:17
nit: drop empty line.
Denis Kuznetsov (DE-MUC)
2014/04/24 17:44:09
Done.
|
| + ASSERT_EQ(original_user_count, UserManager::Get()->GetUsers().size()); |
| + // Created supervised user have to be first in a list. |
|
Nikita (slow)
2014/04/23 09:50:17
nit: insert empty line.
Denis Kuznetsov (DE-MUC)
2014/04/24 17:44:09
Done.
|
| + 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 |