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