Fix unit tests

matrix-org · Apr 22, 2020 · f9b66d9 · f9b66d9
1 parent 9486faa
commit f9b66d9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/replication/slave/storage/_base.py b/tests/replication/slave/storage/_base.py
@@ -57,6 +57,8 @@ def prepare(self, reactor, clock, hs):
         # We now do some gut wrenching so that we have a client that is based
         # off of the slave store rather than the main store.
         self.replication_handler = ReplicationCommandHandler(self.hs)
+        self.replication_handler._store = self.slaved_store
+
         self.replication_handler._replication_data_handler = ReplicationDataHandler(
             self.slaved_store
         )