From 6fe244b20926b3be396423dc966572f7d7fd37e7 Mon Sep 17 00:00:00 2001
From: gvolfing <gabor.volfinger@memgraph.io>
Date: Thu, 19 Jan 2023 17:16:29 +0100
Subject: [PATCH] Add missing MOCK_METHODS to MockedRequestRouter

---
 tests/unit/mock_helpers.hpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/unit/mock_helpers.hpp b/tests/unit/mock_helpers.hpp
index c522b8602..89ae9bb7d 100644
--- a/tests/unit/mock_helpers.hpp
+++ b/tests/unit/mock_helpers.hpp
@@ -42,6 +42,8 @@ class MockedRequestRouter : public RequestRouterInterface {
   MOCK_METHOD(std::optional<storage::v3::LabelId>, MaybeNameToLabel, (const std::string &), (const));
   MOCK_METHOD(bool, IsPrimaryLabel, (storage::v3::LabelId), (const));
   MOCK_METHOD(bool, IsPrimaryKey, (storage::v3::LabelId, storage::v3::PropertyId), (const));
+  MOCK_METHOD(std::optional<std::pair<uint64_t, uint64_t>>, AllocateInitialEdgeIds, (io::Address));
+  MOCK_METHOD(void, InstallSimulatorTicker, (std::function<bool()>));
 };
 
 class MockedLogicalOperator : public plan::LogicalOperator {