| // Copyright 2017 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 HAMMERD_MOCK_PAIR_UTILS_H_ |
| #define HAMMERD_MOCK_PAIR_UTILS_H_ |
| |
| #include <gmock/gmock.h> |
| |
| #include "hammerd/pair_utils.h" |
| |
| namespace hammerd { |
| |
| // Mock internal method GenerateChallenge() to test PairManager itself. |
| class MockPairManager : public PairManager { |
| public: |
| MockPairManager() = default; |
| |
| MOCK_METHOD2(GenerateChallenge, |
| void(PairChallengeRequest* request, uint8_t* private_key)); |
| }; |
| |
| // Mock public method PairChallenge() used to inject into HammerUpdater. |
| class MockPairManagerInterface : public PairManagerInterface { |
| public: |
| MockPairManagerInterface() = default; |
| MOCK_METHOD2(PairChallenge, |
| ChallengeStatus(FirmwareUpdaterInterface* fw_updater, |
| DBusWrapperInterface* dbus_wrapper)); |
| }; |
| |
| } // namespace hammerd |
| #endif // HAMMERD_MOCK_PAIR_UTILS_H_ |