1 files changed, 14 insertions, 0 deletions

diff --git a/src/data_mock.hpp b/src/data_mock.hpp
index 363eb9e..2740165 100644
--- a/src/data_mock.hpp
+++ b/src/data_mock.hpp
@@ -15,6 +15,20 @@ public:
     ~DataSourceMock() override = default;
 
     MOCK_METHOD(mw::E<int64_t>, getSchemaVersion, (), (const override));
+    MOCK_METHOD(mw::E<std::optional<User>>, userByOpenIDUID,
+                (const std::string& uid), (const override));
+    MOCK_METHOD(mw::E<int64_t>, addUser, (User&& u), (const override));
+    MOCK_METHOD(mw::E<std::optional<User>>, userByID, (const int64_t),
+                (const override));
+    MOCK_METHOD(mw::E<std::optional<User>>, userByName,
+                (const std::string& name), (const override));
+    MOCK_METHOD(mw::E<std::optional<LinkItem>>, itemByID, (int64_t id),
+                (const override));
+    MOCK_METHOD(mw::E<std::vector<LinkItem>>, itemsByParent, (int64_t parent),
+                (const override));
+    MOCK_METHOD(mw::E<std::vector<LinkItem>>, itemsTopLevelByUser,
+                (int64_t user_id), (const override));
+    MOCK_METHOD(mw::E<int64_t>, addLink, (LinkItem&& link), (const override));
 
 protected:
     mw::E<void> setSchemaVersion([[maybe_unused]] int64_t v) const override