| // Copyright 2021 The Chromium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "chrome/browser/ash/crosapi/browser_data_migrator.h" |
| |
| #include <string> |
| |
| #include "ash/constants/ash_features.h" |
| #include "ash/constants/ash_switches.h" |
| #include "base/files/file_util.h" |
| #include "base/files/scoped_temp_dir.h" |
| #include "base/run_loop.h" |
| #include "base/strings/string_piece.h" |
| #include "base/test/bind.h" |
| #include "base/test/metrics/histogram_tester.h" |
| #include "base/test/scoped_command_line.h" |
| #include "base/test/scoped_feature_list.h" |
| #include "base/test/task_environment.h" |
| #include "base/version.h" |
| #include "chrome/browser/ash/crosapi/browser_data_migrator_util.h" |
| #include "chrome/browser/ash/crosapi/browser_util.h" |
| #include "chrome/browser/ash/crosapi/fake_migration_progress_tracker.h" |
| #include "chrome/browser/ash/login/users/fake_chrome_user_manager.h" |
| #include "chrome/browser/ash/profiles/profile_helper.h" |
| #include "chromeos/ash/components/dbus/session_manager/fake_session_manager_client.h" |
| #include "components/prefs/testing_pref_service.h" |
| #include "components/user_manager/fake_user_manager.h" |
| #include "components/user_manager/scoped_user_manager.h" |
| #include "components/version_info/version_info.h" |
| #include "content/public/test/browser_task_environment.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| #include "third_party/abseil-cpp/absl/types/optional.h" |
| |
| namespace ash { |
| |
| namespace { |
| constexpr char kFirstRun[] = "First Run"; |
| } // namespace |
| |
| class BrowserDataMigratorImplTest : public ::testing::Test { |
| public: |
| void SetUp() override { |
| // Setup `user_data_dir_` as below. |
| // ./ /* user_data_dir_ */ |
| // |- user/ /* from_dir_ */ |
| // |- data |
| |
| ASSERT_TRUE(user_data_dir_.CreateUniqueTempDir()); |
| from_dir_ = user_data_dir_.GetPath().Append("user"); |
| |
| ASSERT_TRUE(base::CreateDirectory(from_dir_)); |
| |
| BrowserDataMigratorImpl::RegisterLocalStatePrefs(pref_service_.registry()); |
| crosapi::browser_util::RegisterLocalStatePrefs(pref_service_.registry()); |
| } |
| |
| void TearDown() override { EXPECT_TRUE(user_data_dir_.Delete()); } |
| |
| base::ScopedTempDir user_data_dir_; |
| base::FilePath from_dir_; |
| TestingPrefServiceSimple pref_service_; |
| }; |
| |
| TEST_F(BrowserDataMigratorImplTest, ManipulateMigrationAttemptCount) { |
| const std::string user_id_hash = "user"; |
| |
| EXPECT_EQ(BrowserDataMigratorImpl::GetMigrationAttemptCountForUser( |
| &pref_service_, user_id_hash), |
| 0); |
| BrowserDataMigratorImpl::UpdateMigrationAttemptCountForUser(&pref_service_, |
| user_id_hash); |
| EXPECT_EQ(BrowserDataMigratorImpl::GetMigrationAttemptCountForUser( |
| &pref_service_, user_id_hash), |
| 1); |
| |
| BrowserDataMigratorImpl::UpdateMigrationAttemptCountForUser(&pref_service_, |
| user_id_hash); |
| EXPECT_EQ(BrowserDataMigratorImpl::GetMigrationAttemptCountForUser( |
| &pref_service_, user_id_hash), |
| 2); |
| |
| BrowserDataMigratorImpl::ClearMigrationAttemptCountForUser(&pref_service_, |
| user_id_hash); |
| EXPECT_EQ(BrowserDataMigratorImpl::GetMigrationAttemptCountForUser( |
| &pref_service_, user_id_hash), |
| 0); |
| } |
| |
| TEST_F(BrowserDataMigratorImplTest, Migrate) { |
| base::test::TaskEnvironment task_environment; |
| std::unique_ptr<MigrationProgressTracker> progress_tracker = |
| std::make_unique<FakeMigrationProgressTracker>(); |
| const std::string user_id_hash = "abcd"; |
| BrowserDataMigratorImpl::SetMigrationStep( |
| &pref_service_, BrowserDataMigratorImpl::MigrationStep::kRestartCalled); |
| // Set migration attempt count to 1. |
| BrowserDataMigratorImpl::UpdateMigrationAttemptCountForUser(&pref_service_, |
| user_id_hash); |
| |
| base::RunLoop run_loop; |
| std::unique_ptr<BrowserDataMigratorImpl> migrator = |
| std::make_unique<BrowserDataMigratorImpl>( |
| from_dir_, user_id_hash, base::DoNothing(), &pref_service_); |
| absl::optional<BrowserDataMigrator::Result> result; |
| migrator->Migrate( |
| crosapi::browser_util::MigrationMode::kCopy, |
| base::BindLambdaForTesting([&out_result = result, &run_loop]( |
| BrowserDataMigrator::Result result) { |
| run_loop.Quit(); |
| out_result = result; |
| })); |
| run_loop.Run(); |
| |
| const base::FilePath new_user_data_dir = |
| from_dir_.Append(browser_data_migrator_util::kLacrosDir); |
| const base::FilePath new_profile_data_dir = |
| new_user_data_dir.Append("Default"); |
| // Check that `First Run` file is created inside the new data directory. |
| EXPECT_TRUE(base::PathExists(new_user_data_dir.Append(kFirstRun))); |
| // Check that migration is marked as completed for the user. |
| EXPECT_TRUE(crosapi::browser_util::IsProfileMigrationCompletedForUser( |
| &pref_service_, user_id_hash, |
| crosapi::browser_util::MigrationMode::kCopy)); |
| ASSERT_TRUE(result.has_value()); |
| EXPECT_EQ(BrowserDataMigrator::ResultKind::kSucceeded, result->kind); |
| EXPECT_EQ(BrowserDataMigratorImpl::GetMigrationStep(&pref_service_), |
| BrowserDataMigratorImpl::MigrationStep::kEnded); |
| // Successful migration should clear the migration attempt count. |
| EXPECT_EQ(BrowserDataMigratorImpl::GetMigrationAttemptCountForUser( |
| &pref_service_, user_id_hash), |
| 0); |
| // Data version should be updated to the current version after a migration. |
| EXPECT_EQ(crosapi::browser_util::GetDataVer(&pref_service_, user_id_hash), |
| version_info::GetVersion()); |
| } |
| |
| TEST_F(BrowserDataMigratorImplTest, MigrateCancelled) { |
| base::test::TaskEnvironment task_environment; |
| std::unique_ptr<MigrationProgressTracker> progress_tracker = |
| std::make_unique<FakeMigrationProgressTracker>(); |
| const std::string user_id_hash = "abcd"; |
| BrowserDataMigratorImpl::SetMigrationStep( |
| &pref_service_, BrowserDataMigratorImpl::MigrationStep::kRestartCalled); |
| // Set migration attempt count to 1. |
| BrowserDataMigratorImpl::UpdateMigrationAttemptCountForUser(&pref_service_, |
| user_id_hash); |
| |
| base::RunLoop run_loop; |
| std::unique_ptr<BrowserDataMigratorImpl> migrator = |
| std::make_unique<BrowserDataMigratorImpl>( |
| from_dir_, user_id_hash, base::DoNothing(), &pref_service_); |
| absl::optional<BrowserDataMigrator::Result> result; |
| migrator->Migrate( |
| crosapi::browser_util::MigrationMode::kCopy, |
| base::BindLambdaForTesting([&out_result = result, &run_loop]( |
| BrowserDataMigrator::Result result) { |
| run_loop.Quit(); |
| out_result = result; |
| })); |
| migrator->Cancel(); |
| run_loop.Run(); |
| |
| const base::FilePath new_user_data_dir = |
| from_dir_.Append(browser_data_migrator_util::kLacrosDir); |
| const base::FilePath new_profile_data_dir = |
| new_user_data_dir.Append("Default"); |
| EXPECT_FALSE(base::PathExists(new_user_data_dir.Append(kFirstRun))); |
| EXPECT_FALSE(crosapi::browser_util::IsProfileMigrationCompletedForUser( |
| &pref_service_, user_id_hash, |
| crosapi::browser_util::MigrationMode::kCopy)); |
| ASSERT_TRUE(result.has_value()); |
| EXPECT_EQ(BrowserDataMigrator::ResultKind::kCancelled, result->kind); |
| EXPECT_EQ(BrowserDataMigratorImpl::GetMigrationStep(&pref_service_), |
| BrowserDataMigratorImpl::MigrationStep::kEnded); |
| // If migration fails, migration attempt count should not be cleared thus |
| // should remain as 1. |
| EXPECT_EQ(BrowserDataMigratorImpl::GetMigrationAttemptCountForUser( |
| &pref_service_, user_id_hash), |
| 1); |
| // Even if migration is cancelled, lacros data dir is cleared and thus data |
| // version should be updated. |
| EXPECT_EQ(crosapi::browser_util::GetDataVer(&pref_service_, user_id_hash), |
| version_info::GetVersion()); |
| } |
| |
| TEST_F(BrowserDataMigratorImplTest, MigrateOutOfDiskForCopy) { |
| // Emulate the situation of out-of-disk. |
| browser_data_migrator_util::ScopedExtraBytesRequiredToBeFreedForTesting |
| scoped_extra_bytes(100); |
| |
| base::test::TaskEnvironment task_environment; |
| const std::string user_id_hash = "abcd"; |
| BrowserDataMigratorImpl::SetMigrationStep( |
| &pref_service_, BrowserDataMigratorImpl::MigrationStep::kRestartCalled); |
| // Set migration attempt count to 1. |
| BrowserDataMigratorImpl::UpdateMigrationAttemptCountForUser(&pref_service_, |
| user_id_hash); |
| |
| base::RunLoop run_loop; |
| std::unique_ptr<BrowserDataMigratorImpl> migrator = |
| std::make_unique<BrowserDataMigratorImpl>( |
| from_dir_, user_id_hash, base::DoNothing(), &pref_service_); |
| absl::optional<BrowserDataMigrator::Result> result; |
| migrator->Migrate( |
| crosapi::browser_util::MigrationMode::kCopy, |
| base::BindLambdaForTesting([&out_result = result, &run_loop]( |
| BrowserDataMigrator::Result result) { |
| run_loop.Quit(); |
| out_result = result; |
| })); |
| run_loop.Run(); |
| |
| ASSERT_TRUE(result.has_value()); |
| EXPECT_EQ(BrowserDataMigrator::ResultKind::kFailed, result->kind); |
| // |required_size| should carry the data. |
| EXPECT_EQ(100u, result->required_size); |
| } |
| |
| TEST_F(BrowserDataMigratorImplTest, MigrateOutOfDiskForMove) { |
| base::test::ScopedFeatureList feature_list( |
| ash::features::kLacrosMoveProfileMigration); |
| |
| // Emulate the situation of out-of-disk. |
| browser_data_migrator_util::ScopedExtraBytesRequiredToBeFreedForTesting |
| scoped_extra_bytes(100); |
| |
| base::test::TaskEnvironment task_environment; |
| const std::string user_id_hash = "abcd"; |
| BrowserDataMigratorImpl::SetMigrationStep( |
| &pref_service_, BrowserDataMigratorImpl::MigrationStep::kRestartCalled); |
| // Set migration attempt count to 1. |
| BrowserDataMigratorImpl::UpdateMigrationAttemptCountForUser(&pref_service_, |
| user_id_hash); |
| |
| base::RunLoop run_loop; |
| std::unique_ptr<BrowserDataMigratorImpl> migrator = |
| std::make_unique<BrowserDataMigratorImpl>( |
| from_dir_, user_id_hash, base::DoNothing(), &pref_service_); |
| absl::optional<BrowserDataMigrator::Result> result; |
| migrator->Migrate( |
| crosapi::browser_util::MigrationMode::kMove, |
| base::BindLambdaForTesting([&out_result = result, &run_loop]( |
| BrowserDataMigrator::Result result) { |
| run_loop.Quit(); |
| out_result = result; |
| })); |
| run_loop.Run(); |
| |
| ASSERT_TRUE(result.has_value()); |
| EXPECT_EQ(BrowserDataMigrator::ResultKind::kFailed, result->kind); |
| // |required_size| should carry the data. |
| EXPECT_EQ(100u, result->required_size); |
| } |
| |
| class BrowserDataMigratorRestartTest : public ::testing::Test { |
| public: |
| BrowserDataMigratorRestartTest() = default; |
| ~BrowserDataMigratorRestartTest() override = default; |
| |
| void SetUp() override { |
| fake_user_manager_ = new ash::FakeChromeUserManager; |
| scoped_user_manager_ = std::make_unique<user_manager::ScopedUserManager>( |
| base::WrapUnique(fake_user_manager_)); |
| fake_user_manager_->CreateLocalState(); |
| |
| auto* local_state_simple = |
| static_cast<TestingPrefServiceSimple*>(local_state()); |
| BrowserDataMigratorImpl::RegisterLocalStatePrefs( |
| local_state_simple->registry()); |
| crosapi::browser_util::RegisterLocalStatePrefs( |
| local_state_simple->registry()); |
| } |
| |
| void AddRegularUser(const std::string& email) { |
| AccountId account_id = AccountId::FromUserEmail(email); |
| const user_manager::User* user = fake_user_manager_->AddUser(account_id); |
| fake_user_manager_->UserLoggedIn(account_id, user->username_hash(), |
| /*browser_restart=*/false, |
| /*is_child=*/false); |
| ash::ProfileHelper::Get()->SetUserToProfileMappingForTesting( |
| user, &testing_profile_); |
| } |
| |
| protected: |
| TestingProfile* testing_profile() { return &testing_profile_; } |
| ash::FakeChromeUserManager* user_manager() { return fake_user_manager_; } |
| PrefService* local_state() { return fake_user_manager_->GetLocalState(); } |
| |
| private: |
| content::BrowserTaskEnvironment task_environment_; |
| TestingProfile testing_profile_; |
| ash::FakeChromeUserManager* fake_user_manager_ = nullptr; |
| std::unique_ptr<user_manager::ScopedUserManager> scoped_user_manager_; |
| FakeSessionManagerClient session_manager_; |
| }; |
| |
| TEST_F(BrowserDataMigratorRestartTest, MaybeRestartToMigrateWithMigrationStep) { |
| bool restart_called = false; |
| ScopedRestartAttemptForTesting scoped_restart_attempt( |
| base::BindLambdaForTesting( |
| [&restart_called]() { restart_called = true; })); |
| |
| BrowserDataMigratorImpl::SetMigrationStep( |
| local_state(), BrowserDataMigratorImpl::MigrationStep::kRestartCalled); |
| EXPECT_FALSE(BrowserDataMigratorImpl::MaybeRestartToMigrate( |
| AccountId::FromUserEmail("fake@gmail.com"), "abcde", |
| crosapi::browser_util::PolicyInitState::kAfterInit)); |
| |
| BrowserDataMigratorImpl::SetMigrationStep( |
| local_state(), BrowserDataMigratorImpl::MigrationStep::kStarted); |
| EXPECT_FALSE(BrowserDataMigratorImpl::MaybeRestartToMigrate( |
| AccountId::FromUserEmail("fake@gmail.com"), "abcde", |
| crosapi::browser_util::PolicyInitState::kAfterInit)); |
| |
| BrowserDataMigratorImpl::SetMigrationStep( |
| local_state(), BrowserDataMigratorImpl::MigrationStep::kEnded); |
| EXPECT_FALSE(BrowserDataMigratorImpl::MaybeRestartToMigrate( |
| AccountId::FromUserEmail("fake@gmail.com"), "abcde", |
| crosapi::browser_util::PolicyInitState::kAfterInit)); |
| } |
| |
| TEST_F(BrowserDataMigratorRestartTest, MaybeRestartToMigrateWithCommandLine) { |
| bool restart_called = false; |
| ScopedRestartAttemptForTesting scoped_restart_attempt( |
| base::BindLambdaForTesting( |
| [&restart_called]() { restart_called = true; })); |
| |
| base::test::ScopedFeatureList feature_list; |
| AddRegularUser("user@gmail.com"); |
| const user_manager::User* const user = |
| ash::ProfileHelper::Get()->GetUserByProfile(testing_profile()); |
| { |
| base::test::ScopedCommandLine command_line; |
| command_line.GetProcessCommandLine()->AppendSwitchASCII( |
| switches::kForceBrowserDataMigrationForTesting, "force-skip"); |
| EXPECT_FALSE(BrowserDataMigratorImpl::MaybeRestartToMigrate( |
| user->GetAccountId(), user->username_hash(), |
| crosapi::browser_util::PolicyInitState::kAfterInit)); |
| } |
| { |
| base::test::ScopedCommandLine command_line; |
| command_line.GetProcessCommandLine()->AppendSwitchASCII( |
| switches::kForceBrowserDataMigrationForTesting, "force-migration"); |
| EXPECT_TRUE(BrowserDataMigratorImpl::MaybeRestartToMigrate( |
| user->GetAccountId(), user->username_hash(), |
| crosapi::browser_util::PolicyInitState::kAfterInit)); |
| } |
| } |
| |
| TEST_F(BrowserDataMigratorRestartTest, MaybeRestartToMigrateWithDiskCheck) { |
| bool restart_called = false; |
| ScopedRestartAttemptForTesting scoped_restart_attempt( |
| base::BindLambdaForTesting( |
| [&restart_called]() { restart_called = true; })); |
| |
| base::test::ScopedFeatureList feature_list; |
| AddRegularUser("user@gmail.com"); |
| const user_manager::User* const user = |
| ash::ProfileHelper::Get()->GetUserByProfile(testing_profile()); |
| // If MaybeRestartToMigrate will skip the restarting, WithDiskCheck variation |
| // also skips it. |
| { |
| restart_called = false; |
| base::test::ScopedCommandLine command_line; |
| command_line.GetProcessCommandLine()->AppendSwitchASCII( |
| switches::kForceBrowserDataMigrationForTesting, "force-skip"); |
| absl::optional<bool> result; |
| BrowserDataMigratorImpl::MaybeRestartToMigrateWithDiskCheck( |
| user->GetAccountId(), user->username_hash(), |
| base::BindLambdaForTesting( |
| [&out_result = result](bool result, |
| const absl::optional<uint64_t>& size) { |
| out_result = result; |
| })); |
| EXPECT_TRUE(result.has_value()); |
| EXPECT_FALSE(result.value()); |
| EXPECT_FALSE(restart_called); |
| } |
| |
| // If MaybeRestartToMigrate will trigger the restarting, WithDiskCheck |
| // variation will see additional disk size check. |
| { |
| restart_called = false; |
| base::test::ScopedCommandLine command_line; |
| command_line.GetProcessCommandLine()->AppendSwitchASCII( |
| switches::kForceBrowserDataMigrationForTesting, "force-migration"); |
| // Inject the behavior that the disk does not have enough space. |
| browser_data_migrator_util::ScopedExtraBytesRequiredToBeFreedForTesting |
| scoped_extra_bytes(1024 * 1024); |
| |
| absl::optional<bool> result; |
| absl::optional<uint64_t> out_size; |
| base::RunLoop run_loop; |
| BrowserDataMigratorImpl::MaybeRestartToMigrateWithDiskCheck( |
| user->GetAccountId(), user->username_hash(), |
| base::BindLambdaForTesting( |
| [&out_result = result, &out_size, &run_loop]( |
| bool result, const absl::optional<uint64_t>& size) { |
| run_loop.Quit(); |
| out_result = result; |
| out_size = size; |
| })); |
| run_loop.Run(); |
| ASSERT_TRUE(result.has_value()); |
| EXPECT_FALSE(result.value()); |
| EXPECT_EQ(1024u * 1024u, out_size); |
| EXPECT_FALSE(restart_called); |
| } |
| |
| { |
| restart_called = false; |
| base::test::ScopedCommandLine command_line; |
| command_line.GetProcessCommandLine()->AppendSwitchASCII( |
| switches::kForceBrowserDataMigrationForTesting, "force-migration"); |
| // Inject the behavior that the disk has enough space for the migration. |
| browser_data_migrator_util::ScopedExtraBytesRequiredToBeFreedForTesting |
| scoped_extra_bytes(0); |
| |
| absl::optional<bool> result; |
| base::RunLoop run_loop; |
| BrowserDataMigratorImpl::MaybeRestartToMigrateWithDiskCheck( |
| user->GetAccountId(), user->username_hash(), |
| base::BindLambdaForTesting( |
| [&out_result = result, &run_loop]( |
| bool result, const absl::optional<uint64_t>& size) { |
| run_loop.Quit(); |
| out_result = result; |
| })); |
| run_loop.Run(); |
| ASSERT_TRUE(result.has_value()); |
| EXPECT_TRUE(result.value()); |
| EXPECT_TRUE(restart_called); |
| } |
| } |
| |
| TEST_F(BrowserDataMigratorRestartTest, MaybeRestartToMigrateMoveAfterCopy) { |
| // Check that `MaybeRestartToMigrateInternal()` returns true after completion |
| // of copy migration if move migration is enabled. |
| AddRegularUser("user@gmail.com"); |
| const user_manager::User* const user = |
| ash::ProfileHelper::Get()->GetUserByProfile(testing_profile()); |
| |
| crosapi::browser_util::RecordDataVer( |
| local_state(), user->username_hash(), |
| base::Version( |
| base::StringPiece(crosapi::browser_util::kRequiredDataVersion))); |
| |
| { |
| // If Lacros is not enabled, migration should not run. |
| base::test::ScopedFeatureList feature_list; |
| EXPECT_EQ(crosapi::browser_util::GetMigrationMode( |
| user, crosapi::browser_util::PolicyInitState::kAfterInit), |
| crosapi::browser_util::MigrationMode::kCopy); |
| EXPECT_FALSE(BrowserDataMigratorImpl::MaybeRestartToMigrateInternal( |
| user->GetAccountId(), user->username_hash(), |
| crosapi::browser_util::PolicyInitState::kAfterInit)); |
| } |
| |
| { |
| // If Lacros is enabled, migration should run. |
| base::test::ScopedFeatureList feature_list; |
| feature_list.InitWithFeatures({ash::features::kLacrosSupport}, {}); |
| EXPECT_EQ(crosapi::browser_util::GetMigrationMode( |
| user, crosapi::browser_util::PolicyInitState::kAfterInit), |
| crosapi::browser_util::MigrationMode::kCopy); |
| EXPECT_TRUE(BrowserDataMigratorImpl::MaybeRestartToMigrateInternal( |
| user->GetAccountId(), user->username_hash(), |
| crosapi::browser_util::PolicyInitState::kAfterInit)); |
| } |
| |
| // Mark copy migration as completed. |
| crosapi::browser_util::SetProfileMigrationCompletedForUser( |
| local_state(), user->username_hash(), |
| crosapi::browser_util::MigrationMode::kCopy); |
| { |
| // If migration is marked as completed, migration should not run. |
| base::test::ScopedFeatureList feature_list; |
| feature_list.InitWithFeatures({ash::features::kLacrosSupport}, {}); |
| EXPECT_EQ(crosapi::browser_util::GetMigrationMode( |
| user, crosapi::browser_util::PolicyInitState::kAfterInit), |
| crosapi::browser_util::MigrationMode::kCopy); |
| EXPECT_FALSE(BrowserDataMigratorImpl::MaybeRestartToMigrateInternal( |
| user->GetAccountId(), user->username_hash(), |
| crosapi::browser_util::PolicyInitState::kAfterInit)); |
| } |
| |
| { |
| // If copy migration is marked as completed then migration should not run |
| // even if move migration is enabled. |
| // TODO(crbug.com/1340438): This previously checked the opposite where the |
| // expectation was to run move migration even if copy migration is |
| // completed. Revisit this part if we decide to restore that behaviour. |
| base::test::ScopedFeatureList feature_list; |
| feature_list.InitWithFeatures( |
| {ash::features::kLacrosSupport, ash::features::kLacrosPrimary, |
| ash::features::kLacrosOnly}, |
| {}); |
| EXPECT_EQ(crosapi::browser_util::GetMigrationMode( |
| user, crosapi::browser_util::PolicyInitState::kAfterInit), |
| crosapi::browser_util::MigrationMode::kMove); |
| EXPECT_FALSE(BrowserDataMigratorImpl::MaybeRestartToMigrateInternal( |
| user->GetAccountId(), user->username_hash(), |
| crosapi::browser_util::PolicyInitState::kAfterInit)); |
| } |
| |
| // Mark move migration as completed. |
| crosapi::browser_util::ClearProfileMigrationCompletedForUser( |
| local_state(), user->username_hash()); |
| crosapi::browser_util::SetProfileMigrationCompletedForUser( |
| local_state(), user->username_hash(), |
| crosapi::browser_util::MigrationMode::kMove); |
| { |
| // If move migration is marked as completed, move migration should not run. |
| base::test::ScopedFeatureList feature_list; |
| feature_list.InitWithFeatures( |
| {ash::features::kLacrosSupport, ash::features::kLacrosPrimary, |
| ash::features::kLacrosOnly}, |
| {}); |
| EXPECT_EQ(crosapi::browser_util::GetMigrationMode( |
| user, crosapi::browser_util::PolicyInitState::kAfterInit), |
| crosapi::browser_util::MigrationMode::kMove); |
| EXPECT_FALSE(BrowserDataMigratorImpl::MaybeRestartToMigrateInternal( |
| user->GetAccountId(), user->username_hash(), |
| crosapi::browser_util::PolicyInitState::kAfterInit)); |
| } |
| } |
| |
| TEST_F(BrowserDataMigratorRestartTest, |
| LacrosProfileMigrationForAnyUserDisabled) { |
| AddRegularUser("user@gmail.com"); |
| const user_manager::User* const user = |
| ash::ProfileHelper::Get()->GetUserByProfile(testing_profile()); |
| |
| crosapi::browser_util::RecordDataVer( |
| local_state(), user->username_hash(), |
| base::Version( |
| base::StringPiece(crosapi::browser_util::kRequiredDataVersion))); |
| { |
| // If Lacros is enabled, migration should run. |
| base::test::ScopedFeatureList feature_list; |
| feature_list.InitWithFeatures({ash::features::kLacrosSupport}, {}); |
| EXPECT_TRUE(BrowserDataMigratorImpl::MaybeRestartToMigrateInternal( |
| user->GetAccountId(), user->username_hash(), |
| crosapi::browser_util::PolicyInitState::kAfterInit)); |
| } |
| |
| { |
| // If Lacros is enabled but `kLacrosProfileMigrationForAnyUser` is disabled, |
| // migration should not run. |
| base::test::ScopedFeatureList feature_list; |
| feature_list.InitWithFeatures( |
| {}, {ash::features::kLacrosProfileMigrationForAnyUser}); |
| EXPECT_FALSE(BrowserDataMigratorImpl::MaybeRestartToMigrateInternal( |
| user->GetAccountId(), user->username_hash(), |
| crosapi::browser_util::PolicyInitState::kAfterInit)); |
| } |
| } |
| |
| TEST_F(BrowserDataMigratorRestartTest, |
| LacrosProfileMigrationForAnyUserDisabledForGoogler) { |
| AddRegularUser("user@google.com"); |
| const user_manager::User* const user = |
| ash::ProfileHelper::Get()->GetUserByProfile(testing_profile()); |
| |
| crosapi::browser_util::RecordDataVer( |
| local_state(), user->username_hash(), |
| base::Version( |
| base::StringPiece(crosapi::browser_util::kRequiredDataVersion))); |
| { |
| // If Lacros is enabled, migration should run even if |
| // `kLacrosProfileMigrationForAnyUser` is disabled for Googlers. |
| base::test::ScopedFeatureList feature_list; |
| feature_list.InitWithFeatures( |
| {ash::features::kLacrosSupport}, |
| {ash::features::kLacrosProfileMigrationForAnyUser}); |
| EXPECT_TRUE(BrowserDataMigratorImpl::MaybeRestartToMigrateInternal( |
| user->GetAccountId(), user->username_hash(), |
| crosapi::browser_util::PolicyInitState::kAfterInit)); |
| } |
| } |
| |
| } // namespace ash |
