diff options
Diffstat (limited to 'tests/storage/test_stream.py')
| -rw-r--r-- | tests/storage/test_stream.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/storage/test_stream.py b/tests/storage/test_stream.py index 52e41cdab4..78663a53fe 100644 --- a/tests/storage/test_stream.py +++ b/tests/storage/test_stream.py @@ -110,7 +110,9 @@ class PaginationTestCase(HomeserverTestCase): def _filter_messages(self, filter: JsonDict) -> List[EventBase]: """Make a request to /messages with a filter, returns the chunk of events.""" - from_token = self.hs.get_event_sources().get_current_token_for_pagination() + from_token = self.get_success( + self.hs.get_event_sources().get_current_token_for_pagination(self.room_id) + ) events, next_key = self.get_success( self.hs.get_datastores().main.paginate_room_events( |