chrome/browser/profiles/profile_test_util.cc

// Copyright 2022 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/profiles/profile_test_util.h"

#include "base/files/file_path.h"
#include "base/run_loop.h"
#include "base/test/bind.h"
#include "base/test/test_future.h"
#include "build/build_config.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/profiles/profile_manager.h"
#include "testing/gtest/include/gtest/gtest.h"

#if !BUILDFLAG(IS_ANDROID)
#include "chrome/browser/profiles/profile_window.h"
#endif  // BUILDFLAG(IS_ANDROID)

namespace profiles::testing {

Profile* CreateProfileSync(ProfileManager* profile_manager,
                           const base::FilePath& path) {
  Profile* created_profile = nullptr;
  base::RunLoop run_loop;
  profile_manager->CreateProfileAsync(
      path, base::BindLambdaForTesting(
                [&run_loop, &created_profile](Profile* profile,
                                              Profile::CreateStatus status) {
                  switch (status) {
                    case Profile::CREATE_STATUS_LOCAL_FAIL:
                      NOTREACHED();
                      return;
                    case Profile::CREATE_STATUS_CREATED:
                      // Do nothing, wait for the profile to be initialized.
                      return;
                    case Profile::CREATE_STATUS_INITIALIZED:
                      created_profile = profile;
                      run_loop.Quit();
                      return;
                  }
                }));
  run_loop.Run();
  return created_profile;
}

#if !BUILDFLAG(IS_ANDROID)

void SwitchToProfileSync(const base::FilePath& path, bool always_create) {
  base::test::TestFuture<Profile*> future;
  profiles::SwitchToProfile(path, always_create, future.GetCallback());
  ASSERT_TRUE(future.Wait()) << "profiles::SwitchToProfile() did not complete";
}
#endif  // BUILDFLAG(IS_ANDROID)
}  // namespace profiles::testing