|
12 | 12 | # See the License for the specific language governing permissions and
|
13 | 13 | # limitations under the License.
|
14 | 14 |
|
| 15 | +from typing import Optional |
| 16 | + |
| 17 | +from synapse.api.constants import EventTypes, JoinRules |
15 | 18 | from synapse.api.errors import Codes, ResourceLimitError
|
16 | 19 | from synapse.api.filtering import DEFAULT_FILTER_COLLECTION
|
| 20 | +from synapse.api.room_versions import RoomVersions |
17 | 21 | from synapse.handlers.sync import SyncConfig
|
| 22 | +from synapse.rest import admin |
| 23 | +from synapse.rest.client import knock, login, room |
| 24 | +from synapse.server import HomeServer |
18 | 25 | from synapse.types import UserID, create_requester
|
19 | 26 |
|
20 | 27 | import tests.unittest
|
|
24 | 31 | class SyncTestCase(tests.unittest.HomeserverTestCase):
|
25 | 32 | """Tests Sync Handler."""
|
26 | 33 |
|
27 |
| - def prepare(self, reactor, clock, hs): |
28 |
| - self.hs = hs |
| 34 | + servlets = [ |
| 35 | + admin.register_servlets, |
| 36 | + knock.register_servlets, |
| 37 | + login.register_servlets, |
| 38 | + room.register_servlets, |
| 39 | + ] |
| 40 | + |
| 41 | + def prepare(self, reactor, clock, hs: HomeServer): |
29 | 42 | self.sync_handler = self.hs.get_sync_handler()
|
30 | 43 | self.store = self.hs.get_datastore()
|
31 | 44 |
|
@@ -68,12 +81,124 @@ def test_wait_for_sync_for_user_auth_blocking(self):
|
68 | 81 | )
|
69 | 82 | self.assertEquals(e.value.errcode, Codes.RESOURCE_LIMIT_EXCEEDED)
|
70 | 83 |
|
| 84 | + def test_unknown_room_version(self): |
| 85 | + """ |
| 86 | + A room with an unknown room version should not break sync (and should be excluded). |
| 87 | + """ |
| 88 | + inviter = self.register_user("creator", "pass", admin=True) |
| 89 | + inviter_tok = self.login("@creator:test", "pass") |
| 90 | + |
| 91 | + user = self.register_user("user", "pass") |
| 92 | + tok = self.login("user", "pass") |
| 93 | + |
| 94 | + # Do an initial sync on a different device. |
| 95 | + requester = create_requester(user) |
| 96 | + initial_result = self.get_success( |
| 97 | + self.sync_handler.wait_for_sync_for_user( |
| 98 | + requester, sync_config=generate_sync_config(user, device_id="dev") |
| 99 | + ) |
| 100 | + ) |
| 101 | + |
| 102 | + # Create a room as the user. |
| 103 | + joined_room = self.helper.create_room_as(user, tok=tok) |
| 104 | + |
| 105 | + # Invite the user to the room as someone else. |
| 106 | + invite_room = self.helper.create_room_as(inviter, tok=inviter_tok) |
| 107 | + self.helper.invite(invite_room, targ=user, tok=inviter_tok) |
| 108 | + |
| 109 | + knock_room = self.helper.create_room_as( |
| 110 | + inviter, room_version=RoomVersions.V7.identifier, tok=inviter_tok |
| 111 | + ) |
| 112 | + self.helper.send_state( |
| 113 | + knock_room, |
| 114 | + EventTypes.JoinRules, |
| 115 | + {"join_rule": JoinRules.KNOCK}, |
| 116 | + tok=inviter_tok, |
| 117 | + ) |
| 118 | + channel = self.make_request( |
| 119 | + "POST", |
| 120 | + "/_matrix/client/r0/knock/%s" % (knock_room,), |
| 121 | + b"{}", |
| 122 | + tok, |
| 123 | + ) |
| 124 | + self.assertEquals(200, channel.code, channel.result) |
| 125 | + |
| 126 | + # The rooms should appear in the sync response. |
| 127 | + result = self.get_success( |
| 128 | + self.sync_handler.wait_for_sync_for_user( |
| 129 | + requester, sync_config=generate_sync_config(user) |
| 130 | + ) |
| 131 | + ) |
| 132 | + self.assertIn(joined_room, [r.room_id for r in result.joined]) |
| 133 | + self.assertIn(invite_room, [r.room_id for r in result.invited]) |
| 134 | + self.assertIn(knock_room, [r.room_id for r in result.knocked]) |
| 135 | + |
| 136 | + # Test a incremental sync (by providing a since_token). |
| 137 | + result = self.get_success( |
| 138 | + self.sync_handler.wait_for_sync_for_user( |
| 139 | + requester, |
| 140 | + sync_config=generate_sync_config(user, device_id="dev"), |
| 141 | + since_token=initial_result.next_batch, |
| 142 | + ) |
| 143 | + ) |
| 144 | + self.assertIn(joined_room, [r.room_id for r in result.joined]) |
| 145 | + self.assertIn(invite_room, [r.room_id for r in result.invited]) |
| 146 | + self.assertIn(knock_room, [r.room_id for r in result.knocked]) |
| 147 | + |
| 148 | + # Poke the database and update the room version to an unknown one. |
| 149 | + for room_id in (joined_room, invite_room, knock_room): |
| 150 | + self.get_success( |
| 151 | + self.hs.get_datastores().main.db_pool.simple_update( |
| 152 | + "rooms", |
| 153 | + keyvalues={"room_id": room_id}, |
| 154 | + updatevalues={"room_version": "unknown-room-version"}, |
| 155 | + desc="updated-room-version", |
| 156 | + ) |
| 157 | + ) |
| 158 | + |
| 159 | + # Blow away caches (supported room versions can only change due to a restart). |
| 160 | + self.get_success( |
| 161 | + self.store.get_rooms_for_user_with_stream_ordering.invalidate_all() |
| 162 | + ) |
| 163 | + self.store._get_event_cache.clear() |
| 164 | + |
| 165 | + # The rooms should be excluded from the sync response. |
| 166 | + # Get a new request key. |
| 167 | + result = self.get_success( |
| 168 | + self.sync_handler.wait_for_sync_for_user( |
| 169 | + requester, sync_config=generate_sync_config(user) |
| 170 | + ) |
| 171 | + ) |
| 172 | + self.assertNotIn(joined_room, [r.room_id for r in result.joined]) |
| 173 | + self.assertNotIn(invite_room, [r.room_id for r in result.invited]) |
| 174 | + self.assertNotIn(knock_room, [r.room_id for r in result.knocked]) |
| 175 | + |
| 176 | + # The rooms should also not be in an incremental sync. |
| 177 | + result = self.get_success( |
| 178 | + self.sync_handler.wait_for_sync_for_user( |
| 179 | + requester, |
| 180 | + sync_config=generate_sync_config(user, device_id="dev"), |
| 181 | + since_token=initial_result.next_batch, |
| 182 | + ) |
| 183 | + ) |
| 184 | + self.assertNotIn(joined_room, [r.room_id for r in result.joined]) |
| 185 | + self.assertNotIn(invite_room, [r.room_id for r in result.invited]) |
| 186 | + self.assertNotIn(knock_room, [r.room_id for r in result.knocked]) |
| 187 | + |
| 188 | + |
| 189 | +_request_key = 0 |
| 190 | + |
71 | 191 |
|
72 |
| -def generate_sync_config(user_id: str) -> SyncConfig: |
| 192 | +def generate_sync_config( |
| 193 | + user_id: str, device_id: Optional[str] = "device_id" |
| 194 | +) -> SyncConfig: |
| 195 | + """Generate a sync config (with a unique request key).""" |
| 196 | + global _request_key |
| 197 | + _request_key += 1 |
73 | 198 | return SyncConfig(
|
74 |
| - user=UserID(user_id.split(":")[0][1:], user_id.split(":")[1]), |
| 199 | + user=UserID.from_string(user_id), |
75 | 200 | filter_collection=DEFAULT_FILTER_COLLECTION,
|
76 | 201 | is_guest=False,
|
77 |
| - request_key="request_key", |
78 |
| - device_id="device_id", |
| 202 | + request_key=("request_key", _request_key), |
| 203 | + device_id=device_id, |
79 | 204 | )
|
0 commit comments