| Patrick Venture | 54c3b53 | 2018-08-01 11:45:49 -0700 | [diff] [blame] | 1 | #pragma once |
| 2 | |
| 3 | #include <gmock/gmock.h> |
| Patrick Venture | 79e131f | 2018-08-01 13:34:35 -0700 | [diff] [blame] | 4 | #include <vector> |
| Patrick Venture | 54c3b53 | 2018-08-01 11:45:49 -0700 | [diff] [blame] | 5 | |
| 6 | #include "flash-ipmi.hpp" |
| 7 | |
| 8 | class UpdaterMock : public UpdateInterface |
| 9 | { |
| 10 | public: |
| 11 | virtual ~UpdaterMock() = default; |
| 12 | |
| 13 | MOCK_METHOD1(start, bool(uint32_t)); |
| Patrick Venture | 79e131f | 2018-08-01 13:34:35 -0700 | [diff] [blame] | 14 | MOCK_METHOD2(flashData, bool(uint32_t, const std::vector<uint8_t>&)); |
| Patrick Venture | 2c1205d | 2018-08-03 10:23:14 -0700 | [diff] [blame] | 15 | MOCK_METHOD0(flashFinish, bool()); |
| Patrick Venture | 8d9f732 | 2018-08-03 10:39:13 -0700 | [diff] [blame^] | 16 | MOCK_METHOD1(startHash, bool(uint32_t)); |
| Patrick Venture | 54c3b53 | 2018-08-01 11:45:49 -0700 | [diff] [blame] | 17 | }; |