diff --git a/test/data_mock.hpp b/test/data_mock.hpp
index 1fd1fae..a79c562 100644
--- a/test/data_mock.hpp
+++ b/test/data_mock.hpp
@@ -13,6 +13,8 @@
     virtual ~DataHandlerMock() = default;
 
     MOCK_METHOD1(copyFrom, std::vector<std::uint8_t>(std::uint32_t));
+    MOCK_METHOD1(write, bool(const std::vector<std::uint8_t>&));
+    MOCK_METHOD0(read, std::vector<std::uint8_t>());
 };
 
 } // namespace blobs