mirror of
https://github.com/element-hq/synapse.git
synced 2026-05-25 06:25:27 +00:00
Erik Johnston
5542419c28
We have to take a slightly different approach here as we can't subclass the native Event type.
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
#
|
|
# This file is licensed under the Affero General Public License (AGPL) version 3.
|
|
#
|
|
# Copyright (C) 2026 Element Creations 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>.
|
|
#
|
|
#
|
|
|
|
from unittest.mock import Mock
|
|
|
|
from synapse.api.room_versions import RoomVersion, RoomVersions
|
|
from synapse.events import EventBase
|
|
from synapse.events.py_protocol import (
|
|
MSC4242Event,
|
|
all_supports_msc4242_state_dag,
|
|
supports_msc4242_state_dag,
|
|
)
|
|
|
|
from tests.test_utils.event_builders import make_test_event
|
|
from tests.unittest import TestCase
|
|
|
|
|
|
def _make_event(room_version: RoomVersion) -> EventBase:
|
|
"""Helper to make an EventBase with the given room version."""
|
|
return make_test_event(
|
|
{
|
|
"sender": "@user:example.com",
|
|
"type": "m.room.message",
|
|
"room_id": "!room:example.com",
|
|
},
|
|
room_version=room_version,
|
|
)
|
|
|
|
|
|
class TestMetaClass(TestCase):
|
|
def test_is_instance(self) -> None:
|
|
"""Test that isinstance checks on EventProtocol raise
|
|
NotImplementedError, but that isinstance checks on EventBase and
|
|
FrozenEvent still work as normal.
|
|
"""
|
|
# EventBase should work as normal
|
|
self.assertFalse(isinstance(object(), EventBase))
|
|
|
|
with self.assertRaises(NotImplementedError):
|
|
isinstance(object(), MSC4242Event)
|
|
|
|
|
|
class SupportsMSC4242StateDagTestCase(TestCase):
|
|
def test_single_event_msc4242(self) -> None:
|
|
"""A single event in an MSC4242 room is recognised."""
|
|
ev = _make_event(RoomVersions.MSC4242v12)
|
|
self.assertTrue(supports_msc4242_state_dag(ev))
|
|
|
|
def test_single_event_non_msc4242(self) -> None:
|
|
"""A single event in a non-MSC4242 room is not recognised."""
|
|
ev = _make_event(RoomVersions.V11)
|
|
self.assertFalse(supports_msc4242_state_dag(ev))
|
|
|
|
def test_sequence_all_msc4242(self) -> None:
|
|
"""A sequence of MSC4242 (event, context) pairs is recognised."""
|
|
pairs = [(_make_event(RoomVersions.MSC4242v12), Mock()) for _ in range(3)]
|
|
self.assertTrue(all_supports_msc4242_state_dag(pairs))
|
|
|
|
def test_sequence_mixed(self) -> None:
|
|
"""A sequence containing any non-MSC4242 event is not recognised."""
|
|
pairs = [
|
|
(_make_event(RoomVersions.MSC4242v12), Mock()),
|
|
(_make_event(RoomVersions.V11), Mock()),
|
|
]
|
|
self.assertFalse(all_supports_msc4242_state_dag(pairs))
|