1380832ccSPatrick Venture #pragma once 2380832ccSPatrick Venture 3380832ccSPatrick Venture #include "updater.hpp" 4380832ccSPatrick Venture 5328f520fSJie Yang #include <cstdint> 6380832ccSPatrick Venture #include <string> 7328f520fSJie Yang #include <vector> 8380832ccSPatrick Venture 9380832ccSPatrick Venture #include <gmock/gmock.h> 10380832ccSPatrick Venture 11380832ccSPatrick Venture namespace host_tool 12380832ccSPatrick Venture { 13380832ccSPatrick Venture 141f09d414SPatrick Venture class UpdateHandlerMock : public UpdateHandlerInterface 15380832ccSPatrick Venture { 16380832ccSPatrick Venture public: 17*f450486fSWilly Tu MOCK_METHOD(bool, checkAvailable, (const std::string&), (override)); 18*f450486fSWilly Tu MOCK_METHOD(std::vector<uint8_t>, readVersion, (const std::string&), 19*f450486fSWilly Tu (override)); 20*f450486fSWilly Tu MOCK_METHOD(void, sendFile, (const std::string&, const std::string&), 21*f450486fSWilly Tu (override)); 22*f450486fSWilly Tu MOCK_METHOD(bool, verifyFile, (const std::string&, bool), (override)); 23*f450486fSWilly Tu MOCK_METHOD(void, cleanArtifacts, (), (override)); 24380832ccSPatrick Venture }; 25380832ccSPatrick Venture 26380832ccSPatrick Venture } // namespace host_tool 27