| // Copyright (c) 2013 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_MOCK_ATTESTATION_H_ |
| #define CRYPTOHOME_MOCK_ATTESTATION_H_ |
| |
| #include "cryptohome/attestation.h" |
| |
| #include <memory> |
| #include <string> |
| |
| #include <brillo/secure_blob.h> |
| #include <gmock/gmock.h> |
| |
| namespace cryptohome { |
| |
| class Platform; |
| |
| class MockAttestation : public Attestation { |
| public: |
| MockAttestation(): Attestation() { } |
| virtual ~MockAttestation() {} |
| |
| MOCK_METHOD(void, |
| Initialize, |
| (Tpm*, |
| TpmInit*, |
| Platform*, |
| Crypto*, |
| InstallAttributes*, |
| const brillo::SecureBlob&, |
| bool), |
| (override)); |
| MOCK_METHOD(bool, IsPreparedForEnrollment, (), (override)); |
| MOCK_METHOD(bool, IsEnrolled, (), (override)); |
| MOCK_METHOD(void, PrepareForEnrollment, (), (override)); |
| MOCK_METHOD(void, CacheEndorsementData, (), (override)); |
| MOCK_METHOD(void, PrepareForEnrollmentAsync, (), (override)); |
| MOCK_METHOD(bool, Verify, (bool), (override)); |
| MOCK_METHOD(bool, VerifyEK, (bool), (override)); |
| MOCK_METHOD(bool, |
| CreateEnrollRequest, |
| (Attestation::PCAType, brillo::SecureBlob*), |
| (override)); |
| MOCK_METHOD(bool, |
| Enroll, |
| (Attestation::PCAType, const brillo::SecureBlob&), |
| (override)); |
| MOCK_METHOD(bool, |
| CreateCertRequest, |
| (Attestation::PCAType, |
| CertificateProfile, |
| const std::string&, |
| const std::string&, |
| brillo::SecureBlob*), |
| (override)); |
| MOCK_METHOD(bool, |
| FinishCertRequest, |
| (const brillo::SecureBlob&, |
| bool, |
| const std::string&, |
| const std::string&, |
| brillo::SecureBlob*), |
| (override)); |
| MOCK_METHOD( |
| bool, |
| GetCertificateChain, |
| (bool, const std::string&, const std::string&, brillo::SecureBlob*), |
| (override)); |
| MOCK_METHOD( |
| bool, |
| GetPublicKey, |
| (bool, const std::string&, const std::string&, brillo::SecureBlob*), |
| (override)); |
| MOCK_METHOD(bool, |
| DoesKeyExist, |
| (bool, const std::string&, const std::string&), |
| (override)); |
| MOCK_METHOD(bool, |
| SignEnterpriseChallenge, |
| (bool, |
| const std::string&, |
| const std::string&, |
| const std::string&, |
| const brillo::SecureBlob&, |
| bool, |
| const brillo::SecureBlob&, |
| brillo::SecureBlob*), |
| (override)); |
| MOCK_METHOD(bool, |
| SignEnterpriseVaChallenge, |
| (Attestation::VAType, |
| bool, |
| const std::string&, |
| const std::string&, |
| const std::string&, |
| const brillo::SecureBlob&, |
| bool, |
| const brillo::SecureBlob&, |
| const std::string&, |
| brillo::SecureBlob*), |
| (override)); |
| MOCK_METHOD(bool, |
| SignSimpleChallenge, |
| (bool, |
| const std::string&, |
| const std::string&, |
| const brillo::SecureBlob&, |
| brillo::SecureBlob*), |
| (override)); |
| MOCK_METHOD(bool, |
| RegisterKey, |
| (bool, const std::string&, const std::string&, bool), |
| (override)); |
| MOCK_METHOD( |
| bool, |
| GetKeyPayload, |
| (bool, const std::string&, const std::string&, brillo::SecureBlob*), |
| (override)); |
| MOCK_METHOD( |
| bool, |
| SetKeyPayload, |
| (bool, const std::string&, const std::string&, const brillo::SecureBlob&), |
| (override)); |
| MOCK_METHOD(bool, |
| DeleteKeysByPrefix, |
| (bool, const std::string&, const std::string&), |
| (override)); |
| MOCK_METHOD(bool, GetEKInfo, (std::string*), (override)); |
| MOCK_METHOD(bool, |
| GetIdentityResetRequest, |
| (const std::string&, brillo::SecureBlob*), |
| (override)); |
| MOCK_METHOD(void, set_database_path, (const char*), (override)); |
| MOCK_METHOD(void, set_enterprise_test_key, (Attestation::VAType, RSA*)); |
| MOCK_METHOD(void, ThreadMain, (), (override)); |
| MOCK_METHOD(void, OnFinalized, (), (override)); |
| MOCK_METHOD(bool, |
| GetDelegateCredentials, |
| (brillo::Blob*, brillo::Blob*, bool*), |
| (override)); |
| }; |
| |
| } // namespace cryptohome |
| |
| #endif // CRYPTOHOME_MOCK_ATTESTATION_H_ |