summary refs log tree commit diff
path: root/tests/replication/tcp/streams
diff options
context:
space:
mode:
authorAmber Brown <hawkowl@atleastfornow.net>2019-11-28 08:54:07 +1100
committerGitHub <noreply@github.com>2019-11-28 08:54:07 +1100
commit0f87b912aba7e678041632bc9a6d1f7c2d24342c (patch)
tree6381748f864954fb84c63764a1d2c5e8cb5e4e0f /tests/replication/tcp/streams
parentadd etag and count to key backup endpoints (#5858) (diff)
downloadsynapse-0f87b912aba7e678041632bc9a6d1f7c2d24342c.tar.xz
Implementation of MSC2314 (#6176)
Diffstat (limited to 'tests/replication/tcp/streams')
-rw-r--r--tests/replication/tcp/streams/_base.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/tests/replication/tcp/streams/_base.py b/tests/replication/tcp/streams/_base.py
index ce3835ae6a..1d14e77255 100644
--- a/tests/replication/tcp/streams/_base.py
+++ b/tests/replication/tcp/streams/_base.py
@@ -12,6 +12,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+from mock import Mock
+
 from synapse.replication.tcp.commands import ReplicateCommand
 from synapse.replication.tcp.protocol import ClientReplicationStreamProtocol
 from synapse.replication.tcp.resource import ReplicationStreamProtocolFactory
@@ -30,7 +32,9 @@ class BaseStreamTestCase(unittest.HomeserverTestCase):
         server = server_factory.buildProtocol(None)
 
         # build a replication client, with a dummy handler
+        handler_factory = Mock()
         self.test_handler = TestReplicationClientHandler()
+        self.test_handler.factory = handler_factory
         self.client = ClientReplicationStreamProtocol(
             "client", "test", clock, self.test_handler
         )