diff options
| author | Erik Johnston <erik@matrix.org> | 2020-05-01 17:19:56 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-05-01 17:19:56 +0100 |
| commit | 0e719f23981b8294df66ba7f38b8c7cc99fad228 (patch) | |
| tree | 42d9aa97954cdbea46b0969bceefd88d2953a623 /tests/replication/tcp/streams/_base.py | |
| parent | Use `stream.current_token()` and remove `stream_positions()` (#7172) (diff) | |
| download | synapse-0e719f23981b8294df66ba7f38b8c7cc99fad228.tar.xz | |
Thread through instance name to replication client. (#7369)
For in memory streams when fetching updates on workers we need to query the source of the stream, which currently is hard coded to be master. This PR threads through the source instance we received via `POSITION` through to the update function in each stream, which can then be passed to the replication client for in memory streams.
Diffstat (limited to 'tests/replication/tcp/streams/_base.py')
| -rw-r--r-- | tests/replication/tcp/streams/_base.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/replication/tcp/streams/_base.py b/tests/replication/tcp/streams/_base.py index 8c104f8d1d..7b56d2028d 100644 --- a/tests/replication/tcp/streams/_base.py +++ b/tests/replication/tcp/streams/_base.py @@ -183,8 +183,8 @@ class TestReplicationDataHandler(GenericWorkerReplicationHandler): # list of received (stream_name, token, row) tuples self.received_rdata_rows = [] # type: List[Tuple[str, int, Any]] - async def on_rdata(self, stream_name, token, rows): - await super().on_rdata(stream_name, token, rows) + async def on_rdata(self, stream_name, instance_name, token, rows): + await super().on_rdata(stream_name, instance_name, token, rows) for r in rows: self.received_rdata_rows.append((stream_name, token, r)) |