mirror of
https://github.com/element-hq/synapse.git
synced 2024-11-23 18:15:53 +03:00
23740eaa3d
During the migration the automated script to update the copyright headers accidentally got rid of some of the existing copyright lines. Reinstate them.
413 lines
15 KiB
Python
413 lines
15 KiB
Python
#
|
|
# This file is licensed under the Affero General Public License (AGPL) version 3.
|
|
#
|
|
# Copyright 2021 The Matrix.org Foundation C.I.C.
|
|
# Copyright (C) 2023 New Vector, Ltd
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# See the GNU Affero General Public License for more details:
|
|
# <https://www.gnu.org/licenses/agpl-3.0.html>.
|
|
#
|
|
# Originally licensed under the Apache License, Version 2.0:
|
|
# <http://www.apache.org/licenses/LICENSE-2.0>.
|
|
#
|
|
# [This file includes modifications made by New Vector Limited]
|
|
#
|
|
#
|
|
from typing import Optional
|
|
from unittest.mock import patch
|
|
|
|
from twisted.test.proto_helpers import MemoryReactor
|
|
|
|
from synapse.api.constants import EventContentFields, EventTypes, RoomTypes
|
|
from synapse.config.server import DEFAULT_ROOM_VERSION
|
|
from synapse.rest import admin
|
|
from synapse.rest.client import login, room, room_upgrade_rest_servlet
|
|
from synapse.server import HomeServer
|
|
from synapse.util import Clock
|
|
|
|
from tests import unittest
|
|
from tests.server import FakeChannel
|
|
|
|
|
|
class UpgradeRoomTest(unittest.HomeserverTestCase):
|
|
servlets = [
|
|
admin.register_servlets,
|
|
login.register_servlets,
|
|
room.register_servlets,
|
|
room_upgrade_rest_servlet.register_servlets,
|
|
]
|
|
|
|
def prepare(self, reactor: MemoryReactor, clock: Clock, hs: HomeServer) -> None:
|
|
self.store = hs.get_datastores().main
|
|
|
|
self.creator = self.register_user("creator", "pass")
|
|
self.creator_token = self.login(self.creator, "pass")
|
|
|
|
self.other = self.register_user("user", "pass")
|
|
self.other_token = self.login(self.other, "pass")
|
|
|
|
self.room_id = self.helper.create_room_as(self.creator, tok=self.creator_token)
|
|
self.helper.join(self.room_id, self.other, tok=self.other_token)
|
|
|
|
def _upgrade_room(
|
|
self,
|
|
token: Optional[str] = None,
|
|
room_id: Optional[str] = None,
|
|
expire_cache: bool = True,
|
|
) -> FakeChannel:
|
|
if expire_cache:
|
|
# We don't want a cached response.
|
|
self.reactor.advance(5 * 60 + 1)
|
|
|
|
if room_id is None:
|
|
room_id = self.room_id
|
|
|
|
return self.make_request(
|
|
"POST",
|
|
f"/_matrix/client/r0/rooms/{room_id}/upgrade",
|
|
# This will upgrade a room to the same version, but that's fine.
|
|
content={"new_version": DEFAULT_ROOM_VERSION},
|
|
access_token=token or self.creator_token,
|
|
)
|
|
|
|
def test_upgrade(self) -> None:
|
|
"""
|
|
Upgrading a room should work fine.
|
|
"""
|
|
channel = self._upgrade_room()
|
|
self.assertEqual(200, channel.code, channel.result)
|
|
self.assertIn("replacement_room", channel.json_body)
|
|
|
|
new_room_id = channel.json_body["replacement_room"]
|
|
|
|
# Check that the tombstone event points to the new room.
|
|
tombstone_event = self.get_success(
|
|
self.hs.get_storage_controllers().state.get_current_state_event(
|
|
self.room_id, EventTypes.Tombstone, ""
|
|
)
|
|
)
|
|
assert tombstone_event is not None
|
|
self.assertEqual(new_room_id, tombstone_event.content["replacement_room"])
|
|
|
|
# Check that the new room exists.
|
|
room = self.get_success(self.store.get_room(new_room_id))
|
|
self.assertIsNotNone(room)
|
|
|
|
def test_never_in_room(self) -> None:
|
|
"""
|
|
A user who has never been in the room cannot upgrade the room.
|
|
"""
|
|
# The user isn't in the room.
|
|
roomless = self.register_user("roomless", "pass")
|
|
roomless_token = self.login(roomless, "pass")
|
|
|
|
channel = self._upgrade_room(roomless_token)
|
|
self.assertEqual(403, channel.code, channel.result)
|
|
|
|
def test_left_room(self) -> None:
|
|
"""
|
|
A user who is no longer in the room cannot upgrade the room.
|
|
"""
|
|
# Remove the user from the room.
|
|
self.helper.leave(self.room_id, self.creator, tok=self.creator_token)
|
|
|
|
channel = self._upgrade_room(self.creator_token)
|
|
self.assertEqual(403, channel.code, channel.result)
|
|
|
|
def test_power_levels(self) -> None:
|
|
"""
|
|
Another user can upgrade the room if their power level is increased.
|
|
"""
|
|
# The other user doesn't have the proper power level.
|
|
channel = self._upgrade_room(self.other_token)
|
|
self.assertEqual(403, channel.code, channel.result)
|
|
|
|
# Increase the power levels so that this user can upgrade.
|
|
power_levels = self.helper.get_state(
|
|
self.room_id,
|
|
"m.room.power_levels",
|
|
tok=self.creator_token,
|
|
)
|
|
power_levels["users"][self.other] = 100
|
|
self.helper.send_state(
|
|
self.room_id,
|
|
"m.room.power_levels",
|
|
body=power_levels,
|
|
tok=self.creator_token,
|
|
)
|
|
|
|
# The upgrade should succeed!
|
|
channel = self._upgrade_room(self.other_token)
|
|
self.assertEqual(200, channel.code, channel.result)
|
|
|
|
def test_power_levels_user_default(self) -> None:
|
|
"""
|
|
Another user can upgrade the room if the default power level for users is increased.
|
|
"""
|
|
# The other user doesn't have the proper power level.
|
|
channel = self._upgrade_room(self.other_token)
|
|
self.assertEqual(403, channel.code, channel.result)
|
|
|
|
# Increase the power levels so that this user can upgrade.
|
|
power_levels = self.helper.get_state(
|
|
self.room_id,
|
|
"m.room.power_levels",
|
|
tok=self.creator_token,
|
|
)
|
|
power_levels["users_default"] = 100
|
|
self.helper.send_state(
|
|
self.room_id,
|
|
"m.room.power_levels",
|
|
body=power_levels,
|
|
tok=self.creator_token,
|
|
)
|
|
|
|
# The upgrade should succeed!
|
|
channel = self._upgrade_room(self.other_token)
|
|
self.assertEqual(200, channel.code, channel.result)
|
|
|
|
def test_power_levels_tombstone(self) -> None:
|
|
"""
|
|
Another user can upgrade the room if they can send the tombstone event.
|
|
"""
|
|
# The other user doesn't have the proper power level.
|
|
channel = self._upgrade_room(self.other_token)
|
|
self.assertEqual(403, channel.code, channel.result)
|
|
|
|
# Increase the power levels so that this user can upgrade.
|
|
power_levels = self.helper.get_state(
|
|
self.room_id,
|
|
"m.room.power_levels",
|
|
tok=self.creator_token,
|
|
)
|
|
power_levels["events"]["m.room.tombstone"] = 0
|
|
self.helper.send_state(
|
|
self.room_id,
|
|
"m.room.power_levels",
|
|
body=power_levels,
|
|
tok=self.creator_token,
|
|
)
|
|
|
|
# The upgrade should succeed!
|
|
channel = self._upgrade_room(self.other_token)
|
|
self.assertEqual(200, channel.code, channel.result)
|
|
|
|
power_levels = self.helper.get_state(
|
|
self.room_id,
|
|
"m.room.power_levels",
|
|
tok=self.creator_token,
|
|
)
|
|
self.assertNotIn(self.other, power_levels["users"])
|
|
|
|
def test_stringy_power_levels(self) -> None:
|
|
"""The room upgrade converts stringy power levels to proper integers."""
|
|
# Create a room on room version < 10.
|
|
room_id = self.helper.create_room_as(
|
|
self.creator, tok=self.creator_token, room_version="9"
|
|
)
|
|
self.helper.join(room_id, self.other, tok=self.other_token)
|
|
|
|
# Retrieve the room's current power levels.
|
|
power_levels = self.helper.get_state(
|
|
room_id,
|
|
"m.room.power_levels",
|
|
tok=self.creator_token,
|
|
)
|
|
|
|
# Set creator's power level to the string "100" instead of the integer `100`.
|
|
power_levels["users"][self.creator] = "100"
|
|
|
|
# Synapse refuses to accept new stringy power level events. Bypass this by
|
|
# neutering the validation.
|
|
with patch("synapse.events.validator.jsonschema.validate"):
|
|
# Note: https://github.com/matrix-org/matrix-spec/issues/853 plans to forbid
|
|
# string power levels in new rooms. For this test to have a clean
|
|
# conscience, we ought to ensure it's upgrading from a sufficiently old
|
|
# version of room.
|
|
self.helper.send_state(
|
|
room_id,
|
|
"m.room.power_levels",
|
|
body=power_levels,
|
|
tok=self.creator_token,
|
|
)
|
|
|
|
# Upgrade the room. Check the homeserver reports success.
|
|
channel = self._upgrade_room(room_id=room_id)
|
|
self.assertEqual(200, channel.code, channel.result)
|
|
|
|
# Extract the new room ID.
|
|
new_room_id = channel.json_body["replacement_room"]
|
|
|
|
# Fetch the new room's power level event.
|
|
new_power_levels = self.helper.get_state(
|
|
new_room_id,
|
|
"m.room.power_levels",
|
|
tok=self.creator_token,
|
|
)
|
|
|
|
# We should now have an integer power level.
|
|
self.assertEqual(new_power_levels["users"][self.creator], 100, new_power_levels)
|
|
|
|
def test_events_field_missing(self) -> None:
|
|
"""Regression test for https://github.com/matrix-org/synapse/issues/16715."""
|
|
# Create a new room.
|
|
room_id = self.helper.create_room_as(
|
|
self.creator, tok=self.creator_token, room_version="10"
|
|
)
|
|
self.helper.join(room_id, self.other, tok=self.other_token)
|
|
|
|
# Retrieve the room's current power levels.
|
|
power_levels = self.helper.get_state(
|
|
room_id,
|
|
"m.room.power_levels",
|
|
tok=self.creator_token,
|
|
)
|
|
|
|
# Remove the events field and re-set the power levels.
|
|
del power_levels["events"]
|
|
self.helper.send_state(
|
|
room_id,
|
|
"m.room.power_levels",
|
|
body=power_levels,
|
|
tok=self.creator_token,
|
|
)
|
|
|
|
# Upgrade the room. Check the homeserver reports success.
|
|
channel = self._upgrade_room(room_id=room_id)
|
|
self.assertEqual(200, channel.code, channel.result)
|
|
|
|
def test_space(self) -> None:
|
|
"""Test upgrading a space."""
|
|
|
|
# Create a space.
|
|
space_id = self.helper.create_room_as(
|
|
self.creator,
|
|
tok=self.creator_token,
|
|
extra_content={
|
|
"creation_content": {EventContentFields.ROOM_TYPE: RoomTypes.SPACE}
|
|
},
|
|
)
|
|
|
|
# Add the room as a child room.
|
|
self.helper.send_state(
|
|
space_id,
|
|
event_type=EventTypes.SpaceChild,
|
|
body={"via": [self.hs.hostname]},
|
|
tok=self.creator_token,
|
|
state_key=self.room_id,
|
|
)
|
|
|
|
# Also add a room that was removed.
|
|
old_room_id = "!notaroom:" + self.hs.hostname
|
|
self.helper.send_state(
|
|
space_id,
|
|
event_type=EventTypes.SpaceChild,
|
|
body={},
|
|
tok=self.creator_token,
|
|
state_key=old_room_id,
|
|
)
|
|
|
|
# Upgrade the room!
|
|
channel = self._upgrade_room(room_id=space_id)
|
|
self.assertEqual(200, channel.code, channel.result)
|
|
self.assertIn("replacement_room", channel.json_body)
|
|
|
|
new_space_id = channel.json_body["replacement_room"]
|
|
|
|
state_ids = self.get_success(
|
|
self.store.get_partial_current_state_ids(new_space_id)
|
|
)
|
|
|
|
# Ensure the new room is still a space.
|
|
create_event = self.get_success(
|
|
self.store.get_event(state_ids[(EventTypes.Create, "")])
|
|
)
|
|
self.assertEqual(
|
|
create_event.content.get(EventContentFields.ROOM_TYPE), RoomTypes.SPACE
|
|
)
|
|
|
|
# The child link should have been copied over.
|
|
self.assertIn((EventTypes.SpaceChild, self.room_id), state_ids)
|
|
# The child that was removed should not be copied over.
|
|
self.assertNotIn((EventTypes.SpaceChild, old_room_id), state_ids)
|
|
|
|
def test_custom_room_type(self) -> None:
|
|
"""Test upgrading a room that has a custom room type set."""
|
|
test_room_type = "com.example.my_custom_room_type"
|
|
|
|
# Create a room with a custom room type.
|
|
room_id = self.helper.create_room_as(
|
|
self.creator,
|
|
tok=self.creator_token,
|
|
extra_content={
|
|
"creation_content": {EventContentFields.ROOM_TYPE: test_room_type}
|
|
},
|
|
)
|
|
|
|
# Upgrade the room!
|
|
channel = self._upgrade_room(room_id=room_id)
|
|
self.assertEqual(200, channel.code, channel.result)
|
|
self.assertIn("replacement_room", channel.json_body)
|
|
|
|
new_room_id = channel.json_body["replacement_room"]
|
|
|
|
state_ids = self.get_success(
|
|
self.store.get_partial_current_state_ids(new_room_id)
|
|
)
|
|
|
|
# Ensure the new room is the same type as the old room.
|
|
create_event = self.get_success(
|
|
self.store.get_event(state_ids[(EventTypes.Create, "")])
|
|
)
|
|
self.assertEqual(
|
|
create_event.content.get(EventContentFields.ROOM_TYPE), test_room_type
|
|
)
|
|
|
|
def test_second_upgrade_from_same_user(self) -> None:
|
|
"""A second room upgrade from the same user is deduplicated."""
|
|
channel1 = self._upgrade_room()
|
|
self.assertEqual(200, channel1.code, channel1.result)
|
|
|
|
channel2 = self._upgrade_room(expire_cache=False)
|
|
self.assertEqual(200, channel2.code, channel2.result)
|
|
|
|
self.assertEqual(
|
|
channel1.json_body["replacement_room"],
|
|
channel2.json_body["replacement_room"],
|
|
)
|
|
|
|
def test_second_upgrade_after_delay(self) -> None:
|
|
"""A second room upgrade is not deduplicated after some time has passed."""
|
|
channel1 = self._upgrade_room()
|
|
self.assertEqual(200, channel1.code, channel1.result)
|
|
|
|
channel2 = self._upgrade_room(expire_cache=True)
|
|
self.assertEqual(200, channel2.code, channel2.result)
|
|
|
|
self.assertNotEqual(
|
|
channel1.json_body["replacement_room"],
|
|
channel2.json_body["replacement_room"],
|
|
)
|
|
|
|
def test_second_upgrade_from_different_user(self) -> None:
|
|
"""A second room upgrade from a different user is blocked."""
|
|
channel = self._upgrade_room()
|
|
self.assertEqual(200, channel.code, channel.result)
|
|
|
|
channel = self._upgrade_room(self.other_token, expire_cache=False)
|
|
self.assertEqual(400, channel.code, channel.result)
|
|
|
|
def test_first_upgrade_does_not_block_second(self) -> None:
|
|
"""A second room upgrade is not blocked when a previous upgrade attempt was not
|
|
allowed.
|
|
"""
|
|
channel = self._upgrade_room(self.other_token)
|
|
self.assertEqual(403, channel.code, channel.result)
|
|
|
|
channel = self._upgrade_room(expire_cache=False)
|
|
self.assertEqual(200, channel.code, channel.result)
|