| // 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/credentials.h" |
| #include "cryptohome/key_challenge_service.h" |
| #include "cryptohome/vault_keyset.pb.h" |
| |
| namespace cryptohome { |
| |
| class MockChallengeCredentialsHelper : public ChallengeCredentialsHelper { |
| public: |
| MockChallengeCredentialsHelper() = default; |
| ~MockChallengeCredentialsHelper() = default; |
| |
| MOCK_METHOD( |
| void, |
| GenerateNew, |
| (const std::string& account_id, |
| const KeyData& key_data, |
| (const std::vector<std::map<uint32_t, brillo::Blob>>& pcr_restrictions), |
| std::unique_ptr<KeyChallengeService> key_challenge_service, |
| GenerateNewCallback callback), |
| (override)); |
| MOCK_METHOD(void, |
| Decrypt, |
| (const std::string& account_id, |
| const KeyData& key_data, |
| const KeysetSignatureChallengeInfo& keyset_challenge_info, |
| std::unique_ptr<KeyChallengeService> key_challenge_service, |
| DecryptCallback callback), |
| (override)); |
| MOCK_METHOD(void, |
| VerifyKey, |
| (const std::string& account_id, |
| const KeyData& key_data, |
| std::unique_ptr<KeyChallengeService> key_challenge_service, |
| VerifyKeyCallback callback), |
| (override)); |
| }; |
| |
| } // namespace cryptohome |
| |
| #endif // CRYPTOHOME_CHALLENGE_CREDENTIALS_MOCK_CHALLENGE_CREDENTIALS_HELPER_H_ |