chrome/browser/ash/drive/drivefs_test_support.cc

// Copyright 2018 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/drive/drivefs_test_support.h"

#include <string>
#include <utility>

#include "base/files/file_util.h"
#include "base/functional/bind.h"
#include "base/json/json_writer.h"
#include "base/path_service.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/string_util.h"
#include "base/test/bind.h"
#include "chrome/browser/ash/profiles/profile_helper.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/common/chrome_constants.h"
#include "chrome/common/chrome_paths.h"
#include "components/account_id/account_id.h"
#include "components/drive/drive_pref_names.h"
#include "components/prefs/pref_service.h"
#include "components/user_manager/user.h"
#include "crypto/obsolete/md5.h"
#include "google_apis/gaia/gaia_id.h"

namespace drive {

const char FakeDriveFsHelper::kPredefinedProfileSalt[] = "salt";
const char FakeDriveFsHelper::kDefaultUserEmail[] = "testuser@gmail.com";
const GaiaId::Literal FakeDriveFsHelper::kDefaultGaiaId("123456");

FakeDriveFsHelper::FakeDriveFsHelper(Profile* profile,
                                     const base::FilePath& mount_path)
    : mount_path_(mount_path), fake_drivefs_(mount_path_) {
  profile->GetPrefs()->SetString(
      drive::prefs::kDriveFsProfileSalt,
      drive::FakeDriveFsHelper::kPredefinedProfileSalt);
  fake_drivefs_.RegisterMountingForAccountId(
      base::BindLambdaForTesting([profile]() {
        auto* user = ash::ProfileHelper::Get()->GetUserByProfile(profile);
        if (!user)
          return std::string();

        auto md5 = crypto::obsolete::Md5::MakeMd5HasherForTesting();
        md5.Update(FakeDriveFsHelper::kPredefinedProfileSalt);
        md5.Update("-");
        md5.Update(user->GetAccountId().GetAccountIdKey());
        return base::ToLowerASCII(base::HexEncode(md5.Finish()));
      }));
}
FakeDriveFsHelper::~FakeDriveFsHelper() = default;

base::RepeatingCallback<std::unique_ptr<drivefs::DriveFsBootstrapListener>()>
FakeDriveFsHelper::CreateFakeDriveFsListenerFactory() {
  return base::BindRepeating(&drivefs::FakeDriveFs::CreateMojoListener,
                             base::Unretained(&fake_drivefs_));
}

bool SetUpUserDataDirectoryForDriveFsTest() {
  AccountId account_id =
      AccountId::FromUserEmailGaiaId(FakeDriveFsHelper::kDefaultUserEmail,
                                     GaiaId(FakeDriveFsHelper::kDefaultGaiaId));
  return SetUpUserDataDirectoryForDriveFsTest(account_id);
}

bool SetUpUserDataDirectoryForDriveFsTest(const AccountId& account_id) {
  // Account type must be GOOGLE to use Drive.
  CHECK(account_id.GetAccountType() == AccountType::GOOGLE);
  base::Value::List known_users_list;
  base::Value::Dict user_dict;
  user_dict.Set("account_type",
                AccountId::AccountTypeToString(account_id.GetAccountType()));
  user_dict.Set("email", account_id.GetUserEmail());
  user_dict.Set("gaia_id", account_id.GetGaiaId().ToString());
  known_users_list.Append(std::move(user_dict));

  base::Value::Dict local_state;
  local_state.Set("KnownUsers", std::move(known_users_list));

  std::string local_state_json;
  if (!base::JSONWriter::Write(local_state, &local_state_json))
    return false;

  base::FilePath local_state_file;
  if (!base::PathService::Get(chrome::DIR_USER_DATA, &local_state_file))
    return false;
  local_state_file = local_state_file.Append(chrome::kLocalStateFilename);
  return base::WriteFile(local_state_file, local_state_json);
}

}  // namespace drive