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_METHOD1(checkAvailable, bool(const std::string&)); 18 MOCK_METHOD1(readVersion, std::vector<uint8_t>(const std::string&)); 19 MOCK_METHOD2(sendFile, void(const std::string&, const std::string&)); 20 MOCK_METHOD2(verifyFile, bool(const std::string&, bool)); 21 MOCK_METHOD0(cleanArtifacts, void()); 22 }; 23 24 } // namespace host_tool 25