cryptohome/challenge_credentials/mock_challenge_credentials_helper.h

// Copyright 2020 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CRYPTOHOME_CHALLENGE_CREDENTIALS_MOCK_CHALLENGE_CREDENTIALS_HELPER_H_
#define CRYPTOHOME_CHALLENGE_CREDENTIALS_MOCK_CHALLENGE_CREDENTIALS_HELPER_H_

#include <cstdint>
#include <map>
#include <memory>
#include <string>
#include <vector>

#include <base/callback.h>
#include <brillo/secure_blob.h>
#include <gmock/gmock.h>

#include "cryptohome/challenge_credentials/challenge_credentials_helper.h"
#include "cryptohome/key_challenge_service.h"

namespace cryptohome {

class MockChallengeCredentialsHelper : public ChallengeCredentialsHelper {
 public:
  MockChallengeCredentialsHelper() = default;
  ~MockChallengeCredentialsHelper() = default;

  MOCK_METHOD(void,
              GenerateNew,
              (const std::string& account_id,
               const structure::ChallengePublicKeyInfo& public_key_info,
               const std::string& obfuscated_username,
               std::unique_ptr<KeyChallengeService> key_challenge_service,
               GenerateNewCallback callback),
              (override));
  MOCK_METHOD(void,
              Decrypt,
              (const std::string& account_id,
               const structure::ChallengePublicKeyInfo& public_key_info,
               const structure::SignatureChallengeInfo& keyset_challenge_info,
               bool locked_to_single_user,
               std::unique_ptr<KeyChallengeService> key_challenge_service,
               DecryptCallback callback),
              (override));
  MOCK_METHOD(void,
              VerifyKey,
              (const std::string& account_id,
               const structure::ChallengePublicKeyInfo& public_key_info,
               std::unique_ptr<KeyChallengeService> key_challenge_service,
               VerifyKeyCallback callback),
              (override));
};

}  // namespace cryptohome

#endif  // CRYPTOHOME_CHALLENGE_CREDENTIALS_MOCK_CHALLENGE_CREDENTIALS_HELPER_H_