| // Copyright 2019 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 U2FD_MOCK_TPM_VENDOR_CMD_H_ |
| #define U2FD_MOCK_TPM_VENDOR_CMD_H_ |
| |
| #include "u2fd/tpm_vendor_cmd.h" |
| |
| #include <string> |
| |
| #include <gmock/gmock.h> |
| #include <trunks/cr50_headers/u2f.h> |
| |
| namespace u2f { |
| |
| class MockTpmVendorCommandProxy : public TpmVendorCommandProxy { |
| public: |
| MOCK_METHOD(uint32_t, |
| SendU2fGenerate, |
| (const struct u2f_generate_req&, struct u2f_generate_resp*), |
| (override)); |
| MOCK_METHOD(uint32_t, |
| SendU2fGenerate, |
| (const struct u2f_generate_req&, |
| struct u2f_generate_versioned_resp*), |
| (override)); |
| MOCK_METHOD(uint32_t, |
| SendU2fSign, |
| (const struct u2f_sign_req&, struct u2f_sign_resp*), |
| (override)); |
| MOCK_METHOD(uint32_t, |
| SendU2fSign, |
| (const struct u2f_sign_versioned_req&, struct u2f_sign_resp*), |
| (override)); |
| MOCK_METHOD(uint32_t, |
| SendU2fAttest, |
| (const struct u2f_attest_req&, struct u2f_attest_resp*), |
| (override)); |
| MOCK_METHOD(uint32_t, GetG2fCertificate, (std::string*), (override)); |
| }; |
| |
| } // namespace u2f |
| |
| #endif // U2FD_MOCK_TPM_VENDOR_CMD_H_ |