// 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.

#ifndef CHROME_BROWSER_ASH_LOGIN_TEST_CRYPTOHOME_MIXIN_H_
#define CHROME_BROWSER_ASH_LOGIN_TEST_CRYPTOHOME_MIXIN_H_

#include <string>
#include <utility>

#include "chrome/test/base/mixin_based_in_process_browser_test.h"
#include "chromeos/ash/components/dbus/cryptohome/rpc.pb.h"
#include "chromeos/ash/components/dbus/userdataauth/fake_userdataauth_client.h"
#include "components/account_id/account_id.h"

namespace ash {

namespace test {
struct UserAuthConfig;
}

// Mixin that acts as a broker between tests and FakeUserDataAuthClient,
// handling all interactions and transformations.
class CryptohomeMixin : public InProcessBrowserTestMixin,
                        public FakeUserDataAuthClient::TestApi {
 public:
  explicit CryptohomeMixin(InProcessBrowserTestMixinHost* host);
  CryptohomeMixin(const CryptohomeMixin&) = delete;
  CryptohomeMixin& operator=(const CryptohomeMixin&) = delete;
  ~CryptohomeMixin() override;

  void SetUpOnMainThread() override;

  void ApplyAuthConfig(const AccountId& user,
                       const test::UserAuthConfig& config);
  void ApplyAuthConfigIfUserExists(const AccountId& user,
                                   const test::UserAuthConfig& config);

  void MarkUserAsExisting(const AccountId& user);
  // Returns {authsession_id, broadcast_id} pair.
  std::pair<std::string, std::string> AddSession(const AccountId& user,
                                                 bool authenticated);
  void AddGaiaPassword(const AccountId& user, std::string password);
  void AddLocalPassword(const AccountId& user, std::string password);
  void AddCryptohomePin(const AccountId& user,
                        const std::string& pin,
                        const std::string& pin_salt);
  void SetPinLocked(const AccountId& user, bool locked);
  void SetPinType(const AccountId& user, bool legacy);
  bool HasPinFactor(const AccountId& user);
  bool HasLocalPasswordFactor(const AccountId& user);
  bool HasGaiaPasswordFactor(const AccountId& user);
  void AddRecoveryFactor(const AccountId& user);
  bool HasRecoveryFactor(const AccountId& user);

  void SendLegacyFingerprintSuccessScan();
  void SendLegacyFingerprintFailureScan();
  void SendLegacyFingerprintFailureLockoutScan();

  bool IsAuthenticated(const AccountId& user);
};

}  // namespace ash

#endif  // CHROME_BROWSER_ASH_LOGIN_TEST_CRYPTOHOME_MIXIN_H_