| // 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, SetU2fVendorMode, (uint8_t), (override)); |
| MOCK_METHOD(uint32_t, |
| SendU2fGenerate, |
| (const U2F_GENERATE_REQ&, U2F_GENERATE_RESP*), |
| (override)); |
| MOCK_METHOD(uint32_t, |
| SendU2fSign, |
| (const U2F_SIGN_REQ&, U2F_SIGN_RESP*), |
| (override)); |
| MOCK_METHOD(uint32_t, |
| SendU2fAttest, |
| (const U2F_ATTEST_REQ&, U2F_ATTEST_RESP*), |
| (override)); |
| MOCK_METHOD(uint32_t, GetG2fCertificate, (std::string*), (override)); |
| }; |
| |
| } // namespace u2f |
| |
| #endif // U2FD_MOCK_TPM_VENDOR_CMD_H_ |