tools/progress: Add finish and abort functions
We have external progress handlers which would like to implement these
functions, in order to update the screen upon completion or error.
Change-Id: I2df4654d5e2092407a83d2c8bbb020dcce84ac50
Signed-off-by: William A. Kennington III <wak@google.com>
diff --git a/tools/test/progress_mock.hpp b/tools/test/progress_mock.hpp
index 80c0af1..a43d9b3 100644
--- a/tools/test/progress_mock.hpp
+++ b/tools/test/progress_mock.hpp
@@ -12,8 +12,10 @@
class ProgressMock : public ProgressInterface
{
public:
- MOCK_METHOD1(updateProgress, void(std::int64_t));
- MOCK_METHOD1(start, void(std::int64_t));
+ MOCK_METHOD(void, updateProgress, (std::int64_t), (override));
+ MOCK_METHOD(void, start, (std::int64_t), (override));
+ MOCK_METHOD(void, finish, (), (override));
+ MOCK_METHOD(void, abort, (), (override));
};
} // namespace host_tool