1 #pragma once
2 
3 #include "updater.hpp"
4 
5 #include <cstdint>
6 #include <string>
7 #include <vector>
8 
9 #include <gmock/gmock.h>
10 
11 namespace host_tool
12 {
13 
14 class UpdateHandlerMock : public UpdateHandlerInterface
15 {
16   public:
17     MOCK_METHOD(bool, checkAvailable, (const std::string&), (override));
18     MOCK_METHOD(std::vector<uint8_t>, readVersion, (const std::string&),
19                 (override));
20     MOCK_METHOD(void, sendFile, (const std::string&, const std::string&),
21                 (override));
22     MOCK_METHOD(bool, verifyFile, (const std::string&, bool), (override));
23     MOCK_METHOD(void, cleanArtifacts, (), (override));
24 };
25 
26 } // namespace host_tool
27