diff --git a/tests/rest/test_presence.py b/tests/rest/test_presence.py
index 91d4d1ff6c..99287823e4 100644
--- a/tests/rest/test_presence.py
+++ b/tests/rest/test_presence.py
@@ -51,6 +51,7 @@ class PresenceStateTestCase(unittest.TestCase):
hs = HomeServer("test",
db_pool=None,
http_client=None,
+ datastore=None,
resource_for_client=self.mock_server,
resource_for_federation=self.mock_server,
)
@@ -109,6 +110,7 @@ class PresenceListTestCase(unittest.TestCase):
hs = HomeServer("test",
db_pool=None,
http_client=None,
+ datastore=None,
resource_for_client=self.mock_server,
resource_for_federation=self.mock_server
)
diff --git a/tests/rest/test_profile.py b/tests/rest/test_profile.py
index ff1e92805e..ee47daf4c2 100644
--- a/tests/rest/test_profile.py
+++ b/tests/rest/test_profile.py
@@ -46,6 +46,7 @@ class ProfileTestCase(unittest.TestCase):
resource_for_client=self.mock_server,
federation=Mock(),
replication_layer=Mock(),
+ datastore=None,
)
def _get_user_by_token(token=None):
|