mirror of
https://mau.dev/maunium/synapse.git
synced 2024-10-01 01:36:05 -04:00
Replace assertEquals and friends with non-deprecated versions. (#12092)
This commit is contained in:
parent
ab3ef49059
commit
02d708568b
1
changelog.d/12092.misc
Normal file
1
changelog.d/12092.misc
Normal file
@ -0,0 +1 @@
|
|||||||
|
User `assertEqual` instead of the deprecated `assertEquals` in test code.
|
@ -71,7 +71,7 @@ class AuthTestCase(unittest.HomeserverTestCase):
|
|||||||
request.args[b"access_token"] = [self.test_token]
|
request.args[b"access_token"] = [self.test_token]
|
||||||
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
||||||
requester = self.get_success(self.auth.get_user_by_req(request))
|
requester = self.get_success(self.auth.get_user_by_req(request))
|
||||||
self.assertEquals(requester.user.to_string(), self.test_user)
|
self.assertEqual(requester.user.to_string(), self.test_user)
|
||||||
|
|
||||||
def test_get_user_by_req_user_bad_token(self):
|
def test_get_user_by_req_user_bad_token(self):
|
||||||
self.store.get_user_by_access_token = simple_async_mock(None)
|
self.store.get_user_by_access_token = simple_async_mock(None)
|
||||||
@ -109,7 +109,7 @@ class AuthTestCase(unittest.HomeserverTestCase):
|
|||||||
request.args[b"access_token"] = [self.test_token]
|
request.args[b"access_token"] = [self.test_token]
|
||||||
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
||||||
requester = self.get_success(self.auth.get_user_by_req(request))
|
requester = self.get_success(self.auth.get_user_by_req(request))
|
||||||
self.assertEquals(requester.user.to_string(), self.test_user)
|
self.assertEqual(requester.user.to_string(), self.test_user)
|
||||||
|
|
||||||
def test_get_user_by_req_appservice_valid_token_good_ip(self):
|
def test_get_user_by_req_appservice_valid_token_good_ip(self):
|
||||||
from netaddr import IPSet
|
from netaddr import IPSet
|
||||||
@ -128,7 +128,7 @@ class AuthTestCase(unittest.HomeserverTestCase):
|
|||||||
request.args[b"access_token"] = [self.test_token]
|
request.args[b"access_token"] = [self.test_token]
|
||||||
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
||||||
requester = self.get_success(self.auth.get_user_by_req(request))
|
requester = self.get_success(self.auth.get_user_by_req(request))
|
||||||
self.assertEquals(requester.user.to_string(), self.test_user)
|
self.assertEqual(requester.user.to_string(), self.test_user)
|
||||||
|
|
||||||
def test_get_user_by_req_appservice_valid_token_bad_ip(self):
|
def test_get_user_by_req_appservice_valid_token_bad_ip(self):
|
||||||
from netaddr import IPSet
|
from netaddr import IPSet
|
||||||
@ -195,7 +195,7 @@ class AuthTestCase(unittest.HomeserverTestCase):
|
|||||||
request.args[b"user_id"] = [masquerading_user_id]
|
request.args[b"user_id"] = [masquerading_user_id]
|
||||||
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
||||||
requester = self.get_success(self.auth.get_user_by_req(request))
|
requester = self.get_success(self.auth.get_user_by_req(request))
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
requester.user.to_string(), masquerading_user_id.decode("utf8")
|
requester.user.to_string(), masquerading_user_id.decode("utf8")
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -242,10 +242,10 @@ class AuthTestCase(unittest.HomeserverTestCase):
|
|||||||
request.args[b"org.matrix.msc3202.device_id"] = [masquerading_device_id]
|
request.args[b"org.matrix.msc3202.device_id"] = [masquerading_device_id]
|
||||||
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
||||||
requester = self.get_success(self.auth.get_user_by_req(request))
|
requester = self.get_success(self.auth.get_user_by_req(request))
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
requester.user.to_string(), masquerading_user_id.decode("utf8")
|
requester.user.to_string(), masquerading_user_id.decode("utf8")
|
||||||
)
|
)
|
||||||
self.assertEquals(requester.device_id, masquerading_device_id.decode("utf8"))
|
self.assertEqual(requester.device_id, masquerading_device_id.decode("utf8"))
|
||||||
|
|
||||||
@override_config({"experimental_features": {"msc3202_device_masquerading": True}})
|
@override_config({"experimental_features": {"msc3202_device_masquerading": True}})
|
||||||
def test_get_user_by_req_appservice_valid_token_invalid_device_id(self):
|
def test_get_user_by_req_appservice_valid_token_invalid_device_id(self):
|
||||||
@ -275,8 +275,8 @@ class AuthTestCase(unittest.HomeserverTestCase):
|
|||||||
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
||||||
|
|
||||||
failure = self.get_failure(self.auth.get_user_by_req(request), AuthError)
|
failure = self.get_failure(self.auth.get_user_by_req(request), AuthError)
|
||||||
self.assertEquals(failure.value.code, 400)
|
self.assertEqual(failure.value.code, 400)
|
||||||
self.assertEquals(failure.value.errcode, Codes.EXCLUSIVE)
|
self.assertEqual(failure.value.errcode, Codes.EXCLUSIVE)
|
||||||
|
|
||||||
def test_get_user_by_req__puppeted_token__not_tracking_puppeted_mau(self):
|
def test_get_user_by_req__puppeted_token__not_tracking_puppeted_mau(self):
|
||||||
self.store.get_user_by_access_token = simple_async_mock(
|
self.store.get_user_by_access_token = simple_async_mock(
|
||||||
@ -309,7 +309,7 @@ class AuthTestCase(unittest.HomeserverTestCase):
|
|||||||
request.args[b"access_token"] = [self.test_token]
|
request.args[b"access_token"] = [self.test_token]
|
||||||
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
request.requestHeaders.getRawHeaders = mock_getRawHeaders()
|
||||||
self.get_success(self.auth.get_user_by_req(request))
|
self.get_success(self.auth.get_user_by_req(request))
|
||||||
self.assertEquals(self.store.insert_client_ip.call_count, 2)
|
self.assertEqual(self.store.insert_client_ip.call_count, 2)
|
||||||
|
|
||||||
def test_get_user_from_macaroon(self):
|
def test_get_user_from_macaroon(self):
|
||||||
self.store.get_user_by_access_token = simple_async_mock(
|
self.store.get_user_by_access_token = simple_async_mock(
|
||||||
@ -369,9 +369,9 @@ class AuthTestCase(unittest.HomeserverTestCase):
|
|||||||
self.store.get_monthly_active_count = simple_async_mock(lots_of_users)
|
self.store.get_monthly_active_count = simple_async_mock(lots_of_users)
|
||||||
|
|
||||||
e = self.get_failure(self.auth.check_auth_blocking(), ResourceLimitError)
|
e = self.get_failure(self.auth.check_auth_blocking(), ResourceLimitError)
|
||||||
self.assertEquals(e.value.admin_contact, self.hs.config.server.admin_contact)
|
self.assertEqual(e.value.admin_contact, self.hs.config.server.admin_contact)
|
||||||
self.assertEquals(e.value.errcode, Codes.RESOURCE_LIMIT_EXCEEDED)
|
self.assertEqual(e.value.errcode, Codes.RESOURCE_LIMIT_EXCEEDED)
|
||||||
self.assertEquals(e.value.code, 403)
|
self.assertEqual(e.value.code, 403)
|
||||||
|
|
||||||
# Ensure does not throw an error
|
# Ensure does not throw an error
|
||||||
self.store.get_monthly_active_count = simple_async_mock(small_number_of_users)
|
self.store.get_monthly_active_count = simple_async_mock(small_number_of_users)
|
||||||
@ -473,9 +473,9 @@ class AuthTestCase(unittest.HomeserverTestCase):
|
|||||||
self.auth_blocking._hs_disabled = True
|
self.auth_blocking._hs_disabled = True
|
||||||
self.auth_blocking._hs_disabled_message = "Reason for being disabled"
|
self.auth_blocking._hs_disabled_message = "Reason for being disabled"
|
||||||
e = self.get_failure(self.auth.check_auth_blocking(), ResourceLimitError)
|
e = self.get_failure(self.auth.check_auth_blocking(), ResourceLimitError)
|
||||||
self.assertEquals(e.value.admin_contact, self.hs.config.server.admin_contact)
|
self.assertEqual(e.value.admin_contact, self.hs.config.server.admin_contact)
|
||||||
self.assertEquals(e.value.errcode, Codes.RESOURCE_LIMIT_EXCEEDED)
|
self.assertEqual(e.value.errcode, Codes.RESOURCE_LIMIT_EXCEEDED)
|
||||||
self.assertEquals(e.value.code, 403)
|
self.assertEqual(e.value.code, 403)
|
||||||
|
|
||||||
def test_hs_disabled_no_server_notices_user(self):
|
def test_hs_disabled_no_server_notices_user(self):
|
||||||
"""Check that 'hs_disabled_message' works correctly when there is no
|
"""Check that 'hs_disabled_message' works correctly when there is no
|
||||||
@ -488,9 +488,9 @@ class AuthTestCase(unittest.HomeserverTestCase):
|
|||||||
self.auth_blocking._hs_disabled = True
|
self.auth_blocking._hs_disabled = True
|
||||||
self.auth_blocking._hs_disabled_message = "Reason for being disabled"
|
self.auth_blocking._hs_disabled_message = "Reason for being disabled"
|
||||||
e = self.get_failure(self.auth.check_auth_blocking(), ResourceLimitError)
|
e = self.get_failure(self.auth.check_auth_blocking(), ResourceLimitError)
|
||||||
self.assertEquals(e.value.admin_contact, self.hs.config.server.admin_contact)
|
self.assertEqual(e.value.admin_contact, self.hs.config.server.admin_contact)
|
||||||
self.assertEquals(e.value.errcode, Codes.RESOURCE_LIMIT_EXCEEDED)
|
self.assertEqual(e.value.errcode, Codes.RESOURCE_LIMIT_EXCEEDED)
|
||||||
self.assertEquals(e.value.code, 403)
|
self.assertEqual(e.value.code, 403)
|
||||||
|
|
||||||
def test_server_notices_mxid_special_cased(self):
|
def test_server_notices_mxid_special_cased(self):
|
||||||
self.auth_blocking._hs_disabled = True
|
self.auth_blocking._hs_disabled = True
|
||||||
|
@ -364,7 +364,7 @@ class FilteringTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
results = self.get_success(user_filter.filter_presence(events=events))
|
results = self.get_success(user_filter.filter_presence(events=events))
|
||||||
self.assertEquals(events, results)
|
self.assertEqual(events, results)
|
||||||
|
|
||||||
def test_filter_presence_no_match(self):
|
def test_filter_presence_no_match(self):
|
||||||
user_filter_json = {"presence": {"types": ["m.*"]}}
|
user_filter_json = {"presence": {"types": ["m.*"]}}
|
||||||
@ -388,7 +388,7 @@ class FilteringTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
results = self.get_success(user_filter.filter_presence(events=events))
|
results = self.get_success(user_filter.filter_presence(events=events))
|
||||||
self.assertEquals([], results)
|
self.assertEqual([], results)
|
||||||
|
|
||||||
def test_filter_room_state_match(self):
|
def test_filter_room_state_match(self):
|
||||||
user_filter_json = {"room": {"state": {"types": ["m.*"]}}}
|
user_filter_json = {"room": {"state": {"types": ["m.*"]}}}
|
||||||
@ -407,7 +407,7 @@ class FilteringTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
results = self.get_success(user_filter.filter_room_state(events=events))
|
results = self.get_success(user_filter.filter_room_state(events=events))
|
||||||
self.assertEquals(events, results)
|
self.assertEqual(events, results)
|
||||||
|
|
||||||
def test_filter_room_state_no_match(self):
|
def test_filter_room_state_no_match(self):
|
||||||
user_filter_json = {"room": {"state": {"types": ["m.*"]}}}
|
user_filter_json = {"room": {"state": {"types": ["m.*"]}}}
|
||||||
@ -428,7 +428,7 @@ class FilteringTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
results = self.get_success(user_filter.filter_room_state(events))
|
results = self.get_success(user_filter.filter_room_state(events))
|
||||||
self.assertEquals([], results)
|
self.assertEqual([], results)
|
||||||
|
|
||||||
def test_filter_rooms(self):
|
def test_filter_rooms(self):
|
||||||
definition = {
|
definition = {
|
||||||
@ -444,7 +444,7 @@ class FilteringTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
filtered_room_ids = list(Filter(self.hs, definition).filter_rooms(room_ids))
|
filtered_room_ids = list(Filter(self.hs, definition).filter_rooms(room_ids))
|
||||||
|
|
||||||
self.assertEquals(filtered_room_ids, ["!allowed:example.com"])
|
self.assertEqual(filtered_room_ids, ["!allowed:example.com"])
|
||||||
|
|
||||||
@unittest.override_config({"experimental_features": {"msc3440_enabled": True}})
|
@unittest.override_config({"experimental_features": {"msc3440_enabled": True}})
|
||||||
def test_filter_relations(self):
|
def test_filter_relations(self):
|
||||||
@ -486,7 +486,7 @@ class FilteringTestCase(unittest.HomeserverTestCase):
|
|||||||
Filter(self.hs, definition)._check_event_relations(events)
|
Filter(self.hs, definition)._check_event_relations(events)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(filtered_events, events[1:])
|
self.assertEqual(filtered_events, events[1:])
|
||||||
|
|
||||||
def test_add_filter(self):
|
def test_add_filter(self):
|
||||||
user_filter_json = {"room": {"state": {"types": ["m.*"]}}}
|
user_filter_json = {"room": {"state": {"types": ["m.*"]}}}
|
||||||
@ -497,8 +497,8 @@ class FilteringTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(filter_id, 0)
|
self.assertEqual(filter_id, 0)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
user_filter_json,
|
user_filter_json,
|
||||||
(
|
(
|
||||||
self.get_success(
|
self.get_success(
|
||||||
@ -524,6 +524,6 @@ class FilteringTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(filter.get_filter_json(), user_filter_json)
|
self.assertEqual(filter.get_filter_json(), user_filter_json)
|
||||||
|
|
||||||
self.assertRegexpMatches(repr(filter), r"<FilterCollection \{.*\}>")
|
self.assertRegex(repr(filter), r"<FilterCollection \{.*\}>")
|
||||||
|
@ -14,19 +14,19 @@ class TestRatelimiter(unittest.HomeserverTestCase):
|
|||||||
limiter.can_do_action(None, key="test_id", _time_now_s=0)
|
limiter.can_do_action(None, key="test_id", _time_now_s=0)
|
||||||
)
|
)
|
||||||
self.assertTrue(allowed)
|
self.assertTrue(allowed)
|
||||||
self.assertEquals(10.0, time_allowed)
|
self.assertEqual(10.0, time_allowed)
|
||||||
|
|
||||||
allowed, time_allowed = self.get_success_or_raise(
|
allowed, time_allowed = self.get_success_or_raise(
|
||||||
limiter.can_do_action(None, key="test_id", _time_now_s=5)
|
limiter.can_do_action(None, key="test_id", _time_now_s=5)
|
||||||
)
|
)
|
||||||
self.assertFalse(allowed)
|
self.assertFalse(allowed)
|
||||||
self.assertEquals(10.0, time_allowed)
|
self.assertEqual(10.0, time_allowed)
|
||||||
|
|
||||||
allowed, time_allowed = self.get_success_or_raise(
|
allowed, time_allowed = self.get_success_or_raise(
|
||||||
limiter.can_do_action(None, key="test_id", _time_now_s=10)
|
limiter.can_do_action(None, key="test_id", _time_now_s=10)
|
||||||
)
|
)
|
||||||
self.assertTrue(allowed)
|
self.assertTrue(allowed)
|
||||||
self.assertEquals(20.0, time_allowed)
|
self.assertEqual(20.0, time_allowed)
|
||||||
|
|
||||||
def test_allowed_appservice_ratelimited_via_can_requester_do_action(self):
|
def test_allowed_appservice_ratelimited_via_can_requester_do_action(self):
|
||||||
appservice = ApplicationService(
|
appservice = ApplicationService(
|
||||||
@ -45,19 +45,19 @@ class TestRatelimiter(unittest.HomeserverTestCase):
|
|||||||
limiter.can_do_action(as_requester, _time_now_s=0)
|
limiter.can_do_action(as_requester, _time_now_s=0)
|
||||||
)
|
)
|
||||||
self.assertTrue(allowed)
|
self.assertTrue(allowed)
|
||||||
self.assertEquals(10.0, time_allowed)
|
self.assertEqual(10.0, time_allowed)
|
||||||
|
|
||||||
allowed, time_allowed = self.get_success_or_raise(
|
allowed, time_allowed = self.get_success_or_raise(
|
||||||
limiter.can_do_action(as_requester, _time_now_s=5)
|
limiter.can_do_action(as_requester, _time_now_s=5)
|
||||||
)
|
)
|
||||||
self.assertFalse(allowed)
|
self.assertFalse(allowed)
|
||||||
self.assertEquals(10.0, time_allowed)
|
self.assertEqual(10.0, time_allowed)
|
||||||
|
|
||||||
allowed, time_allowed = self.get_success_or_raise(
|
allowed, time_allowed = self.get_success_or_raise(
|
||||||
limiter.can_do_action(as_requester, _time_now_s=10)
|
limiter.can_do_action(as_requester, _time_now_s=10)
|
||||||
)
|
)
|
||||||
self.assertTrue(allowed)
|
self.assertTrue(allowed)
|
||||||
self.assertEquals(20.0, time_allowed)
|
self.assertEqual(20.0, time_allowed)
|
||||||
|
|
||||||
def test_allowed_appservice_via_can_requester_do_action(self):
|
def test_allowed_appservice_via_can_requester_do_action(self):
|
||||||
appservice = ApplicationService(
|
appservice = ApplicationService(
|
||||||
@ -76,19 +76,19 @@ class TestRatelimiter(unittest.HomeserverTestCase):
|
|||||||
limiter.can_do_action(as_requester, _time_now_s=0)
|
limiter.can_do_action(as_requester, _time_now_s=0)
|
||||||
)
|
)
|
||||||
self.assertTrue(allowed)
|
self.assertTrue(allowed)
|
||||||
self.assertEquals(-1, time_allowed)
|
self.assertEqual(-1, time_allowed)
|
||||||
|
|
||||||
allowed, time_allowed = self.get_success_or_raise(
|
allowed, time_allowed = self.get_success_or_raise(
|
||||||
limiter.can_do_action(as_requester, _time_now_s=5)
|
limiter.can_do_action(as_requester, _time_now_s=5)
|
||||||
)
|
)
|
||||||
self.assertTrue(allowed)
|
self.assertTrue(allowed)
|
||||||
self.assertEquals(-1, time_allowed)
|
self.assertEqual(-1, time_allowed)
|
||||||
|
|
||||||
allowed, time_allowed = self.get_success_or_raise(
|
allowed, time_allowed = self.get_success_or_raise(
|
||||||
limiter.can_do_action(as_requester, _time_now_s=10)
|
limiter.can_do_action(as_requester, _time_now_s=10)
|
||||||
)
|
)
|
||||||
self.assertTrue(allowed)
|
self.assertTrue(allowed)
|
||||||
self.assertEquals(-1, time_allowed)
|
self.assertEqual(-1, time_allowed)
|
||||||
|
|
||||||
def test_allowed_via_ratelimit(self):
|
def test_allowed_via_ratelimit(self):
|
||||||
limiter = Ratelimiter(
|
limiter = Ratelimiter(
|
||||||
@ -246,7 +246,7 @@ class TestRatelimiter(unittest.HomeserverTestCase):
|
|||||||
limiter.can_do_action(None, key="test_id", n_actions=3, _time_now_s=0)
|
limiter.can_do_action(None, key="test_id", n_actions=3, _time_now_s=0)
|
||||||
)
|
)
|
||||||
self.assertTrue(allowed)
|
self.assertTrue(allowed)
|
||||||
self.assertEquals(10.0, time_allowed)
|
self.assertEqual(10.0, time_allowed)
|
||||||
|
|
||||||
# Test that, after doing these 3 actions, we can't do any more action without
|
# Test that, after doing these 3 actions, we can't do any more action without
|
||||||
# waiting.
|
# waiting.
|
||||||
@ -254,7 +254,7 @@ class TestRatelimiter(unittest.HomeserverTestCase):
|
|||||||
limiter.can_do_action(None, key="test_id", n_actions=1, _time_now_s=0)
|
limiter.can_do_action(None, key="test_id", n_actions=1, _time_now_s=0)
|
||||||
)
|
)
|
||||||
self.assertFalse(allowed)
|
self.assertFalse(allowed)
|
||||||
self.assertEquals(10.0, time_allowed)
|
self.assertEqual(10.0, time_allowed)
|
||||||
|
|
||||||
# Test that after waiting we can do only 1 action.
|
# Test that after waiting we can do only 1 action.
|
||||||
allowed, time_allowed = self.get_success_or_raise(
|
allowed, time_allowed = self.get_success_or_raise(
|
||||||
@ -269,7 +269,7 @@ class TestRatelimiter(unittest.HomeserverTestCase):
|
|||||||
self.assertTrue(allowed)
|
self.assertTrue(allowed)
|
||||||
# The time allowed is the current time because we could still repeat the action
|
# The time allowed is the current time because we could still repeat the action
|
||||||
# once.
|
# once.
|
||||||
self.assertEquals(10.0, time_allowed)
|
self.assertEqual(10.0, time_allowed)
|
||||||
|
|
||||||
allowed, time_allowed = self.get_success_or_raise(
|
allowed, time_allowed = self.get_success_or_raise(
|
||||||
limiter.can_do_action(None, key="test_id", n_actions=2, _time_now_s=10)
|
limiter.can_do_action(None, key="test_id", n_actions=2, _time_now_s=10)
|
||||||
@ -277,7 +277,7 @@ class TestRatelimiter(unittest.HomeserverTestCase):
|
|||||||
self.assertFalse(allowed)
|
self.assertFalse(allowed)
|
||||||
# The time allowed doesn't change despite allowed being False because, while we
|
# The time allowed doesn't change despite allowed being False because, while we
|
||||||
# don't allow 2 actions, we could still do 1.
|
# don't allow 2 actions, we could still do 1.
|
||||||
self.assertEquals(10.0, time_allowed)
|
self.assertEqual(10.0, time_allowed)
|
||||||
|
|
||||||
# Test that after waiting a bit more we can do 2 actions.
|
# Test that after waiting a bit more we can do 2 actions.
|
||||||
allowed, time_allowed = self.get_success_or_raise(
|
allowed, time_allowed = self.get_success_or_raise(
|
||||||
@ -286,4 +286,4 @@ class TestRatelimiter(unittest.HomeserverTestCase):
|
|||||||
self.assertTrue(allowed)
|
self.assertTrue(allowed)
|
||||||
# The time allowed is the current time because we could still repeat the action
|
# The time allowed is the current time because we could still repeat the action
|
||||||
# once.
|
# once.
|
||||||
self.assertEquals(20.0, time_allowed)
|
self.assertEqual(20.0, time_allowed)
|
||||||
|
@ -71,7 +71,7 @@ class ApplicationServiceSchedulerTransactionCtrlTestCase(unittest.TestCase):
|
|||||||
one_time_key_counts={},
|
one_time_key_counts={},
|
||||||
unused_fallback_keys={},
|
unused_fallback_keys={},
|
||||||
)
|
)
|
||||||
self.assertEquals(0, len(self.txnctrl.recoverers)) # no recoverer made
|
self.assertEqual(0, len(self.txnctrl.recoverers)) # no recoverer made
|
||||||
txn.complete.assert_called_once_with(self.store) # txn completed
|
txn.complete.assert_called_once_with(self.store) # txn completed
|
||||||
|
|
||||||
def test_single_service_down(self):
|
def test_single_service_down(self):
|
||||||
@ -97,8 +97,8 @@ class ApplicationServiceSchedulerTransactionCtrlTestCase(unittest.TestCase):
|
|||||||
one_time_key_counts={},
|
one_time_key_counts={},
|
||||||
unused_fallback_keys={},
|
unused_fallback_keys={},
|
||||||
)
|
)
|
||||||
self.assertEquals(0, txn.send.call_count) # txn not sent though
|
self.assertEqual(0, txn.send.call_count) # txn not sent though
|
||||||
self.assertEquals(0, txn.complete.call_count) # or completed
|
self.assertEqual(0, txn.complete.call_count) # or completed
|
||||||
|
|
||||||
def test_single_service_up_txn_not_sent(self):
|
def test_single_service_up_txn_not_sent(self):
|
||||||
# Test: The AS is up and the txn is not sent. A Recoverer is made and
|
# Test: The AS is up and the txn is not sent. A Recoverer is made and
|
||||||
@ -125,10 +125,10 @@ class ApplicationServiceSchedulerTransactionCtrlTestCase(unittest.TestCase):
|
|||||||
one_time_key_counts={},
|
one_time_key_counts={},
|
||||||
unused_fallback_keys={},
|
unused_fallback_keys={},
|
||||||
)
|
)
|
||||||
self.assertEquals(1, self.recoverer_fn.call_count) # recoverer made
|
self.assertEqual(1, self.recoverer_fn.call_count) # recoverer made
|
||||||
self.assertEquals(1, self.recoverer.recover.call_count) # and invoked
|
self.assertEqual(1, self.recoverer.recover.call_count) # and invoked
|
||||||
self.assertEquals(1, len(self.txnctrl.recoverers)) # and stored
|
self.assertEqual(1, len(self.txnctrl.recoverers)) # and stored
|
||||||
self.assertEquals(0, txn.complete.call_count) # txn not completed
|
self.assertEqual(0, txn.complete.call_count) # txn not completed
|
||||||
self.store.set_appservice_state.assert_called_once_with(
|
self.store.set_appservice_state.assert_called_once_with(
|
||||||
service, ApplicationServiceState.DOWN # service marked as down
|
service, ApplicationServiceState.DOWN # service marked as down
|
||||||
)
|
)
|
||||||
@ -161,17 +161,17 @@ class ApplicationServiceSchedulerRecovererTestCase(unittest.TestCase):
|
|||||||
|
|
||||||
self.recoverer.recover()
|
self.recoverer.recover()
|
||||||
# shouldn't have called anything prior to waiting for exp backoff
|
# shouldn't have called anything prior to waiting for exp backoff
|
||||||
self.assertEquals(0, self.store.get_oldest_unsent_txn.call_count)
|
self.assertEqual(0, self.store.get_oldest_unsent_txn.call_count)
|
||||||
txn.send = simple_async_mock(True)
|
txn.send = simple_async_mock(True)
|
||||||
txn.complete = simple_async_mock(None)
|
txn.complete = simple_async_mock(None)
|
||||||
# wait for exp backoff
|
# wait for exp backoff
|
||||||
self.clock.advance_time(2)
|
self.clock.advance_time(2)
|
||||||
self.assertEquals(1, txn.send.call_count)
|
self.assertEqual(1, txn.send.call_count)
|
||||||
self.assertEquals(1, txn.complete.call_count)
|
self.assertEqual(1, txn.complete.call_count)
|
||||||
# 2 because it needs to get None to know there are no more txns
|
# 2 because it needs to get None to know there are no more txns
|
||||||
self.assertEquals(2, self.store.get_oldest_unsent_txn.call_count)
|
self.assertEqual(2, self.store.get_oldest_unsent_txn.call_count)
|
||||||
self.callback.assert_called_once_with(self.recoverer)
|
self.callback.assert_called_once_with(self.recoverer)
|
||||||
self.assertEquals(self.recoverer.service, self.service)
|
self.assertEqual(self.recoverer.service, self.service)
|
||||||
|
|
||||||
def test_recover_retry_txn(self):
|
def test_recover_retry_txn(self):
|
||||||
txn = Mock()
|
txn = Mock()
|
||||||
@ -187,26 +187,26 @@ class ApplicationServiceSchedulerRecovererTestCase(unittest.TestCase):
|
|||||||
self.store.get_oldest_unsent_txn = Mock(side_effect=take_txn)
|
self.store.get_oldest_unsent_txn = Mock(side_effect=take_txn)
|
||||||
|
|
||||||
self.recoverer.recover()
|
self.recoverer.recover()
|
||||||
self.assertEquals(0, self.store.get_oldest_unsent_txn.call_count)
|
self.assertEqual(0, self.store.get_oldest_unsent_txn.call_count)
|
||||||
txn.send = simple_async_mock(False)
|
txn.send = simple_async_mock(False)
|
||||||
txn.complete = simple_async_mock(None)
|
txn.complete = simple_async_mock(None)
|
||||||
self.clock.advance_time(2)
|
self.clock.advance_time(2)
|
||||||
self.assertEquals(1, txn.send.call_count)
|
self.assertEqual(1, txn.send.call_count)
|
||||||
self.assertEquals(0, txn.complete.call_count)
|
self.assertEqual(0, txn.complete.call_count)
|
||||||
self.assertEquals(0, self.callback.call_count)
|
self.assertEqual(0, self.callback.call_count)
|
||||||
self.clock.advance_time(4)
|
self.clock.advance_time(4)
|
||||||
self.assertEquals(2, txn.send.call_count)
|
self.assertEqual(2, txn.send.call_count)
|
||||||
self.assertEquals(0, txn.complete.call_count)
|
self.assertEqual(0, txn.complete.call_count)
|
||||||
self.assertEquals(0, self.callback.call_count)
|
self.assertEqual(0, self.callback.call_count)
|
||||||
self.clock.advance_time(8)
|
self.clock.advance_time(8)
|
||||||
self.assertEquals(3, txn.send.call_count)
|
self.assertEqual(3, txn.send.call_count)
|
||||||
self.assertEquals(0, txn.complete.call_count)
|
self.assertEqual(0, txn.complete.call_count)
|
||||||
self.assertEquals(0, self.callback.call_count)
|
self.assertEqual(0, self.callback.call_count)
|
||||||
txn.send = simple_async_mock(True) # successfully send the txn
|
txn.send = simple_async_mock(True) # successfully send the txn
|
||||||
pop_txn = True # returns the txn the first time, then no more.
|
pop_txn = True # returns the txn the first time, then no more.
|
||||||
self.clock.advance_time(16)
|
self.clock.advance_time(16)
|
||||||
self.assertEquals(1, txn.send.call_count) # new mock reset call count
|
self.assertEqual(1, txn.send.call_count) # new mock reset call count
|
||||||
self.assertEquals(1, txn.complete.call_count)
|
self.assertEqual(1, txn.complete.call_count)
|
||||||
self.callback.assert_called_once_with(self.recoverer)
|
self.callback.assert_called_once_with(self.recoverer)
|
||||||
|
|
||||||
|
|
||||||
@ -241,13 +241,13 @@ class ApplicationServiceSchedulerQueuerTestCase(unittest.HomeserverTestCase):
|
|||||||
self.scheduler.enqueue_for_appservice(service, events=[event2])
|
self.scheduler.enqueue_for_appservice(service, events=[event2])
|
||||||
self.scheduler.enqueue_for_appservice(service, events=[event3])
|
self.scheduler.enqueue_for_appservice(service, events=[event3])
|
||||||
self.txn_ctrl.send.assert_called_with(service, [event], [], [], None, None)
|
self.txn_ctrl.send.assert_called_with(service, [event], [], [], None, None)
|
||||||
self.assertEquals(1, self.txn_ctrl.send.call_count)
|
self.assertEqual(1, self.txn_ctrl.send.call_count)
|
||||||
# Resolve the send event: expect the queued events to be sent
|
# Resolve the send event: expect the queued events to be sent
|
||||||
d.callback(service)
|
d.callback(service)
|
||||||
self.txn_ctrl.send.assert_called_with(
|
self.txn_ctrl.send.assert_called_with(
|
||||||
service, [event2, event3], [], [], None, None
|
service, [event2, event3], [], [], None, None
|
||||||
)
|
)
|
||||||
self.assertEquals(2, self.txn_ctrl.send.call_count)
|
self.assertEqual(2, self.txn_ctrl.send.call_count)
|
||||||
|
|
||||||
def test_multiple_service_queues(self):
|
def test_multiple_service_queues(self):
|
||||||
# Tests that each service has its own queue, and that they don't block
|
# Tests that each service has its own queue, and that they don't block
|
||||||
@ -281,7 +281,7 @@ class ApplicationServiceSchedulerQueuerTestCase(unittest.HomeserverTestCase):
|
|||||||
# service
|
# service
|
||||||
srv_2_defer.callback(srv2)
|
srv_2_defer.callback(srv2)
|
||||||
self.txn_ctrl.send.assert_called_with(srv2, [srv_2_event2], [], [], None, None)
|
self.txn_ctrl.send.assert_called_with(srv2, [srv_2_event2], [], [], None, None)
|
||||||
self.assertEquals(3, self.txn_ctrl.send.call_count)
|
self.assertEqual(3, self.txn_ctrl.send.call_count)
|
||||||
|
|
||||||
def test_send_large_txns(self):
|
def test_send_large_txns(self):
|
||||||
srv_1_defer = defer.Deferred()
|
srv_1_defer = defer.Deferred()
|
||||||
@ -312,7 +312,7 @@ class ApplicationServiceSchedulerQueuerTestCase(unittest.HomeserverTestCase):
|
|||||||
self.txn_ctrl.send.assert_called_with(
|
self.txn_ctrl.send.assert_called_with(
|
||||||
service, event_list[101:], [], [], None, None
|
service, event_list[101:], [], [], None, None
|
||||||
)
|
)
|
||||||
self.assertEquals(3, self.txn_ctrl.send.call_count)
|
self.assertEqual(3, self.txn_ctrl.send.call_count)
|
||||||
|
|
||||||
def test_send_single_ephemeral_no_queue(self):
|
def test_send_single_ephemeral_no_queue(self):
|
||||||
# Expect the event to be sent immediately.
|
# Expect the event to be sent immediately.
|
||||||
@ -346,14 +346,14 @@ class ApplicationServiceSchedulerQueuerTestCase(unittest.HomeserverTestCase):
|
|||||||
self.scheduler.enqueue_for_appservice(service, ephemeral=event_list_2)
|
self.scheduler.enqueue_for_appservice(service, ephemeral=event_list_2)
|
||||||
self.scheduler.enqueue_for_appservice(service, ephemeral=event_list_3)
|
self.scheduler.enqueue_for_appservice(service, ephemeral=event_list_3)
|
||||||
self.txn_ctrl.send.assert_called_with(service, [], event_list_1, [], None, None)
|
self.txn_ctrl.send.assert_called_with(service, [], event_list_1, [], None, None)
|
||||||
self.assertEquals(1, self.txn_ctrl.send.call_count)
|
self.assertEqual(1, self.txn_ctrl.send.call_count)
|
||||||
# Resolve txn_ctrl.send
|
# Resolve txn_ctrl.send
|
||||||
d.callback(service)
|
d.callback(service)
|
||||||
# Expect the queued events to be sent
|
# Expect the queued events to be sent
|
||||||
self.txn_ctrl.send.assert_called_with(
|
self.txn_ctrl.send.assert_called_with(
|
||||||
service, [], event_list_2 + event_list_3, [], None, None
|
service, [], event_list_2 + event_list_3, [], None, None
|
||||||
)
|
)
|
||||||
self.assertEquals(2, self.txn_ctrl.send.call_count)
|
self.assertEqual(2, self.txn_ctrl.send.call_count)
|
||||||
|
|
||||||
def test_send_large_txns_ephemeral(self):
|
def test_send_large_txns_ephemeral(self):
|
||||||
d = defer.Deferred()
|
d = defer.Deferred()
|
||||||
@ -369,4 +369,4 @@ class ApplicationServiceSchedulerQueuerTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
d.callback(service)
|
d.callback(service)
|
||||||
self.txn_ctrl.send.assert_called_with(service, [], second_chunk, [], None, None)
|
self.txn_ctrl.send.assert_called_with(service, [], second_chunk, [], None, None)
|
||||||
self.assertEquals(2, self.txn_ctrl.send.call_count)
|
self.assertEqual(2, self.txn_ctrl.send.call_count)
|
||||||
|
@ -63,14 +63,14 @@ class EventSigningTestCase(unittest.TestCase):
|
|||||||
|
|
||||||
self.assertTrue(hasattr(event, "hashes"))
|
self.assertTrue(hasattr(event, "hashes"))
|
||||||
self.assertIn("sha256", event.hashes)
|
self.assertIn("sha256", event.hashes)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
event.hashes["sha256"], "6tJjLpXtggfke8UxFhAKg82QVkJzvKOVOOSjUDK4ZSI"
|
event.hashes["sha256"], "6tJjLpXtggfke8UxFhAKg82QVkJzvKOVOOSjUDK4ZSI"
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertTrue(hasattr(event, "signatures"))
|
self.assertTrue(hasattr(event, "signatures"))
|
||||||
self.assertIn(HOSTNAME, event.signatures)
|
self.assertIn(HOSTNAME, event.signatures)
|
||||||
self.assertIn(KEY_NAME, event.signatures["domain"])
|
self.assertIn(KEY_NAME, event.signatures["domain"])
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
event.signatures[HOSTNAME][KEY_NAME],
|
event.signatures[HOSTNAME][KEY_NAME],
|
||||||
"2Wptgo4CwmLo/Y8B8qinxApKaCkBG2fjTWB7AbP5Uy+"
|
"2Wptgo4CwmLo/Y8B8qinxApKaCkBG2fjTWB7AbP5Uy+"
|
||||||
"aIbygsSdLOFzvdDjww8zUVKCmI02eP9xtyJxc/cLiBA",
|
"aIbygsSdLOFzvdDjww8zUVKCmI02eP9xtyJxc/cLiBA",
|
||||||
@ -97,14 +97,14 @@ class EventSigningTestCase(unittest.TestCase):
|
|||||||
|
|
||||||
self.assertTrue(hasattr(event, "hashes"))
|
self.assertTrue(hasattr(event, "hashes"))
|
||||||
self.assertIn("sha256", event.hashes)
|
self.assertIn("sha256", event.hashes)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
event.hashes["sha256"], "onLKD1bGljeBWQhWZ1kaP9SorVmRQNdN5aM2JYU2n/g"
|
event.hashes["sha256"], "onLKD1bGljeBWQhWZ1kaP9SorVmRQNdN5aM2JYU2n/g"
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertTrue(hasattr(event, "signatures"))
|
self.assertTrue(hasattr(event, "signatures"))
|
||||||
self.assertIn(HOSTNAME, event.signatures)
|
self.assertIn(HOSTNAME, event.signatures)
|
||||||
self.assertIn(KEY_NAME, event.signatures["domain"])
|
self.assertIn(KEY_NAME, event.signatures["domain"])
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
event.signatures[HOSTNAME][KEY_NAME],
|
event.signatures[HOSTNAME][KEY_NAME],
|
||||||
"Wm+VzmOUOz08Ds+0NTWb1d4CZrVsJSikkeRxh6aCcUw"
|
"Wm+VzmOUOz08Ds+0NTWb1d4CZrVsJSikkeRxh6aCcUw"
|
||||||
"u6pNC78FunoD7KNWzqFn241eYHYMGCA5McEiVPdhzBA",
|
"u6pNC78FunoD7KNWzqFn241eYHYMGCA5McEiVPdhzBA",
|
||||||
|
@ -76,7 +76,7 @@ class FakeRequest:
|
|||||||
@logcontext_clean
|
@logcontext_clean
|
||||||
class KeyringTestCase(unittest.HomeserverTestCase):
|
class KeyringTestCase(unittest.HomeserverTestCase):
|
||||||
def check_context(self, val, expected):
|
def check_context(self, val, expected):
|
||||||
self.assertEquals(getattr(current_context(), "request", None), expected)
|
self.assertEqual(getattr(current_context(), "request", None), expected)
|
||||||
return val
|
return val
|
||||||
|
|
||||||
def test_verify_json_objects_for_server_awaits_previous_requests(self):
|
def test_verify_json_objects_for_server_awaits_previous_requests(self):
|
||||||
@ -96,7 +96,7 @@ class KeyringTestCase(unittest.HomeserverTestCase):
|
|||||||
async def first_lookup_fetch(
|
async def first_lookup_fetch(
|
||||||
server_name: str, key_ids: List[str], minimum_valid_until_ts: int
|
server_name: str, key_ids: List[str], minimum_valid_until_ts: int
|
||||||
) -> Dict[str, FetchKeyResult]:
|
) -> Dict[str, FetchKeyResult]:
|
||||||
# self.assertEquals(current_context().request.id, "context_11")
|
# self.assertEqual(current_context().request.id, "context_11")
|
||||||
self.assertEqual(server_name, "server10")
|
self.assertEqual(server_name, "server10")
|
||||||
self.assertEqual(key_ids, [get_key_id(key1)])
|
self.assertEqual(key_ids, [get_key_id(key1)])
|
||||||
self.assertEqual(minimum_valid_until_ts, 0)
|
self.assertEqual(minimum_valid_until_ts, 0)
|
||||||
@ -137,7 +137,7 @@ class KeyringTestCase(unittest.HomeserverTestCase):
|
|||||||
async def second_lookup_fetch(
|
async def second_lookup_fetch(
|
||||||
server_name: str, key_ids: List[str], minimum_valid_until_ts: int
|
server_name: str, key_ids: List[str], minimum_valid_until_ts: int
|
||||||
) -> Dict[str, FetchKeyResult]:
|
) -> Dict[str, FetchKeyResult]:
|
||||||
# self.assertEquals(current_context().request.id, "context_12")
|
# self.assertEqual(current_context().request.id, "context_12")
|
||||||
return {get_key_id(key1): FetchKeyResult(get_verify_key(key1), 100)}
|
return {get_key_id(key1): FetchKeyResult(get_verify_key(key1), 100)}
|
||||||
|
|
||||||
mock_fetcher.get_keys.reset_mock()
|
mock_fetcher.get_keys.reset_mock()
|
||||||
|
@ -395,7 +395,7 @@ class SerializeEventTestCase(unittest.TestCase):
|
|||||||
return serialize_event(ev, 1479807801915, only_event_fields=fields)
|
return serialize_event(ev, 1479807801915, only_event_fields=fields)
|
||||||
|
|
||||||
def test_event_fields_works_with_keys(self):
|
def test_event_fields_works_with_keys(self):
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
self.serialize(
|
self.serialize(
|
||||||
MockEvent(sender="@alice:localhost", room_id="!foo:bar"), ["room_id"]
|
MockEvent(sender="@alice:localhost", room_id="!foo:bar"), ["room_id"]
|
||||||
),
|
),
|
||||||
@ -403,7 +403,7 @@ class SerializeEventTestCase(unittest.TestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def test_event_fields_works_with_nested_keys(self):
|
def test_event_fields_works_with_nested_keys(self):
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
self.serialize(
|
self.serialize(
|
||||||
MockEvent(
|
MockEvent(
|
||||||
sender="@alice:localhost",
|
sender="@alice:localhost",
|
||||||
@ -416,7 +416,7 @@ class SerializeEventTestCase(unittest.TestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def test_event_fields_works_with_dot_keys(self):
|
def test_event_fields_works_with_dot_keys(self):
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
self.serialize(
|
self.serialize(
|
||||||
MockEvent(
|
MockEvent(
|
||||||
sender="@alice:localhost",
|
sender="@alice:localhost",
|
||||||
@ -429,7 +429,7 @@ class SerializeEventTestCase(unittest.TestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def test_event_fields_works_with_nested_dot_keys(self):
|
def test_event_fields_works_with_nested_dot_keys(self):
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
self.serialize(
|
self.serialize(
|
||||||
MockEvent(
|
MockEvent(
|
||||||
sender="@alice:localhost",
|
sender="@alice:localhost",
|
||||||
@ -445,7 +445,7 @@ class SerializeEventTestCase(unittest.TestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def test_event_fields_nops_with_unknown_keys(self):
|
def test_event_fields_nops_with_unknown_keys(self):
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
self.serialize(
|
self.serialize(
|
||||||
MockEvent(
|
MockEvent(
|
||||||
sender="@alice:localhost",
|
sender="@alice:localhost",
|
||||||
@ -458,7 +458,7 @@ class SerializeEventTestCase(unittest.TestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def test_event_fields_nops_with_non_dict_keys(self):
|
def test_event_fields_nops_with_non_dict_keys(self):
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
self.serialize(
|
self.serialize(
|
||||||
MockEvent(
|
MockEvent(
|
||||||
sender="@alice:localhost",
|
sender="@alice:localhost",
|
||||||
@ -471,7 +471,7 @@ class SerializeEventTestCase(unittest.TestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def test_event_fields_nops_with_array_keys(self):
|
def test_event_fields_nops_with_array_keys(self):
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
self.serialize(
|
self.serialize(
|
||||||
MockEvent(
|
MockEvent(
|
||||||
sender="@alice:localhost",
|
sender="@alice:localhost",
|
||||||
@ -484,7 +484,7 @@ class SerializeEventTestCase(unittest.TestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def test_event_fields_all_fields_if_empty(self):
|
def test_event_fields_all_fields_if_empty(self):
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
self.serialize(
|
self.serialize(
|
||||||
MockEvent(
|
MockEvent(
|
||||||
type="foo",
|
type="foo",
|
||||||
|
@ -50,7 +50,7 @@ class RoomComplexityTests(unittest.FederatingHomeserverTestCase):
|
|||||||
channel = self.make_signed_federation_request(
|
channel = self.make_signed_federation_request(
|
||||||
"GET", "/_matrix/federation/unstable/rooms/%s/complexity" % (room_1,)
|
"GET", "/_matrix/federation/unstable/rooms/%s/complexity" % (room_1,)
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
complexity = channel.json_body["v1"]
|
complexity = channel.json_body["v1"]
|
||||||
self.assertTrue(complexity > 0, complexity)
|
self.assertTrue(complexity > 0, complexity)
|
||||||
|
|
||||||
@ -62,7 +62,7 @@ class RoomComplexityTests(unittest.FederatingHomeserverTestCase):
|
|||||||
channel = self.make_signed_federation_request(
|
channel = self.make_signed_federation_request(
|
||||||
"GET", "/_matrix/federation/unstable/rooms/%s/complexity" % (room_1,)
|
"GET", "/_matrix/federation/unstable/rooms/%s/complexity" % (room_1,)
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
complexity = channel.json_body["v1"]
|
complexity = channel.json_body["v1"]
|
||||||
self.assertEqual(complexity, 1.23)
|
self.assertEqual(complexity, 1.23)
|
||||||
|
|
||||||
|
@ -59,7 +59,7 @@ class FederationServerTests(unittest.FederatingHomeserverTestCase):
|
|||||||
"/_matrix/federation/v1/get_missing_events/%s" % (room_1,),
|
"/_matrix/federation/v1/get_missing_events/%s" % (room_1,),
|
||||||
query_content,
|
query_content,
|
||||||
)
|
)
|
||||||
self.assertEquals(400, channel.code, channel.result)
|
self.assertEqual(400, channel.code, channel.result)
|
||||||
self.assertEqual(channel.json_body["errcode"], "M_NOT_JSON")
|
self.assertEqual(channel.json_body["errcode"], "M_NOT_JSON")
|
||||||
|
|
||||||
|
|
||||||
@ -125,7 +125,7 @@ class StateQueryTests(unittest.FederatingHomeserverTestCase):
|
|||||||
channel = self.make_signed_federation_request(
|
channel = self.make_signed_federation_request(
|
||||||
"GET", "/_matrix/federation/v1/state/%s" % (room_1,)
|
"GET", "/_matrix/federation/v1/state/%s" % (room_1,)
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
channel.json_body["room_version"],
|
channel.json_body["room_version"],
|
||||||
@ -157,7 +157,7 @@ class StateQueryTests(unittest.FederatingHomeserverTestCase):
|
|||||||
channel = self.make_signed_federation_request(
|
channel = self.make_signed_federation_request(
|
||||||
"GET", "/_matrix/federation/v1/state/%s" % (room_1,)
|
"GET", "/_matrix/federation/v1/state/%s" % (room_1,)
|
||||||
)
|
)
|
||||||
self.assertEquals(403, channel.code, channel.result)
|
self.assertEqual(403, channel.code, channel.result)
|
||||||
self.assertEqual(channel.json_body["errcode"], "M_FORBIDDEN")
|
self.assertEqual(channel.json_body["errcode"], "M_FORBIDDEN")
|
||||||
|
|
||||||
|
|
||||||
@ -189,7 +189,7 @@ class SendJoinFederationTests(unittest.FederatingHomeserverTestCase):
|
|||||||
f"/_matrix/federation/v1/make_join/{self._room_id}/{user_id}"
|
f"/_matrix/federation/v1/make_join/{self._room_id}/{user_id}"
|
||||||
f"?ver={DEFAULT_ROOM_VERSION}",
|
f"?ver={DEFAULT_ROOM_VERSION}",
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.code, 200, channel.json_body)
|
self.assertEqual(channel.code, 200, channel.json_body)
|
||||||
return channel.json_body
|
return channel.json_body
|
||||||
|
|
||||||
def test_send_join(self):
|
def test_send_join(self):
|
||||||
@ -209,7 +209,7 @@ class SendJoinFederationTests(unittest.FederatingHomeserverTestCase):
|
|||||||
f"/_matrix/federation/v2/send_join/{self._room_id}/x",
|
f"/_matrix/federation/v2/send_join/{self._room_id}/x",
|
||||||
content=join_event_dict,
|
content=join_event_dict,
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.code, 200, channel.json_body)
|
self.assertEqual(channel.code, 200, channel.json_body)
|
||||||
|
|
||||||
# we should get complete room state back
|
# we should get complete room state back
|
||||||
returned_state = [
|
returned_state = [
|
||||||
@ -266,7 +266,7 @@ class SendJoinFederationTests(unittest.FederatingHomeserverTestCase):
|
|||||||
f"/_matrix/federation/v2/send_join/{self._room_id}/x?org.matrix.msc3706.partial_state=true",
|
f"/_matrix/federation/v2/send_join/{self._room_id}/x?org.matrix.msc3706.partial_state=true",
|
||||||
content=join_event_dict,
|
content=join_event_dict,
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.code, 200, channel.json_body)
|
self.assertEqual(channel.code, 200, channel.json_body)
|
||||||
|
|
||||||
# expect a reduced room state
|
# expect a reduced room state
|
||||||
returned_state = [
|
returned_state = [
|
||||||
|
@ -169,7 +169,7 @@ class KnockingStrippedStateEventHelperMixin(TestCase):
|
|||||||
self.assertIn(event_type, expected_room_state)
|
self.assertIn(event_type, expected_room_state)
|
||||||
|
|
||||||
# Check the state content matches
|
# Check the state content matches
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
expected_room_state[event_type]["content"], event["content"]
|
expected_room_state[event_type]["content"], event["content"]
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -256,7 +256,7 @@ class FederationKnockingTestCase(
|
|||||||
RoomVersions.V7.identifier,
|
RoomVersions.V7.identifier,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
# Note: We don't expect the knock membership event to be sent over federation as
|
# Note: We don't expect the knock membership event to be sent over federation as
|
||||||
# part of the stripped room state, as the knocking homeserver already has that
|
# part of the stripped room state, as the knocking homeserver already has that
|
||||||
@ -266,11 +266,11 @@ class FederationKnockingTestCase(
|
|||||||
knock_event = channel.json_body["event"]
|
knock_event = channel.json_body["event"]
|
||||||
|
|
||||||
# Check that the event has things we expect in it
|
# Check that the event has things we expect in it
|
||||||
self.assertEquals(knock_event["room_id"], room_id)
|
self.assertEqual(knock_event["room_id"], room_id)
|
||||||
self.assertEquals(knock_event["sender"], fake_knocking_user_id)
|
self.assertEqual(knock_event["sender"], fake_knocking_user_id)
|
||||||
self.assertEquals(knock_event["state_key"], fake_knocking_user_id)
|
self.assertEqual(knock_event["state_key"], fake_knocking_user_id)
|
||||||
self.assertEquals(knock_event["type"], EventTypes.Member)
|
self.assertEqual(knock_event["type"], EventTypes.Member)
|
||||||
self.assertEquals(knock_event["content"]["membership"], Membership.KNOCK)
|
self.assertEqual(knock_event["content"]["membership"], Membership.KNOCK)
|
||||||
|
|
||||||
# Turn the event json dict into a proper event.
|
# Turn the event json dict into a proper event.
|
||||||
# We won't sign it properly, but that's OK as we stub out event auth in `prepare`
|
# We won't sign it properly, but that's OK as we stub out event auth in `prepare`
|
||||||
@ -294,7 +294,7 @@ class FederationKnockingTestCase(
|
|||||||
% (room_id, signed_knock_event.event_id),
|
% (room_id, signed_knock_event.event_id),
|
||||||
signed_knock_event_json,
|
signed_knock_event_json,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
# Check that we got the stripped room state in return
|
# Check that we got the stripped room state in return
|
||||||
room_state_events = channel.json_body["knock_state_events"]
|
room_state_events = channel.json_body["knock_state_events"]
|
||||||
|
@ -26,7 +26,7 @@ class RoomDirectoryFederationTests(unittest.FederatingHomeserverTestCase):
|
|||||||
"GET",
|
"GET",
|
||||||
"/_matrix/federation/v1/publicRooms",
|
"/_matrix/federation/v1/publicRooms",
|
||||||
)
|
)
|
||||||
self.assertEquals(403, channel.code)
|
self.assertEqual(403, channel.code)
|
||||||
|
|
||||||
@override_config({"allow_public_rooms_over_federation": True})
|
@override_config({"allow_public_rooms_over_federation": True})
|
||||||
def test_open_public_room_list_over_federation(self):
|
def test_open_public_room_list_over_federation(self):
|
||||||
@ -37,4 +37,4 @@ class RoomDirectoryFederationTests(unittest.FederatingHomeserverTestCase):
|
|||||||
"GET",
|
"GET",
|
||||||
"/_matrix/federation/v1/publicRooms",
|
"/_matrix/federation/v1/publicRooms",
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
|
@ -147,8 +147,8 @@ class AppServiceHandlerTestCase(unittest.TestCase):
|
|||||||
self.mock_as_api.query_alias.assert_called_once_with(
|
self.mock_as_api.query_alias.assert_called_once_with(
|
||||||
interested_service, room_alias_str
|
interested_service, room_alias_str
|
||||||
)
|
)
|
||||||
self.assertEquals(result.room_id, room_id)
|
self.assertEqual(result.room_id, room_id)
|
||||||
self.assertEquals(result.servers, servers)
|
self.assertEqual(result.servers, servers)
|
||||||
|
|
||||||
def test_get_3pe_protocols_no_appservices(self):
|
def test_get_3pe_protocols_no_appservices(self):
|
||||||
self.mock_store.get_app_services.return_value = []
|
self.mock_store.get_app_services.return_value = []
|
||||||
@ -156,7 +156,7 @@ class AppServiceHandlerTestCase(unittest.TestCase):
|
|||||||
defer.ensureDeferred(self.handler.get_3pe_protocols("my-protocol"))
|
defer.ensureDeferred(self.handler.get_3pe_protocols("my-protocol"))
|
||||||
)
|
)
|
||||||
self.mock_as_api.get_3pe_protocol.assert_not_called()
|
self.mock_as_api.get_3pe_protocol.assert_not_called()
|
||||||
self.assertEquals(response, {})
|
self.assertEqual(response, {})
|
||||||
|
|
||||||
def test_get_3pe_protocols_no_protocols(self):
|
def test_get_3pe_protocols_no_protocols(self):
|
||||||
service = self._mkservice(False, [])
|
service = self._mkservice(False, [])
|
||||||
@ -165,7 +165,7 @@ class AppServiceHandlerTestCase(unittest.TestCase):
|
|||||||
defer.ensureDeferred(self.handler.get_3pe_protocols())
|
defer.ensureDeferred(self.handler.get_3pe_protocols())
|
||||||
)
|
)
|
||||||
self.mock_as_api.get_3pe_protocol.assert_not_called()
|
self.mock_as_api.get_3pe_protocol.assert_not_called()
|
||||||
self.assertEquals(response, {})
|
self.assertEqual(response, {})
|
||||||
|
|
||||||
def test_get_3pe_protocols_protocol_no_response(self):
|
def test_get_3pe_protocols_protocol_no_response(self):
|
||||||
service = self._mkservice(False, ["my-protocol"])
|
service = self._mkservice(False, ["my-protocol"])
|
||||||
@ -177,7 +177,7 @@ class AppServiceHandlerTestCase(unittest.TestCase):
|
|||||||
self.mock_as_api.get_3pe_protocol.assert_called_once_with(
|
self.mock_as_api.get_3pe_protocol.assert_called_once_with(
|
||||||
service, "my-protocol"
|
service, "my-protocol"
|
||||||
)
|
)
|
||||||
self.assertEquals(response, {})
|
self.assertEqual(response, {})
|
||||||
|
|
||||||
def test_get_3pe_protocols_select_one_protocol(self):
|
def test_get_3pe_protocols_select_one_protocol(self):
|
||||||
service = self._mkservice(False, ["my-protocol"])
|
service = self._mkservice(False, ["my-protocol"])
|
||||||
@ -191,7 +191,7 @@ class AppServiceHandlerTestCase(unittest.TestCase):
|
|||||||
self.mock_as_api.get_3pe_protocol.assert_called_once_with(
|
self.mock_as_api.get_3pe_protocol.assert_called_once_with(
|
||||||
service, "my-protocol"
|
service, "my-protocol"
|
||||||
)
|
)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
response, {"my-protocol": {"x-protocol-data": 42, "instances": []}}
|
response, {"my-protocol": {"x-protocol-data": 42, "instances": []}}
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -207,7 +207,7 @@ class AppServiceHandlerTestCase(unittest.TestCase):
|
|||||||
self.mock_as_api.get_3pe_protocol.assert_called_once_with(
|
self.mock_as_api.get_3pe_protocol.assert_called_once_with(
|
||||||
service, "my-protocol"
|
service, "my-protocol"
|
||||||
)
|
)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
response, {"my-protocol": {"x-protocol-data": 42, "instances": []}}
|
response, {"my-protocol": {"x-protocol-data": 42, "instances": []}}
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -222,7 +222,7 @@ class AppServiceHandlerTestCase(unittest.TestCase):
|
|||||||
defer.ensureDeferred(self.handler.get_3pe_protocols())
|
defer.ensureDeferred(self.handler.get_3pe_protocols())
|
||||||
)
|
)
|
||||||
self.mock_as_api.get_3pe_protocol.assert_called()
|
self.mock_as_api.get_3pe_protocol.assert_called()
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
response,
|
response,
|
||||||
{
|
{
|
||||||
"my-protocol": {"x-protocol-data": 42, "instances": []},
|
"my-protocol": {"x-protocol-data": 42, "instances": []},
|
||||||
@ -254,7 +254,7 @@ class AppServiceHandlerTestCase(unittest.TestCase):
|
|||||||
defer.ensureDeferred(self.handler.get_3pe_protocols())
|
defer.ensureDeferred(self.handler.get_3pe_protocols())
|
||||||
)
|
)
|
||||||
# It's expected that the second service's data doesn't appear in the response
|
# It's expected that the second service's data doesn't appear in the response
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
response,
|
response,
|
||||||
{
|
{
|
||||||
"my-protocol": {
|
"my-protocol": {
|
||||||
|
@ -63,7 +63,7 @@ class DirectoryTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
result = self.get_success(self.handler.get_association(self.my_room))
|
result = self.get_success(self.handler.get_association(self.my_room))
|
||||||
|
|
||||||
self.assertEquals({"room_id": "!8765qwer:test", "servers": ["test"]}, result)
|
self.assertEqual({"room_id": "!8765qwer:test", "servers": ["test"]}, result)
|
||||||
|
|
||||||
def test_get_remote_association(self):
|
def test_get_remote_association(self):
|
||||||
self.mock_federation.make_query.return_value = make_awaitable(
|
self.mock_federation.make_query.return_value = make_awaitable(
|
||||||
@ -72,7 +72,7 @@ class DirectoryTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
result = self.get_success(self.handler.get_association(self.remote_room))
|
result = self.get_success(self.handler.get_association(self.remote_room))
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
{"room_id": "!8765qwer:test", "servers": ["test", "remote"]}, result
|
{"room_id": "!8765qwer:test", "servers": ["test", "remote"]}, result
|
||||||
)
|
)
|
||||||
self.mock_federation.make_query.assert_called_with(
|
self.mock_federation.make_query.assert_called_with(
|
||||||
@ -94,7 +94,7 @@ class DirectoryTestCase(unittest.HomeserverTestCase):
|
|||||||
self.handler.on_directory_query({"room_alias": "#your-room:test"})
|
self.handler.on_directory_query({"room_alias": "#your-room:test"})
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals({"room_id": "!8765asdf:test", "servers": ["test"]}, response)
|
self.assertEqual({"room_id": "!8765asdf:test", "servers": ["test"]}, response)
|
||||||
|
|
||||||
|
|
||||||
class TestCreateAlias(unittest.HomeserverTestCase):
|
class TestCreateAlias(unittest.HomeserverTestCase):
|
||||||
@ -224,7 +224,7 @@ class TestDeleteAlias(unittest.HomeserverTestCase):
|
|||||||
create_requester(self.test_user), self.room_alias
|
create_requester(self.test_user), self.room_alias
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(self.room_id, result)
|
self.assertEqual(self.room_id, result)
|
||||||
|
|
||||||
# Confirm the alias is gone.
|
# Confirm the alias is gone.
|
||||||
self.get_failure(
|
self.get_failure(
|
||||||
@ -243,7 +243,7 @@ class TestDeleteAlias(unittest.HomeserverTestCase):
|
|||||||
create_requester(self.admin_user), self.room_alias
|
create_requester(self.admin_user), self.room_alias
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(self.room_id, result)
|
self.assertEqual(self.room_id, result)
|
||||||
|
|
||||||
# Confirm the alias is gone.
|
# Confirm the alias is gone.
|
||||||
self.get_failure(
|
self.get_failure(
|
||||||
@ -269,7 +269,7 @@ class TestDeleteAlias(unittest.HomeserverTestCase):
|
|||||||
create_requester(self.test_user), self.room_alias
|
create_requester(self.test_user), self.room_alias
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(self.room_id, result)
|
self.assertEqual(self.room_id, result)
|
||||||
|
|
||||||
# Confirm the alias is gone.
|
# Confirm the alias is gone.
|
||||||
self.get_failure(
|
self.get_failure(
|
||||||
@ -411,7 +411,7 @@ class TestCreateAliasACL(unittest.HomeserverTestCase):
|
|||||||
b"directory/room/%23test%3Atest",
|
b"directory/room/%23test%3Atest",
|
||||||
{"room_id": room_id},
|
{"room_id": room_id},
|
||||||
)
|
)
|
||||||
self.assertEquals(403, channel.code, channel.result)
|
self.assertEqual(403, channel.code, channel.result)
|
||||||
|
|
||||||
def test_allowed(self):
|
def test_allowed(self):
|
||||||
room_id = self.helper.create_room_as(self.user_id)
|
room_id = self.helper.create_room_as(self.user_id)
|
||||||
@ -421,7 +421,7 @@ class TestCreateAliasACL(unittest.HomeserverTestCase):
|
|||||||
b"directory/room/%23unofficial_test%3Atest",
|
b"directory/room/%23unofficial_test%3Atest",
|
||||||
{"room_id": room_id},
|
{"room_id": room_id},
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
def test_denied_during_creation(self):
|
def test_denied_during_creation(self):
|
||||||
"""A room alias that is not allowed should be rejected during creation."""
|
"""A room alias that is not allowed should be rejected during creation."""
|
||||||
@ -443,8 +443,8 @@ class TestCreateAliasACL(unittest.HomeserverTestCase):
|
|||||||
"GET",
|
"GET",
|
||||||
b"directory/room/%23unofficial_test%3Atest",
|
b"directory/room/%23unofficial_test%3Atest",
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertEquals(channel.json_body["room_id"], room_id)
|
self.assertEqual(channel.json_body["room_id"], room_id)
|
||||||
|
|
||||||
|
|
||||||
class TestCreatePublishedRoomACL(unittest.HomeserverTestCase):
|
class TestCreatePublishedRoomACL(unittest.HomeserverTestCase):
|
||||||
@ -572,7 +572,7 @@ class TestRoomListSearchDisabled(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"PUT", b"directory/list/room/%s" % (room_id.encode("ascii"),), b"{}"
|
"PUT", b"directory/list/room/%s" % (room_id.encode("ascii"),), b"{}"
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
self.room_list_handler = hs.get_room_list_handler()
|
self.room_list_handler = hs.get_room_list_handler()
|
||||||
self.directory_handler = hs.get_directory_handler()
|
self.directory_handler = hs.get_directory_handler()
|
||||||
@ -585,7 +585,7 @@ class TestRoomListSearchDisabled(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# Room list is enabled so we should get some results
|
# Room list is enabled so we should get some results
|
||||||
channel = self.make_request("GET", b"publicRooms")
|
channel = self.make_request("GET", b"publicRooms")
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertTrue(len(channel.json_body["chunk"]) > 0)
|
self.assertTrue(len(channel.json_body["chunk"]) > 0)
|
||||||
|
|
||||||
self.room_list_handler.enable_room_list_search = False
|
self.room_list_handler.enable_room_list_search = False
|
||||||
@ -593,7 +593,7 @@ class TestRoomListSearchDisabled(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# Room list disabled so we should get no results
|
# Room list disabled so we should get no results
|
||||||
channel = self.make_request("GET", b"publicRooms")
|
channel = self.make_request("GET", b"publicRooms")
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertTrue(len(channel.json_body["chunk"]) == 0)
|
self.assertTrue(len(channel.json_body["chunk"]) == 0)
|
||||||
|
|
||||||
# Room list disabled so we shouldn't be allowed to publish rooms
|
# Room list disabled so we shouldn't be allowed to publish rooms
|
||||||
@ -601,4 +601,4 @@ class TestRoomListSearchDisabled(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"PUT", b"directory/list/room/%s" % (room_id.encode("ascii"),), b"{}"
|
"PUT", b"directory/list/room/%s" % (room_id.encode("ascii"),), b"{}"
|
||||||
)
|
)
|
||||||
self.assertEquals(403, channel.code, channel.result)
|
self.assertEqual(403, channel.code, channel.result)
|
||||||
|
@ -61,11 +61,11 @@ class PresenceUpdateTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
self.assertTrue(persist_and_notify)
|
self.assertTrue(persist_and_notify)
|
||||||
self.assertTrue(state.currently_active)
|
self.assertTrue(state.currently_active)
|
||||||
self.assertEquals(new_state.state, state.state)
|
self.assertEqual(new_state.state, state.state)
|
||||||
self.assertEquals(new_state.status_msg, state.status_msg)
|
self.assertEqual(new_state.status_msg, state.status_msg)
|
||||||
self.assertEquals(state.last_federation_update_ts, now)
|
self.assertEqual(state.last_federation_update_ts, now)
|
||||||
|
|
||||||
self.assertEquals(wheel_timer.insert.call_count, 3)
|
self.assertEqual(wheel_timer.insert.call_count, 3)
|
||||||
wheel_timer.insert.assert_has_calls(
|
wheel_timer.insert.assert_has_calls(
|
||||||
[
|
[
|
||||||
call(now=now, obj=user_id, then=new_state.last_active_ts + IDLE_TIMER),
|
call(now=now, obj=user_id, then=new_state.last_active_ts + IDLE_TIMER),
|
||||||
@ -104,11 +104,11 @@ class PresenceUpdateTestCase(unittest.HomeserverTestCase):
|
|||||||
self.assertFalse(persist_and_notify)
|
self.assertFalse(persist_and_notify)
|
||||||
self.assertTrue(federation_ping)
|
self.assertTrue(federation_ping)
|
||||||
self.assertTrue(state.currently_active)
|
self.assertTrue(state.currently_active)
|
||||||
self.assertEquals(new_state.state, state.state)
|
self.assertEqual(new_state.state, state.state)
|
||||||
self.assertEquals(new_state.status_msg, state.status_msg)
|
self.assertEqual(new_state.status_msg, state.status_msg)
|
||||||
self.assertEquals(state.last_federation_update_ts, now)
|
self.assertEqual(state.last_federation_update_ts, now)
|
||||||
|
|
||||||
self.assertEquals(wheel_timer.insert.call_count, 3)
|
self.assertEqual(wheel_timer.insert.call_count, 3)
|
||||||
wheel_timer.insert.assert_has_calls(
|
wheel_timer.insert.assert_has_calls(
|
||||||
[
|
[
|
||||||
call(now=now, obj=user_id, then=new_state.last_active_ts + IDLE_TIMER),
|
call(now=now, obj=user_id, then=new_state.last_active_ts + IDLE_TIMER),
|
||||||
@ -149,11 +149,11 @@ class PresenceUpdateTestCase(unittest.HomeserverTestCase):
|
|||||||
self.assertFalse(persist_and_notify)
|
self.assertFalse(persist_and_notify)
|
||||||
self.assertTrue(federation_ping)
|
self.assertTrue(federation_ping)
|
||||||
self.assertTrue(state.currently_active)
|
self.assertTrue(state.currently_active)
|
||||||
self.assertEquals(new_state.state, state.state)
|
self.assertEqual(new_state.state, state.state)
|
||||||
self.assertEquals(new_state.status_msg, state.status_msg)
|
self.assertEqual(new_state.status_msg, state.status_msg)
|
||||||
self.assertEquals(state.last_federation_update_ts, now)
|
self.assertEqual(state.last_federation_update_ts, now)
|
||||||
|
|
||||||
self.assertEquals(wheel_timer.insert.call_count, 3)
|
self.assertEqual(wheel_timer.insert.call_count, 3)
|
||||||
wheel_timer.insert.assert_has_calls(
|
wheel_timer.insert.assert_has_calls(
|
||||||
[
|
[
|
||||||
call(now=now, obj=user_id, then=new_state.last_active_ts + IDLE_TIMER),
|
call(now=now, obj=user_id, then=new_state.last_active_ts + IDLE_TIMER),
|
||||||
@ -191,11 +191,11 @@ class PresenceUpdateTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
self.assertTrue(persist_and_notify)
|
self.assertTrue(persist_and_notify)
|
||||||
self.assertFalse(state.currently_active)
|
self.assertFalse(state.currently_active)
|
||||||
self.assertEquals(new_state.state, state.state)
|
self.assertEqual(new_state.state, state.state)
|
||||||
self.assertEquals(new_state.status_msg, state.status_msg)
|
self.assertEqual(new_state.status_msg, state.status_msg)
|
||||||
self.assertEquals(state.last_federation_update_ts, now)
|
self.assertEqual(state.last_federation_update_ts, now)
|
||||||
|
|
||||||
self.assertEquals(wheel_timer.insert.call_count, 2)
|
self.assertEqual(wheel_timer.insert.call_count, 2)
|
||||||
wheel_timer.insert.assert_has_calls(
|
wheel_timer.insert.assert_has_calls(
|
||||||
[
|
[
|
||||||
call(now=now, obj=user_id, then=new_state.last_active_ts + IDLE_TIMER),
|
call(now=now, obj=user_id, then=new_state.last_active_ts + IDLE_TIMER),
|
||||||
@ -227,10 +227,10 @@ class PresenceUpdateTestCase(unittest.HomeserverTestCase):
|
|||||||
self.assertFalse(persist_and_notify)
|
self.assertFalse(persist_and_notify)
|
||||||
self.assertFalse(federation_ping)
|
self.assertFalse(federation_ping)
|
||||||
self.assertFalse(state.currently_active)
|
self.assertFalse(state.currently_active)
|
||||||
self.assertEquals(new_state.state, state.state)
|
self.assertEqual(new_state.state, state.state)
|
||||||
self.assertEquals(new_state.status_msg, state.status_msg)
|
self.assertEqual(new_state.status_msg, state.status_msg)
|
||||||
|
|
||||||
self.assertEquals(wheel_timer.insert.call_count, 1)
|
self.assertEqual(wheel_timer.insert.call_count, 1)
|
||||||
wheel_timer.insert.assert_has_calls(
|
wheel_timer.insert.assert_has_calls(
|
||||||
[
|
[
|
||||||
call(
|
call(
|
||||||
@ -259,10 +259,10 @@ class PresenceUpdateTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
self.assertTrue(persist_and_notify)
|
self.assertTrue(persist_and_notify)
|
||||||
self.assertEquals(new_state.state, state.state)
|
self.assertEqual(new_state.state, state.state)
|
||||||
self.assertEquals(state.last_federation_update_ts, now)
|
self.assertEqual(state.last_federation_update_ts, now)
|
||||||
|
|
||||||
self.assertEquals(wheel_timer.insert.call_count, 0)
|
self.assertEqual(wheel_timer.insert.call_count, 0)
|
||||||
|
|
||||||
def test_online_to_idle(self):
|
def test_online_to_idle(self):
|
||||||
wheel_timer = Mock()
|
wheel_timer = Mock()
|
||||||
@ -281,12 +281,12 @@ class PresenceUpdateTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
self.assertTrue(persist_and_notify)
|
self.assertTrue(persist_and_notify)
|
||||||
self.assertEquals(new_state.state, state.state)
|
self.assertEqual(new_state.state, state.state)
|
||||||
self.assertEquals(state.last_federation_update_ts, now)
|
self.assertEqual(state.last_federation_update_ts, now)
|
||||||
self.assertEquals(new_state.state, state.state)
|
self.assertEqual(new_state.state, state.state)
|
||||||
self.assertEquals(new_state.status_msg, state.status_msg)
|
self.assertEqual(new_state.status_msg, state.status_msg)
|
||||||
|
|
||||||
self.assertEquals(wheel_timer.insert.call_count, 1)
|
self.assertEqual(wheel_timer.insert.call_count, 1)
|
||||||
wheel_timer.insert.assert_has_calls(
|
wheel_timer.insert.assert_has_calls(
|
||||||
[
|
[
|
||||||
call(
|
call(
|
||||||
@ -357,8 +357,8 @@ class PresenceTimeoutTestCase(unittest.TestCase):
|
|||||||
new_state = handle_timeout(state, is_mine=True, syncing_user_ids=set(), now=now)
|
new_state = handle_timeout(state, is_mine=True, syncing_user_ids=set(), now=now)
|
||||||
|
|
||||||
self.assertIsNotNone(new_state)
|
self.assertIsNotNone(new_state)
|
||||||
self.assertEquals(new_state.state, PresenceState.UNAVAILABLE)
|
self.assertEqual(new_state.state, PresenceState.UNAVAILABLE)
|
||||||
self.assertEquals(new_state.status_msg, status_msg)
|
self.assertEqual(new_state.status_msg, status_msg)
|
||||||
|
|
||||||
def test_busy_no_idle(self):
|
def test_busy_no_idle(self):
|
||||||
"""
|
"""
|
||||||
@ -380,8 +380,8 @@ class PresenceTimeoutTestCase(unittest.TestCase):
|
|||||||
new_state = handle_timeout(state, is_mine=True, syncing_user_ids=set(), now=now)
|
new_state = handle_timeout(state, is_mine=True, syncing_user_ids=set(), now=now)
|
||||||
|
|
||||||
self.assertIsNotNone(new_state)
|
self.assertIsNotNone(new_state)
|
||||||
self.assertEquals(new_state.state, PresenceState.BUSY)
|
self.assertEqual(new_state.state, PresenceState.BUSY)
|
||||||
self.assertEquals(new_state.status_msg, status_msg)
|
self.assertEqual(new_state.status_msg, status_msg)
|
||||||
|
|
||||||
def test_sync_timeout(self):
|
def test_sync_timeout(self):
|
||||||
user_id = "@foo:bar"
|
user_id = "@foo:bar"
|
||||||
@ -399,8 +399,8 @@ class PresenceTimeoutTestCase(unittest.TestCase):
|
|||||||
new_state = handle_timeout(state, is_mine=True, syncing_user_ids=set(), now=now)
|
new_state = handle_timeout(state, is_mine=True, syncing_user_ids=set(), now=now)
|
||||||
|
|
||||||
self.assertIsNotNone(new_state)
|
self.assertIsNotNone(new_state)
|
||||||
self.assertEquals(new_state.state, PresenceState.OFFLINE)
|
self.assertEqual(new_state.state, PresenceState.OFFLINE)
|
||||||
self.assertEquals(new_state.status_msg, status_msg)
|
self.assertEqual(new_state.status_msg, status_msg)
|
||||||
|
|
||||||
def test_sync_online(self):
|
def test_sync_online(self):
|
||||||
user_id = "@foo:bar"
|
user_id = "@foo:bar"
|
||||||
@ -420,8 +420,8 @@ class PresenceTimeoutTestCase(unittest.TestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
self.assertIsNotNone(new_state)
|
self.assertIsNotNone(new_state)
|
||||||
self.assertEquals(new_state.state, PresenceState.ONLINE)
|
self.assertEqual(new_state.state, PresenceState.ONLINE)
|
||||||
self.assertEquals(new_state.status_msg, status_msg)
|
self.assertEqual(new_state.status_msg, status_msg)
|
||||||
|
|
||||||
def test_federation_ping(self):
|
def test_federation_ping(self):
|
||||||
user_id = "@foo:bar"
|
user_id = "@foo:bar"
|
||||||
@ -440,7 +440,7 @@ class PresenceTimeoutTestCase(unittest.TestCase):
|
|||||||
new_state = handle_timeout(state, is_mine=True, syncing_user_ids=set(), now=now)
|
new_state = handle_timeout(state, is_mine=True, syncing_user_ids=set(), now=now)
|
||||||
|
|
||||||
self.assertIsNotNone(new_state)
|
self.assertIsNotNone(new_state)
|
||||||
self.assertEquals(state, new_state)
|
self.assertEqual(state, new_state)
|
||||||
|
|
||||||
def test_no_timeout(self):
|
def test_no_timeout(self):
|
||||||
user_id = "@foo:bar"
|
user_id = "@foo:bar"
|
||||||
@ -477,8 +477,8 @@ class PresenceTimeoutTestCase(unittest.TestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
self.assertIsNotNone(new_state)
|
self.assertIsNotNone(new_state)
|
||||||
self.assertEquals(new_state.state, PresenceState.OFFLINE)
|
self.assertEqual(new_state.state, PresenceState.OFFLINE)
|
||||||
self.assertEquals(new_state.status_msg, status_msg)
|
self.assertEqual(new_state.status_msg, status_msg)
|
||||||
|
|
||||||
def test_last_active(self):
|
def test_last_active(self):
|
||||||
user_id = "@foo:bar"
|
user_id = "@foo:bar"
|
||||||
@ -497,7 +497,7 @@ class PresenceTimeoutTestCase(unittest.TestCase):
|
|||||||
new_state = handle_timeout(state, is_mine=True, syncing_user_ids=set(), now=now)
|
new_state = handle_timeout(state, is_mine=True, syncing_user_ids=set(), now=now)
|
||||||
|
|
||||||
self.assertIsNotNone(new_state)
|
self.assertIsNotNone(new_state)
|
||||||
self.assertEquals(state, new_state)
|
self.assertEqual(state, new_state)
|
||||||
|
|
||||||
|
|
||||||
class PresenceHandlerTestCase(unittest.HomeserverTestCase):
|
class PresenceHandlerTestCase(unittest.HomeserverTestCase):
|
||||||
|
@ -65,7 +65,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
displayname = self.get_success(self.handler.get_displayname(self.frank))
|
displayname = self.get_success(self.handler.get_displayname(self.frank))
|
||||||
|
|
||||||
self.assertEquals("Frank", displayname)
|
self.assertEqual("Frank", displayname)
|
||||||
|
|
||||||
def test_set_my_name(self):
|
def test_set_my_name(self):
|
||||||
self.get_success(
|
self.get_success(
|
||||||
@ -74,7 +74,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
(
|
(
|
||||||
self.get_success(
|
self.get_success(
|
||||||
self.store.get_profile_displayname(self.frank.localpart)
|
self.store.get_profile_displayname(self.frank.localpart)
|
||||||
@ -90,7 +90,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
(
|
(
|
||||||
self.get_success(
|
self.get_success(
|
||||||
self.store.get_profile_displayname(self.frank.localpart)
|
self.store.get_profile_displayname(self.frank.localpart)
|
||||||
@ -118,7 +118,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
|
|||||||
self.store.set_profile_displayname(self.frank.localpart, "Frank")
|
self.store.set_profile_displayname(self.frank.localpart, "Frank")
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
(
|
(
|
||||||
self.get_success(
|
self.get_success(
|
||||||
self.store.get_profile_displayname(self.frank.localpart)
|
self.store.get_profile_displayname(self.frank.localpart)
|
||||||
@ -150,7 +150,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
displayname = self.get_success(self.handler.get_displayname(self.alice))
|
displayname = self.get_success(self.handler.get_displayname(self.alice))
|
||||||
|
|
||||||
self.assertEquals(displayname, "Alice")
|
self.assertEqual(displayname, "Alice")
|
||||||
self.mock_federation.make_query.assert_called_with(
|
self.mock_federation.make_query.assert_called_with(
|
||||||
destination="remote",
|
destination="remote",
|
||||||
query_type="profile",
|
query_type="profile",
|
||||||
@ -172,7 +172,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals({"displayname": "Caroline"}, response)
|
self.assertEqual({"displayname": "Caroline"}, response)
|
||||||
|
|
||||||
def test_get_my_avatar(self):
|
def test_get_my_avatar(self):
|
||||||
self.get_success(
|
self.get_success(
|
||||||
@ -182,7 +182,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
avatar_url = self.get_success(self.handler.get_avatar_url(self.frank))
|
avatar_url = self.get_success(self.handler.get_avatar_url(self.frank))
|
||||||
|
|
||||||
self.assertEquals("http://my.server/me.png", avatar_url)
|
self.assertEqual("http://my.server/me.png", avatar_url)
|
||||||
|
|
||||||
def test_set_my_avatar(self):
|
def test_set_my_avatar(self):
|
||||||
self.get_success(
|
self.get_success(
|
||||||
@ -193,7 +193,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
(self.get_success(self.store.get_profile_avatar_url(self.frank.localpart))),
|
(self.get_success(self.store.get_profile_avatar_url(self.frank.localpart))),
|
||||||
"http://my.server/pic.gif",
|
"http://my.server/pic.gif",
|
||||||
)
|
)
|
||||||
@ -207,7 +207,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
(self.get_success(self.store.get_profile_avatar_url(self.frank.localpart))),
|
(self.get_success(self.store.get_profile_avatar_url(self.frank.localpart))),
|
||||||
"http://my.server/me.png",
|
"http://my.server/me.png",
|
||||||
)
|
)
|
||||||
@ -235,7 +235,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
(self.get_success(self.store.get_profile_avatar_url(self.frank.localpart))),
|
(self.get_success(self.store.get_profile_avatar_url(self.frank.localpart))),
|
||||||
"http://my.server/me.png",
|
"http://my.server/me.png",
|
||||||
)
|
)
|
||||||
|
@ -314,4 +314,4 @@ class ReceiptsTestCase(unittest.HomeserverTestCase):
|
|||||||
):
|
):
|
||||||
"""Tests that the _filter_out_hidden returns the expected output"""
|
"""Tests that the _filter_out_hidden returns the expected output"""
|
||||||
filtered_events = self.event_source.filter_out_hidden(events, "@me:server.org")
|
filtered_events = self.event_source.filter_out_hidden(events, "@me:server.org")
|
||||||
self.assertEquals(filtered_events, expected_output)
|
self.assertEqual(filtered_events, expected_output)
|
||||||
|
@ -167,7 +167,7 @@ class RegistrationTestCase(unittest.HomeserverTestCase):
|
|||||||
result_user_id, result_token = self.get_success(
|
result_user_id, result_token = self.get_success(
|
||||||
self.get_or_create_user(requester, frank.localpart, "Frankie")
|
self.get_or_create_user(requester, frank.localpart, "Frankie")
|
||||||
)
|
)
|
||||||
self.assertEquals(result_user_id, user_id)
|
self.assertEqual(result_user_id, user_id)
|
||||||
self.assertIsInstance(result_token, str)
|
self.assertIsInstance(result_token, str)
|
||||||
self.assertGreater(len(result_token), 20)
|
self.assertGreater(len(result_token), 20)
|
||||||
|
|
||||||
@ -183,7 +183,7 @@ class RegistrationTestCase(unittest.HomeserverTestCase):
|
|||||||
result_user_id, result_token = self.get_success(
|
result_user_id, result_token = self.get_success(
|
||||||
self.get_or_create_user(requester, local_part, None)
|
self.get_or_create_user(requester, local_part, None)
|
||||||
)
|
)
|
||||||
self.assertEquals(result_user_id, user_id)
|
self.assertEqual(result_user_id, user_id)
|
||||||
self.assertTrue(result_token is not None)
|
self.assertTrue(result_token is not None)
|
||||||
|
|
||||||
@override_config({"limit_usage_by_mau": False})
|
@override_config({"limit_usage_by_mau": False})
|
||||||
|
@ -69,7 +69,7 @@ class SyncTestCase(tests.unittest.HomeserverTestCase):
|
|||||||
self.sync_handler.wait_for_sync_for_user(requester, sync_config),
|
self.sync_handler.wait_for_sync_for_user(requester, sync_config),
|
||||||
ResourceLimitError,
|
ResourceLimitError,
|
||||||
)
|
)
|
||||||
self.assertEquals(e.value.errcode, Codes.RESOURCE_LIMIT_EXCEEDED)
|
self.assertEqual(e.value.errcode, Codes.RESOURCE_LIMIT_EXCEEDED)
|
||||||
|
|
||||||
self.auth_blocking._hs_disabled = False
|
self.auth_blocking._hs_disabled = False
|
||||||
|
|
||||||
@ -80,7 +80,7 @@ class SyncTestCase(tests.unittest.HomeserverTestCase):
|
|||||||
self.sync_handler.wait_for_sync_for_user(requester, sync_config),
|
self.sync_handler.wait_for_sync_for_user(requester, sync_config),
|
||||||
ResourceLimitError,
|
ResourceLimitError,
|
||||||
)
|
)
|
||||||
self.assertEquals(e.value.errcode, Codes.RESOURCE_LIMIT_EXCEEDED)
|
self.assertEqual(e.value.errcode, Codes.RESOURCE_LIMIT_EXCEEDED)
|
||||||
|
|
||||||
def test_unknown_room_version(self):
|
def test_unknown_room_version(self):
|
||||||
"""
|
"""
|
||||||
@ -122,7 +122,7 @@ class SyncTestCase(tests.unittest.HomeserverTestCase):
|
|||||||
b"{}",
|
b"{}",
|
||||||
tok,
|
tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
# The rooms should appear in the sync response.
|
# The rooms should appear in the sync response.
|
||||||
result = self.get_success(
|
result = self.get_success(
|
||||||
|
@ -156,7 +156,7 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
def test_started_typing_local(self):
|
def test_started_typing_local(self):
|
||||||
self.room_members = [U_APPLE, U_BANANA]
|
self.room_members = [U_APPLE, U_BANANA]
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 0)
|
self.assertEqual(self.event_source.get_current_key(), 0)
|
||||||
|
|
||||||
self.get_success(
|
self.get_success(
|
||||||
self.handler.started_typing(
|
self.handler.started_typing(
|
||||||
@ -169,13 +169,13 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
self.on_new_event.assert_has_calls([call("typing_key", 1, rooms=[ROOM_ID])])
|
self.on_new_event.assert_has_calls([call("typing_key", 1, rooms=[ROOM_ID])])
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 1)
|
self.assertEqual(self.event_source.get_current_key(), 1)
|
||||||
events = self.get_success(
|
events = self.get_success(
|
||||||
self.event_source.get_new_events(
|
self.event_source.get_new_events(
|
||||||
user=U_APPLE, from_key=0, limit=None, room_ids=[ROOM_ID], is_guest=False
|
user=U_APPLE, from_key=0, limit=None, room_ids=[ROOM_ID], is_guest=False
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
events[0],
|
events[0],
|
||||||
[
|
[
|
||||||
{
|
{
|
||||||
@ -220,7 +220,7 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
def test_started_typing_remote_recv(self):
|
def test_started_typing_remote_recv(self):
|
||||||
self.room_members = [U_APPLE, U_ONION]
|
self.room_members = [U_APPLE, U_ONION]
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 0)
|
self.assertEqual(self.event_source.get_current_key(), 0)
|
||||||
|
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"PUT",
|
"PUT",
|
||||||
@ -239,13 +239,13 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
self.on_new_event.assert_has_calls([call("typing_key", 1, rooms=[ROOM_ID])])
|
self.on_new_event.assert_has_calls([call("typing_key", 1, rooms=[ROOM_ID])])
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 1)
|
self.assertEqual(self.event_source.get_current_key(), 1)
|
||||||
events = self.get_success(
|
events = self.get_success(
|
||||||
self.event_source.get_new_events(
|
self.event_source.get_new_events(
|
||||||
user=U_APPLE, from_key=0, limit=None, room_ids=[ROOM_ID], is_guest=False
|
user=U_APPLE, from_key=0, limit=None, room_ids=[ROOM_ID], is_guest=False
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
events[0],
|
events[0],
|
||||||
[
|
[
|
||||||
{
|
{
|
||||||
@ -259,7 +259,7 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
def test_started_typing_remote_recv_not_in_room(self):
|
def test_started_typing_remote_recv_not_in_room(self):
|
||||||
self.room_members = [U_APPLE, U_ONION]
|
self.room_members = [U_APPLE, U_ONION]
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 0)
|
self.assertEqual(self.event_source.get_current_key(), 0)
|
||||||
|
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"PUT",
|
"PUT",
|
||||||
@ -278,7 +278,7 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
self.on_new_event.assert_not_called()
|
self.on_new_event.assert_not_called()
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 0)
|
self.assertEqual(self.event_source.get_current_key(), 0)
|
||||||
events = self.get_success(
|
events = self.get_success(
|
||||||
self.event_source.get_new_events(
|
self.event_source.get_new_events(
|
||||||
user=U_APPLE,
|
user=U_APPLE,
|
||||||
@ -288,8 +288,8 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
is_guest=False,
|
is_guest=False,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(events[0], [])
|
self.assertEqual(events[0], [])
|
||||||
self.assertEquals(events[1], 0)
|
self.assertEqual(events[1], 0)
|
||||||
|
|
||||||
@override_config({"send_federation": True})
|
@override_config({"send_federation": True})
|
||||||
def test_stopped_typing(self):
|
def test_stopped_typing(self):
|
||||||
@ -302,7 +302,7 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
self.handler._member_typing_until[member] = 1002000
|
self.handler._member_typing_until[member] = 1002000
|
||||||
self.handler._room_typing[ROOM_ID] = {U_APPLE.to_string()}
|
self.handler._room_typing[ROOM_ID] = {U_APPLE.to_string()}
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 0)
|
self.assertEqual(self.event_source.get_current_key(), 0)
|
||||||
|
|
||||||
self.get_success(
|
self.get_success(
|
||||||
self.handler.stopped_typing(
|
self.handler.stopped_typing(
|
||||||
@ -332,13 +332,13 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
try_trailing_slash_on_400=True,
|
try_trailing_slash_on_400=True,
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 1)
|
self.assertEqual(self.event_source.get_current_key(), 1)
|
||||||
events = self.get_success(
|
events = self.get_success(
|
||||||
self.event_source.get_new_events(
|
self.event_source.get_new_events(
|
||||||
user=U_APPLE, from_key=0, limit=None, room_ids=[ROOM_ID], is_guest=False
|
user=U_APPLE, from_key=0, limit=None, room_ids=[ROOM_ID], is_guest=False
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
events[0],
|
events[0],
|
||||||
[{"type": "m.typing", "room_id": ROOM_ID, "content": {"user_ids": []}}],
|
[{"type": "m.typing", "room_id": ROOM_ID, "content": {"user_ids": []}}],
|
||||||
)
|
)
|
||||||
@ -346,7 +346,7 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
def test_typing_timeout(self):
|
def test_typing_timeout(self):
|
||||||
self.room_members = [U_APPLE, U_BANANA]
|
self.room_members = [U_APPLE, U_BANANA]
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 0)
|
self.assertEqual(self.event_source.get_current_key(), 0)
|
||||||
|
|
||||||
self.get_success(
|
self.get_success(
|
||||||
self.handler.started_typing(
|
self.handler.started_typing(
|
||||||
@ -360,7 +360,7 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
self.on_new_event.assert_has_calls([call("typing_key", 1, rooms=[ROOM_ID])])
|
self.on_new_event.assert_has_calls([call("typing_key", 1, rooms=[ROOM_ID])])
|
||||||
self.on_new_event.reset_mock()
|
self.on_new_event.reset_mock()
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 1)
|
self.assertEqual(self.event_source.get_current_key(), 1)
|
||||||
events = self.get_success(
|
events = self.get_success(
|
||||||
self.event_source.get_new_events(
|
self.event_source.get_new_events(
|
||||||
user=U_APPLE,
|
user=U_APPLE,
|
||||||
@ -370,7 +370,7 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
is_guest=False,
|
is_guest=False,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
events[0],
|
events[0],
|
||||||
[
|
[
|
||||||
{
|
{
|
||||||
@ -385,7 +385,7 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
self.on_new_event.assert_has_calls([call("typing_key", 2, rooms=[ROOM_ID])])
|
self.on_new_event.assert_has_calls([call("typing_key", 2, rooms=[ROOM_ID])])
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 2)
|
self.assertEqual(self.event_source.get_current_key(), 2)
|
||||||
events = self.get_success(
|
events = self.get_success(
|
||||||
self.event_source.get_new_events(
|
self.event_source.get_new_events(
|
||||||
user=U_APPLE,
|
user=U_APPLE,
|
||||||
@ -395,7 +395,7 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
is_guest=False,
|
is_guest=False,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
events[0],
|
events[0],
|
||||||
[{"type": "m.typing", "room_id": ROOM_ID, "content": {"user_ids": []}}],
|
[{"type": "m.typing", "room_id": ROOM_ID, "content": {"user_ids": []}}],
|
||||||
)
|
)
|
||||||
@ -414,7 +414,7 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
self.on_new_event.assert_has_calls([call("typing_key", 3, rooms=[ROOM_ID])])
|
self.on_new_event.assert_has_calls([call("typing_key", 3, rooms=[ROOM_ID])])
|
||||||
self.on_new_event.reset_mock()
|
self.on_new_event.reset_mock()
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 3)
|
self.assertEqual(self.event_source.get_current_key(), 3)
|
||||||
events = self.get_success(
|
events = self.get_success(
|
||||||
self.event_source.get_new_events(
|
self.event_source.get_new_events(
|
||||||
user=U_APPLE,
|
user=U_APPLE,
|
||||||
@ -424,7 +424,7 @@ class TypingNotificationsTestCase(unittest.HomeserverTestCase):
|
|||||||
is_guest=False,
|
is_guest=False,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
events[0],
|
events[0],
|
||||||
[
|
[
|
||||||
{
|
{
|
||||||
|
@ -1042,7 +1042,7 @@ class TestUserDirSearchDisabled(unittest.HomeserverTestCase):
|
|||||||
b'{"search_term":"user2"}',
|
b'{"search_term":"user2"}',
|
||||||
access_token=u1_token,
|
access_token=u1_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertTrue(len(channel.json_body["results"]) > 0)
|
self.assertTrue(len(channel.json_body["results"]) > 0)
|
||||||
|
|
||||||
# Disable user directory and check search returns nothing
|
# Disable user directory and check search returns nothing
|
||||||
@ -1053,5 +1053,5 @@ class TestUserDirSearchDisabled(unittest.HomeserverTestCase):
|
|||||||
b'{"search_term":"user2"}',
|
b'{"search_term":"user2"}',
|
||||||
access_token=u1_token,
|
access_token=u1_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertTrue(len(channel.json_body["results"]) == 0)
|
self.assertTrue(len(channel.json_body["results"]) == 0)
|
||||||
|
@ -65,9 +65,9 @@ class SrvResolverTestCase(unittest.TestCase):
|
|||||||
|
|
||||||
servers = self.successResultOf(test_d)
|
servers = self.successResultOf(test_d)
|
||||||
|
|
||||||
self.assertEquals(len(servers), 1)
|
self.assertEqual(len(servers), 1)
|
||||||
self.assertEquals(servers, cache[service_name])
|
self.assertEqual(servers, cache[service_name])
|
||||||
self.assertEquals(servers[0].host, host_name)
|
self.assertEqual(servers[0].host, host_name)
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def test_from_cache_expired_and_dns_fail(self):
|
def test_from_cache_expired_and_dns_fail(self):
|
||||||
@ -88,8 +88,8 @@ class SrvResolverTestCase(unittest.TestCase):
|
|||||||
|
|
||||||
dns_client_mock.lookupService.assert_called_once_with(service_name)
|
dns_client_mock.lookupService.assert_called_once_with(service_name)
|
||||||
|
|
||||||
self.assertEquals(len(servers), 1)
|
self.assertEqual(len(servers), 1)
|
||||||
self.assertEquals(servers, cache[service_name])
|
self.assertEqual(servers, cache[service_name])
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def test_from_cache(self):
|
def test_from_cache(self):
|
||||||
@ -114,8 +114,8 @@ class SrvResolverTestCase(unittest.TestCase):
|
|||||||
|
|
||||||
self.assertFalse(dns_client_mock.lookupService.called)
|
self.assertFalse(dns_client_mock.lookupService.called)
|
||||||
|
|
||||||
self.assertEquals(len(servers), 1)
|
self.assertEqual(len(servers), 1)
|
||||||
self.assertEquals(servers, cache[service_name])
|
self.assertEqual(servers, cache[service_name])
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def test_empty_cache(self):
|
def test_empty_cache(self):
|
||||||
@ -144,8 +144,8 @@ class SrvResolverTestCase(unittest.TestCase):
|
|||||||
|
|
||||||
servers = yield defer.ensureDeferred(resolver.resolve_service(service_name))
|
servers = yield defer.ensureDeferred(resolver.resolve_service(service_name))
|
||||||
|
|
||||||
self.assertEquals(len(servers), 0)
|
self.assertEqual(len(servers), 0)
|
||||||
self.assertEquals(len(cache), 0)
|
self.assertEqual(len(cache), 0)
|
||||||
|
|
||||||
def test_disabled_service(self):
|
def test_disabled_service(self):
|
||||||
"""
|
"""
|
||||||
@ -201,6 +201,6 @@ class SrvResolverTestCase(unittest.TestCase):
|
|||||||
|
|
||||||
servers = self.successResultOf(resolve_d)
|
servers = self.successResultOf(resolve_d)
|
||||||
|
|
||||||
self.assertEquals(len(servers), 1)
|
self.assertEqual(len(servers), 1)
|
||||||
self.assertEquals(servers, cache[service_name])
|
self.assertEqual(servers, cache[service_name])
|
||||||
self.assertEquals(servers[0].host, b"host")
|
self.assertEqual(servers[0].host, b"host")
|
||||||
|
@ -59,7 +59,7 @@ class SlavedEventStoreTestCase(BaseSlavedStoreTestCase):
|
|||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
# Patch up the equality operator for events so that we can check
|
# Patch up the equality operator for events so that we can check
|
||||||
# whether lists of events match using assertEquals
|
# whether lists of events match using assertEqual
|
||||||
self.unpatches = [patch__eq__(_EventInternalMetadata), patch__eq__(FrozenEvent)]
|
self.unpatches = [patch__eq__(_EventInternalMetadata), patch__eq__(FrozenEvent)]
|
||||||
return super().setUp()
|
return super().setUp()
|
||||||
|
|
||||||
|
@ -1909,7 +1909,7 @@ class JoinAliasRoomTestCase(unittest.HomeserverTestCase):
|
|||||||
"/_matrix/client/r0/joined_rooms",
|
"/_matrix/client/r0/joined_rooms",
|
||||||
access_token=self.second_tok,
|
access_token=self.second_tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(HTTPStatus.OK, channel.code, msg=channel.json_body)
|
self.assertEqual(HTTPStatus.OK, channel.code, msg=channel.json_body)
|
||||||
self.assertEqual(self.public_room_id, channel.json_body["joined_rooms"][0])
|
self.assertEqual(self.public_room_id, channel.json_body["joined_rooms"][0])
|
||||||
|
|
||||||
def test_join_private_room_if_not_member(self) -> None:
|
def test_join_private_room_if_not_member(self) -> None:
|
||||||
@ -1957,7 +1957,7 @@ class JoinAliasRoomTestCase(unittest.HomeserverTestCase):
|
|||||||
"/_matrix/client/r0/joined_rooms",
|
"/_matrix/client/r0/joined_rooms",
|
||||||
access_token=self.admin_user_tok,
|
access_token=self.admin_user_tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(HTTPStatus.OK, channel.code, msg=channel.json_body)
|
self.assertEqual(HTTPStatus.OK, channel.code, msg=channel.json_body)
|
||||||
self.assertEqual(private_room_id, channel.json_body["joined_rooms"][0])
|
self.assertEqual(private_room_id, channel.json_body["joined_rooms"][0])
|
||||||
|
|
||||||
# Join user to room.
|
# Join user to room.
|
||||||
@ -1980,7 +1980,7 @@ class JoinAliasRoomTestCase(unittest.HomeserverTestCase):
|
|||||||
"/_matrix/client/r0/joined_rooms",
|
"/_matrix/client/r0/joined_rooms",
|
||||||
access_token=self.second_tok,
|
access_token=self.second_tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(HTTPStatus.OK, channel.code, msg=channel.json_body)
|
self.assertEqual(HTTPStatus.OK, channel.code, msg=channel.json_body)
|
||||||
self.assertEqual(private_room_id, channel.json_body["joined_rooms"][0])
|
self.assertEqual(private_room_id, channel.json_body["joined_rooms"][0])
|
||||||
|
|
||||||
def test_join_private_room_if_owner(self) -> None:
|
def test_join_private_room_if_owner(self) -> None:
|
||||||
@ -2010,7 +2010,7 @@ class JoinAliasRoomTestCase(unittest.HomeserverTestCase):
|
|||||||
"/_matrix/client/r0/joined_rooms",
|
"/_matrix/client/r0/joined_rooms",
|
||||||
access_token=self.second_tok,
|
access_token=self.second_tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(HTTPStatus.OK, channel.code, msg=channel.json_body)
|
self.assertEqual(HTTPStatus.OK, channel.code, msg=channel.json_body)
|
||||||
self.assertEqual(private_room_id, channel.json_body["joined_rooms"][0])
|
self.assertEqual(private_room_id, channel.json_body["joined_rooms"][0])
|
||||||
|
|
||||||
def test_context_as_non_admin(self) -> None:
|
def test_context_as_non_admin(self) -> None:
|
||||||
@ -2044,7 +2044,7 @@ class JoinAliasRoomTestCase(unittest.HomeserverTestCase):
|
|||||||
% (room_id, events[midway]["event_id"]),
|
% (room_id, events[midway]["event_id"]),
|
||||||
access_token=tok,
|
access_token=tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(HTTPStatus.FORBIDDEN, channel.code, msg=channel.json_body)
|
self.assertEqual(HTTPStatus.FORBIDDEN, channel.code, msg=channel.json_body)
|
||||||
self.assertEqual(Codes.FORBIDDEN, channel.json_body["errcode"])
|
self.assertEqual(Codes.FORBIDDEN, channel.json_body["errcode"])
|
||||||
|
|
||||||
def test_context_as_admin(self) -> None:
|
def test_context_as_admin(self) -> None:
|
||||||
@ -2074,8 +2074,8 @@ class JoinAliasRoomTestCase(unittest.HomeserverTestCase):
|
|||||||
% (room_id, events[midway]["event_id"]),
|
% (room_id, events[midway]["event_id"]),
|
||||||
access_token=self.admin_user_tok,
|
access_token=self.admin_user_tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(HTTPStatus.OK, channel.code, msg=channel.json_body)
|
self.assertEqual(HTTPStatus.OK, channel.code, msg=channel.json_body)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
channel.json_body["event"]["event_id"], events[midway]["event_id"]
|
channel.json_body["event"]["event_id"], events[midway]["event_id"]
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -104,7 +104,7 @@ class PasswordResetTestCase(unittest.HomeserverTestCase):
|
|||||||
client_secret = "foobar"
|
client_secret = "foobar"
|
||||||
session_id = self._request_token(email, client_secret)
|
session_id = self._request_token(email, client_secret)
|
||||||
|
|
||||||
self.assertEquals(len(self.email_attempts), 1)
|
self.assertEqual(len(self.email_attempts), 1)
|
||||||
link = self._get_link_from_email()
|
link = self._get_link_from_email()
|
||||||
|
|
||||||
self._validate_token(link)
|
self._validate_token(link)
|
||||||
@ -143,7 +143,7 @@ class PasswordResetTestCase(unittest.HomeserverTestCase):
|
|||||||
client_secret = "foobar"
|
client_secret = "foobar"
|
||||||
session_id = self._request_token(email, client_secret, ip)
|
session_id = self._request_token(email, client_secret, ip)
|
||||||
|
|
||||||
self.assertEquals(len(self.email_attempts), 1)
|
self.assertEqual(len(self.email_attempts), 1)
|
||||||
link = self._get_link_from_email()
|
link = self._get_link_from_email()
|
||||||
|
|
||||||
self._validate_token(link)
|
self._validate_token(link)
|
||||||
@ -193,7 +193,7 @@ class PasswordResetTestCase(unittest.HomeserverTestCase):
|
|||||||
client_secret = "foobar"
|
client_secret = "foobar"
|
||||||
session_id = self._request_token(email_passwort_reset, client_secret)
|
session_id = self._request_token(email_passwort_reset, client_secret)
|
||||||
|
|
||||||
self.assertEquals(len(self.email_attempts), 1)
|
self.assertEqual(len(self.email_attempts), 1)
|
||||||
link = self._get_link_from_email()
|
link = self._get_link_from_email()
|
||||||
|
|
||||||
self._validate_token(link)
|
self._validate_token(link)
|
||||||
@ -230,7 +230,7 @@ class PasswordResetTestCase(unittest.HomeserverTestCase):
|
|||||||
client_secret = "foobar"
|
client_secret = "foobar"
|
||||||
session_id = self._request_token(email, client_secret)
|
session_id = self._request_token(email, client_secret)
|
||||||
|
|
||||||
self.assertEquals(len(self.email_attempts), 1)
|
self.assertEqual(len(self.email_attempts), 1)
|
||||||
|
|
||||||
# Attempt to reset password without clicking the link
|
# Attempt to reset password without clicking the link
|
||||||
self._reset_password(new_password, session_id, client_secret, expected_code=401)
|
self._reset_password(new_password, session_id, client_secret, expected_code=401)
|
||||||
@ -322,7 +322,7 @@ class PasswordResetTestCase(unittest.HomeserverTestCase):
|
|||||||
shorthand=False,
|
shorthand=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
# Now POST to the same endpoint, mimicking the same behaviour as clicking the
|
# Now POST to the same endpoint, mimicking the same behaviour as clicking the
|
||||||
# password reset confirm button
|
# password reset confirm button
|
||||||
@ -337,7 +337,7 @@ class PasswordResetTestCase(unittest.HomeserverTestCase):
|
|||||||
shorthand=False,
|
shorthand=False,
|
||||||
content_is_form=True,
|
content_is_form=True,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
def _get_link_from_email(self):
|
def _get_link_from_email(self):
|
||||||
assert self.email_attempts, "No emails have been sent"
|
assert self.email_attempts, "No emails have been sent"
|
||||||
@ -376,7 +376,7 @@ class PasswordResetTestCase(unittest.HomeserverTestCase):
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
self.assertEquals(expected_code, channel.code, channel.result)
|
self.assertEqual(expected_code, channel.code, channel.result)
|
||||||
|
|
||||||
|
|
||||||
class DeactivateTestCase(unittest.HomeserverTestCase):
|
class DeactivateTestCase(unittest.HomeserverTestCase):
|
||||||
@ -676,7 +676,7 @@ class ThreepidEmailRestTestCase(unittest.HomeserverTestCase):
|
|||||||
client_secret = "foobar"
|
client_secret = "foobar"
|
||||||
session_id = self._request_token(self.email, client_secret)
|
session_id = self._request_token(self.email, client_secret)
|
||||||
|
|
||||||
self.assertEquals(len(self.email_attempts), 1)
|
self.assertEqual(len(self.email_attempts), 1)
|
||||||
link = self._get_link_from_email()
|
link = self._get_link_from_email()
|
||||||
|
|
||||||
self._validate_token(link)
|
self._validate_token(link)
|
||||||
@ -780,7 +780,7 @@ class ThreepidEmailRestTestCase(unittest.HomeserverTestCase):
|
|||||||
client_secret = "foobar"
|
client_secret = "foobar"
|
||||||
session_id = self._request_token(self.email, client_secret)
|
session_id = self._request_token(self.email, client_secret)
|
||||||
|
|
||||||
self.assertEquals(len(self.email_attempts), 1)
|
self.assertEqual(len(self.email_attempts), 1)
|
||||||
|
|
||||||
# Attempt to add email without clicking the link
|
# Attempt to add email without clicking the link
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -981,7 +981,7 @@ class ThreepidEmailRestTestCase(unittest.HomeserverTestCase):
|
|||||||
path = link.replace("https://example.com", "")
|
path = link.replace("https://example.com", "")
|
||||||
|
|
||||||
channel = self.make_request("GET", path, shorthand=False)
|
channel = self.make_request("GET", path, shorthand=False)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
def _get_link_from_email(self):
|
def _get_link_from_email(self):
|
||||||
assert self.email_attempts, "No emails have been sent"
|
assert self.email_attempts, "No emails have been sent"
|
||||||
@ -1010,7 +1010,7 @@ class ThreepidEmailRestTestCase(unittest.HomeserverTestCase):
|
|||||||
client_secret = "foobar"
|
client_secret = "foobar"
|
||||||
session_id = self._request_token(request_email, client_secret)
|
session_id = self._request_token(request_email, client_secret)
|
||||||
|
|
||||||
self.assertEquals(len(self.email_attempts) - previous_email_attempts, 1)
|
self.assertEqual(len(self.email_attempts) - previous_email_attempts, 1)
|
||||||
link = self._get_link_from_email()
|
link = self._get_link_from_email()
|
||||||
|
|
||||||
self._validate_token(link)
|
self._validate_token(link)
|
||||||
|
@ -65,13 +65,13 @@ class EventStreamPermissionsTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET", "/events?access_token=%s" % ("invalid" + self.token,)
|
"GET", "/events?access_token=%s" % ("invalid" + self.token,)
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.code, 401, msg=channel.result)
|
self.assertEqual(channel.code, 401, msg=channel.result)
|
||||||
|
|
||||||
# valid token, expect content
|
# valid token, expect content
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET", "/events?access_token=%s&timeout=0" % (self.token,)
|
"GET", "/events?access_token=%s&timeout=0" % (self.token,)
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.code, 200, msg=channel.result)
|
self.assertEqual(channel.code, 200, msg=channel.result)
|
||||||
self.assertTrue("chunk" in channel.json_body)
|
self.assertTrue("chunk" in channel.json_body)
|
||||||
self.assertTrue("start" in channel.json_body)
|
self.assertTrue("start" in channel.json_body)
|
||||||
self.assertTrue("end" in channel.json_body)
|
self.assertTrue("end" in channel.json_body)
|
||||||
@ -89,10 +89,10 @@ class EventStreamPermissionsTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET", "/events?access_token=%s&timeout=0" % (self.token,)
|
"GET", "/events?access_token=%s&timeout=0" % (self.token,)
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.code, 200, msg=channel.result)
|
self.assertEqual(channel.code, 200, msg=channel.result)
|
||||||
|
|
||||||
# We may get a presence event for ourselves down
|
# We may get a presence event for ourselves down
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
0,
|
0,
|
||||||
len(
|
len(
|
||||||
[
|
[
|
||||||
@ -153,4 +153,4 @@ class GetEventsTestCase(unittest.HomeserverTestCase):
|
|||||||
"/events/" + event_id,
|
"/events/" + event_id,
|
||||||
access_token=self.token,
|
access_token=self.token,
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.code, 200, msg=channel.result)
|
self.assertEqual(channel.code, 200, msg=channel.result)
|
||||||
|
@ -45,7 +45,7 @@ class FilterTestCase(unittest.HomeserverTestCase):
|
|||||||
self.assertEqual(channel.json_body, {"filter_id": "0"})
|
self.assertEqual(channel.json_body, {"filter_id": "0"})
|
||||||
filter = self.store.get_user_filter(user_localpart="apple", filter_id=0)
|
filter = self.store.get_user_filter(user_localpart="apple", filter_id=0)
|
||||||
self.pump()
|
self.pump()
|
||||||
self.assertEquals(filter.result, self.EXAMPLE_FILTER)
|
self.assertEqual(filter.result, self.EXAMPLE_FILTER)
|
||||||
|
|
||||||
def test_add_filter_for_other_user(self):
|
def test_add_filter_for_other_user(self):
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -55,7 +55,7 @@ class FilterTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
self.assertEqual(channel.result["code"], b"403")
|
self.assertEqual(channel.result["code"], b"403")
|
||||||
self.assertEquals(channel.json_body["errcode"], Codes.FORBIDDEN)
|
self.assertEqual(channel.json_body["errcode"], Codes.FORBIDDEN)
|
||||||
|
|
||||||
def test_add_filter_non_local_user(self):
|
def test_add_filter_non_local_user(self):
|
||||||
_is_mine = self.hs.is_mine
|
_is_mine = self.hs.is_mine
|
||||||
@ -68,7 +68,7 @@ class FilterTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
self.hs.is_mine = _is_mine
|
self.hs.is_mine = _is_mine
|
||||||
self.assertEqual(channel.result["code"], b"403")
|
self.assertEqual(channel.result["code"], b"403")
|
||||||
self.assertEquals(channel.json_body["errcode"], Codes.FORBIDDEN)
|
self.assertEqual(channel.json_body["errcode"], Codes.FORBIDDEN)
|
||||||
|
|
||||||
def test_get_filter(self):
|
def test_get_filter(self):
|
||||||
filter_id = defer.ensureDeferred(
|
filter_id = defer.ensureDeferred(
|
||||||
@ -83,7 +83,7 @@ class FilterTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
self.assertEqual(channel.result["code"], b"200")
|
self.assertEqual(channel.result["code"], b"200")
|
||||||
self.assertEquals(channel.json_body, self.EXAMPLE_FILTER)
|
self.assertEqual(channel.json_body, self.EXAMPLE_FILTER)
|
||||||
|
|
||||||
def test_get_filter_non_existant(self):
|
def test_get_filter_non_existant(self):
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -91,7 +91,7 @@ class FilterTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
self.assertEqual(channel.result["code"], b"404")
|
self.assertEqual(channel.result["code"], b"404")
|
||||||
self.assertEquals(channel.json_body["errcode"], Codes.NOT_FOUND)
|
self.assertEqual(channel.json_body["errcode"], Codes.NOT_FOUND)
|
||||||
|
|
||||||
# Currently invalid params do not have an appropriate errcode
|
# Currently invalid params do not have an appropriate errcode
|
||||||
# in errors.py
|
# in errors.py
|
||||||
|
@ -30,8 +30,8 @@ class GroupsTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# Alice creates a group
|
# Alice creates a group
|
||||||
channel = self.make_request("POST", "/create_group", {"localpart": "spqr"})
|
channel = self.make_request("POST", "/create_group", {"localpart": "spqr"})
|
||||||
self.assertEquals(channel.code, 200, msg=channel.text_body)
|
self.assertEqual(channel.code, 200, msg=channel.text_body)
|
||||||
self.assertEquals(channel.json_body, {"group_id": group_id})
|
self.assertEqual(channel.json_body, {"group_id": group_id})
|
||||||
|
|
||||||
# Bob creates a private room
|
# Bob creates a private room
|
||||||
room_id = self.helper.create_room_as(self.room_creator_user_id, is_public=False)
|
room_id = self.helper.create_room_as(self.room_creator_user_id, is_public=False)
|
||||||
@ -45,12 +45,12 @@ class GroupsTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"PUT", f"/groups/{group_id}/admin/rooms/{room_id}", {}
|
"PUT", f"/groups/{group_id}/admin/rooms/{room_id}", {}
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.code, 200, msg=channel.text_body)
|
self.assertEqual(channel.code, 200, msg=channel.text_body)
|
||||||
self.assertEquals(channel.json_body, {})
|
self.assertEqual(channel.json_body, {})
|
||||||
|
|
||||||
# Alice now tries to retrieve the room list of the space.
|
# Alice now tries to retrieve the room list of the space.
|
||||||
channel = self.make_request("GET", f"/groups/{group_id}/rooms")
|
channel = self.make_request("GET", f"/groups/{group_id}/rooms")
|
||||||
self.assertEquals(channel.code, 200, msg=channel.text_body)
|
self.assertEqual(channel.code, 200, msg=channel.text_body)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
channel.json_body, {"chunk": [], "total_room_count_estimate": 0}
|
channel.json_body, {"chunk": [], "total_room_count_estimate": 0}
|
||||||
)
|
)
|
||||||
|
@ -136,10 +136,10 @@ class LoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(b"POST", LOGIN_URL, params)
|
channel = self.make_request(b"POST", LOGIN_URL, params)
|
||||||
|
|
||||||
if i == 5:
|
if i == 5:
|
||||||
self.assertEquals(channel.result["code"], b"429", channel.result)
|
self.assertEqual(channel.result["code"], b"429", channel.result)
|
||||||
retry_after_ms = int(channel.json_body["retry_after_ms"])
|
retry_after_ms = int(channel.json_body["retry_after_ms"])
|
||||||
else:
|
else:
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
# Since we're ratelimiting at 1 request/min, retry_after_ms should be lower
|
# Since we're ratelimiting at 1 request/min, retry_after_ms should be lower
|
||||||
# than 1min.
|
# than 1min.
|
||||||
@ -154,7 +154,7 @@ class LoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
}
|
}
|
||||||
channel = self.make_request(b"POST", LOGIN_URL, params)
|
channel = self.make_request(b"POST", LOGIN_URL, params)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
@override_config(
|
@override_config(
|
||||||
{
|
{
|
||||||
@ -181,10 +181,10 @@ class LoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(b"POST", LOGIN_URL, params)
|
channel = self.make_request(b"POST", LOGIN_URL, params)
|
||||||
|
|
||||||
if i == 5:
|
if i == 5:
|
||||||
self.assertEquals(channel.result["code"], b"429", channel.result)
|
self.assertEqual(channel.result["code"], b"429", channel.result)
|
||||||
retry_after_ms = int(channel.json_body["retry_after_ms"])
|
retry_after_ms = int(channel.json_body["retry_after_ms"])
|
||||||
else:
|
else:
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
# Since we're ratelimiting at 1 request/min, retry_after_ms should be lower
|
# Since we're ratelimiting at 1 request/min, retry_after_ms should be lower
|
||||||
# than 1min.
|
# than 1min.
|
||||||
@ -199,7 +199,7 @@ class LoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
}
|
}
|
||||||
channel = self.make_request(b"POST", LOGIN_URL, params)
|
channel = self.make_request(b"POST", LOGIN_URL, params)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
@override_config(
|
@override_config(
|
||||||
{
|
{
|
||||||
@ -226,10 +226,10 @@ class LoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(b"POST", LOGIN_URL, params)
|
channel = self.make_request(b"POST", LOGIN_URL, params)
|
||||||
|
|
||||||
if i == 5:
|
if i == 5:
|
||||||
self.assertEquals(channel.result["code"], b"429", channel.result)
|
self.assertEqual(channel.result["code"], b"429", channel.result)
|
||||||
retry_after_ms = int(channel.json_body["retry_after_ms"])
|
retry_after_ms = int(channel.json_body["retry_after_ms"])
|
||||||
else:
|
else:
|
||||||
self.assertEquals(channel.result["code"], b"403", channel.result)
|
self.assertEqual(channel.result["code"], b"403", channel.result)
|
||||||
|
|
||||||
# Since we're ratelimiting at 1 request/min, retry_after_ms should be lower
|
# Since we're ratelimiting at 1 request/min, retry_after_ms should be lower
|
||||||
# than 1min.
|
# than 1min.
|
||||||
@ -244,7 +244,7 @@ class LoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
}
|
}
|
||||||
channel = self.make_request(b"POST", LOGIN_URL, params)
|
channel = self.make_request(b"POST", LOGIN_URL, params)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"403", channel.result)
|
self.assertEqual(channel.result["code"], b"403", channel.result)
|
||||||
|
|
||||||
@override_config({"session_lifetime": "24h"})
|
@override_config({"session_lifetime": "24h"})
|
||||||
def test_soft_logout(self) -> None:
|
def test_soft_logout(self) -> None:
|
||||||
@ -252,8 +252,8 @@ class LoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# we shouldn't be able to make requests without an access token
|
# we shouldn't be able to make requests without an access token
|
||||||
channel = self.make_request(b"GET", TEST_URL)
|
channel = self.make_request(b"GET", TEST_URL)
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
self.assertEquals(channel.json_body["errcode"], "M_MISSING_TOKEN")
|
self.assertEqual(channel.json_body["errcode"], "M_MISSING_TOKEN")
|
||||||
|
|
||||||
# log in as normal
|
# log in as normal
|
||||||
params = {
|
params = {
|
||||||
@ -263,22 +263,22 @@ class LoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
}
|
}
|
||||||
channel = self.make_request(b"POST", LOGIN_URL, params)
|
channel = self.make_request(b"POST", LOGIN_URL, params)
|
||||||
|
|
||||||
self.assertEquals(channel.code, 200, channel.result)
|
self.assertEqual(channel.code, 200, channel.result)
|
||||||
access_token = channel.json_body["access_token"]
|
access_token = channel.json_body["access_token"]
|
||||||
device_id = channel.json_body["device_id"]
|
device_id = channel.json_body["device_id"]
|
||||||
|
|
||||||
# we should now be able to make requests with the access token
|
# we should now be able to make requests with the access token
|
||||||
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
||||||
self.assertEquals(channel.code, 200, channel.result)
|
self.assertEqual(channel.code, 200, channel.result)
|
||||||
|
|
||||||
# time passes
|
# time passes
|
||||||
self.reactor.advance(24 * 3600)
|
self.reactor.advance(24 * 3600)
|
||||||
|
|
||||||
# ... and we should be soft-logouted
|
# ... and we should be soft-logouted
|
||||||
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
||||||
self.assertEquals(channel.code, 401, channel.result)
|
self.assertEqual(channel.code, 401, channel.result)
|
||||||
self.assertEquals(channel.json_body["errcode"], "M_UNKNOWN_TOKEN")
|
self.assertEqual(channel.json_body["errcode"], "M_UNKNOWN_TOKEN")
|
||||||
self.assertEquals(channel.json_body["soft_logout"], True)
|
self.assertEqual(channel.json_body["soft_logout"], True)
|
||||||
|
|
||||||
#
|
#
|
||||||
# test behaviour after deleting the expired device
|
# test behaviour after deleting the expired device
|
||||||
@ -290,17 +290,17 @@ class LoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
# more requests with the expired token should still return a soft-logout
|
# more requests with the expired token should still return a soft-logout
|
||||||
self.reactor.advance(3600)
|
self.reactor.advance(3600)
|
||||||
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
||||||
self.assertEquals(channel.code, 401, channel.result)
|
self.assertEqual(channel.code, 401, channel.result)
|
||||||
self.assertEquals(channel.json_body["errcode"], "M_UNKNOWN_TOKEN")
|
self.assertEqual(channel.json_body["errcode"], "M_UNKNOWN_TOKEN")
|
||||||
self.assertEquals(channel.json_body["soft_logout"], True)
|
self.assertEqual(channel.json_body["soft_logout"], True)
|
||||||
|
|
||||||
# ... but if we delete that device, it will be a proper logout
|
# ... but if we delete that device, it will be a proper logout
|
||||||
self._delete_device(access_token_2, "kermit", "monkey", device_id)
|
self._delete_device(access_token_2, "kermit", "monkey", device_id)
|
||||||
|
|
||||||
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
||||||
self.assertEquals(channel.code, 401, channel.result)
|
self.assertEqual(channel.code, 401, channel.result)
|
||||||
self.assertEquals(channel.json_body["errcode"], "M_UNKNOWN_TOKEN")
|
self.assertEqual(channel.json_body["errcode"], "M_UNKNOWN_TOKEN")
|
||||||
self.assertEquals(channel.json_body["soft_logout"], False)
|
self.assertEqual(channel.json_body["soft_logout"], False)
|
||||||
|
|
||||||
def _delete_device(
|
def _delete_device(
|
||||||
self, access_token: str, user_id: str, password: str, device_id: str
|
self, access_token: str, user_id: str, password: str, device_id: str
|
||||||
@ -309,7 +309,7 @@ class LoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
b"DELETE", "devices/" + device_id, access_token=access_token
|
b"DELETE", "devices/" + device_id, access_token=access_token
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.code, 401, channel.result)
|
self.assertEqual(channel.code, 401, channel.result)
|
||||||
# check it's a UI-Auth fail
|
# check it's a UI-Auth fail
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
set(channel.json_body.keys()),
|
set(channel.json_body.keys()),
|
||||||
@ -332,7 +332,7 @@ class LoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
access_token=access_token,
|
access_token=access_token,
|
||||||
content={"auth": auth},
|
content={"auth": auth},
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.code, 200, channel.result)
|
self.assertEqual(channel.code, 200, channel.result)
|
||||||
|
|
||||||
@override_config({"session_lifetime": "24h"})
|
@override_config({"session_lifetime": "24h"})
|
||||||
def test_session_can_hard_logout_after_being_soft_logged_out(self) -> None:
|
def test_session_can_hard_logout_after_being_soft_logged_out(self) -> None:
|
||||||
@ -343,20 +343,20 @@ class LoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# we should now be able to make requests with the access token
|
# we should now be able to make requests with the access token
|
||||||
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
||||||
self.assertEquals(channel.code, 200, channel.result)
|
self.assertEqual(channel.code, 200, channel.result)
|
||||||
|
|
||||||
# time passes
|
# time passes
|
||||||
self.reactor.advance(24 * 3600)
|
self.reactor.advance(24 * 3600)
|
||||||
|
|
||||||
# ... and we should be soft-logouted
|
# ... and we should be soft-logouted
|
||||||
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
||||||
self.assertEquals(channel.code, 401, channel.result)
|
self.assertEqual(channel.code, 401, channel.result)
|
||||||
self.assertEquals(channel.json_body["errcode"], "M_UNKNOWN_TOKEN")
|
self.assertEqual(channel.json_body["errcode"], "M_UNKNOWN_TOKEN")
|
||||||
self.assertEquals(channel.json_body["soft_logout"], True)
|
self.assertEqual(channel.json_body["soft_logout"], True)
|
||||||
|
|
||||||
# Now try to hard logout this session
|
# Now try to hard logout this session
|
||||||
channel = self.make_request(b"POST", "/logout", access_token=access_token)
|
channel = self.make_request(b"POST", "/logout", access_token=access_token)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
@override_config({"session_lifetime": "24h"})
|
@override_config({"session_lifetime": "24h"})
|
||||||
def test_session_can_hard_logout_all_sessions_after_being_soft_logged_out(
|
def test_session_can_hard_logout_all_sessions_after_being_soft_logged_out(
|
||||||
@ -369,20 +369,20 @@ class LoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# we should now be able to make requests with the access token
|
# we should now be able to make requests with the access token
|
||||||
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
||||||
self.assertEquals(channel.code, 200, channel.result)
|
self.assertEqual(channel.code, 200, channel.result)
|
||||||
|
|
||||||
# time passes
|
# time passes
|
||||||
self.reactor.advance(24 * 3600)
|
self.reactor.advance(24 * 3600)
|
||||||
|
|
||||||
# ... and we should be soft-logouted
|
# ... and we should be soft-logouted
|
||||||
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
channel = self.make_request(b"GET", TEST_URL, access_token=access_token)
|
||||||
self.assertEquals(channel.code, 401, channel.result)
|
self.assertEqual(channel.code, 401, channel.result)
|
||||||
self.assertEquals(channel.json_body["errcode"], "M_UNKNOWN_TOKEN")
|
self.assertEqual(channel.json_body["errcode"], "M_UNKNOWN_TOKEN")
|
||||||
self.assertEquals(channel.json_body["soft_logout"], True)
|
self.assertEqual(channel.json_body["soft_logout"], True)
|
||||||
|
|
||||||
# Now try to hard log out all of the user's sessions
|
# Now try to hard log out all of the user's sessions
|
||||||
channel = self.make_request(b"POST", "/logout/all", access_token=access_token)
|
channel = self.make_request(b"POST", "/logout/all", access_token=access_token)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
|
|
||||||
@skip_unless(has_saml2 and HAS_OIDC, "Requires SAML2 and OIDC")
|
@skip_unless(has_saml2 and HAS_OIDC, "Requires SAML2 and OIDC")
|
||||||
@ -1129,7 +1129,7 @@ class AppserviceLoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"POST", LOGIN_URL, params, access_token=self.service.token
|
b"POST", LOGIN_URL, params, access_token=self.service.token
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
def test_login_appservice_user_bot(self) -> None:
|
def test_login_appservice_user_bot(self) -> None:
|
||||||
"""Test that the appservice bot can use /login"""
|
"""Test that the appservice bot can use /login"""
|
||||||
@ -1143,7 +1143,7 @@ class AppserviceLoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"POST", LOGIN_URL, params, access_token=self.service.token
|
b"POST", LOGIN_URL, params, access_token=self.service.token
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
def test_login_appservice_wrong_user(self) -> None:
|
def test_login_appservice_wrong_user(self) -> None:
|
||||||
"""Test that non-as users cannot login with the as token"""
|
"""Test that non-as users cannot login with the as token"""
|
||||||
@ -1157,7 +1157,7 @@ class AppserviceLoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"POST", LOGIN_URL, params, access_token=self.service.token
|
b"POST", LOGIN_URL, params, access_token=self.service.token
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"403", channel.result)
|
self.assertEqual(channel.result["code"], b"403", channel.result)
|
||||||
|
|
||||||
def test_login_appservice_wrong_as(self) -> None:
|
def test_login_appservice_wrong_as(self) -> None:
|
||||||
"""Test that as users cannot login with wrong as token"""
|
"""Test that as users cannot login with wrong as token"""
|
||||||
@ -1171,7 +1171,7 @@ class AppserviceLoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"POST", LOGIN_URL, params, access_token=self.another_service.token
|
b"POST", LOGIN_URL, params, access_token=self.another_service.token
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"403", channel.result)
|
self.assertEqual(channel.result["code"], b"403", channel.result)
|
||||||
|
|
||||||
def test_login_appservice_no_token(self) -> None:
|
def test_login_appservice_no_token(self) -> None:
|
||||||
"""Test that users must provide a token when using the appservice
|
"""Test that users must provide a token when using the appservice
|
||||||
@ -1185,7 +1185,7 @@ class AppserviceLoginRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
}
|
}
|
||||||
channel = self.make_request(b"POST", LOGIN_URL, params)
|
channel = self.make_request(b"POST", LOGIN_URL, params)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
|
|
||||||
|
|
||||||
@skip_unless(HAS_OIDC, "requires OIDC")
|
@skip_unless(HAS_OIDC, "requires OIDC")
|
||||||
|
@ -80,7 +80,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
def test_get_displayname_other(self):
|
def test_get_displayname_other(self):
|
||||||
res = self._get_displayname(self.other)
|
res = self._get_displayname(self.other)
|
||||||
self.assertEquals(res, "Bob")
|
self.assertEqual(res, "Bob")
|
||||||
|
|
||||||
def test_set_displayname_other(self):
|
def test_set_displayname_other(self):
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
|
@ -65,7 +65,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"POST", self.url + b"?access_token=i_am_an_app_service", request_data
|
b"POST", self.url + b"?access_token=i_am_an_app_service", request_data
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
det_data = {"user_id": user_id, "home_server": self.hs.hostname}
|
det_data = {"user_id": user_id, "home_server": self.hs.hostname}
|
||||||
self.assertDictContainsSubset(det_data, channel.json_body)
|
self.assertDictContainsSubset(det_data, channel.json_body)
|
||||||
|
|
||||||
@ -87,7 +87,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"POST", self.url + b"?access_token=i_am_an_app_service", request_data
|
b"POST", self.url + b"?access_token=i_am_an_app_service", request_data
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"400", channel.result)
|
self.assertEqual(channel.result["code"], b"400", channel.result)
|
||||||
|
|
||||||
def test_POST_appservice_registration_invalid(self):
|
def test_POST_appservice_registration_invalid(self):
|
||||||
self.appservice = None # no application service exists
|
self.appservice = None # no application service exists
|
||||||
@ -98,21 +98,21 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"POST", self.url + b"?access_token=i_am_an_app_service", request_data
|
b"POST", self.url + b"?access_token=i_am_an_app_service", request_data
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
|
|
||||||
def test_POST_bad_password(self):
|
def test_POST_bad_password(self):
|
||||||
request_data = json.dumps({"username": "kermit", "password": 666})
|
request_data = json.dumps({"username": "kermit", "password": 666})
|
||||||
channel = self.make_request(b"POST", self.url, request_data)
|
channel = self.make_request(b"POST", self.url, request_data)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"400", channel.result)
|
self.assertEqual(channel.result["code"], b"400", channel.result)
|
||||||
self.assertEquals(channel.json_body["error"], "Invalid password")
|
self.assertEqual(channel.json_body["error"], "Invalid password")
|
||||||
|
|
||||||
def test_POST_bad_username(self):
|
def test_POST_bad_username(self):
|
||||||
request_data = json.dumps({"username": 777, "password": "monkey"})
|
request_data = json.dumps({"username": 777, "password": "monkey"})
|
||||||
channel = self.make_request(b"POST", self.url, request_data)
|
channel = self.make_request(b"POST", self.url, request_data)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"400", channel.result)
|
self.assertEqual(channel.result["code"], b"400", channel.result)
|
||||||
self.assertEquals(channel.json_body["error"], "Invalid username")
|
self.assertEqual(channel.json_body["error"], "Invalid username")
|
||||||
|
|
||||||
def test_POST_user_valid(self):
|
def test_POST_user_valid(self):
|
||||||
user_id = "@kermit:test"
|
user_id = "@kermit:test"
|
||||||
@ -131,7 +131,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
"home_server": self.hs.hostname,
|
"home_server": self.hs.hostname,
|
||||||
"device_id": device_id,
|
"device_id": device_id,
|
||||||
}
|
}
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
self.assertDictContainsSubset(det_data, channel.json_body)
|
self.assertDictContainsSubset(det_data, channel.json_body)
|
||||||
|
|
||||||
@override_config({"enable_registration": False})
|
@override_config({"enable_registration": False})
|
||||||
@ -141,9 +141,9 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
channel = self.make_request(b"POST", self.url, request_data)
|
channel = self.make_request(b"POST", self.url, request_data)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"403", channel.result)
|
self.assertEqual(channel.result["code"], b"403", channel.result)
|
||||||
self.assertEquals(channel.json_body["error"], "Registration has been disabled")
|
self.assertEqual(channel.json_body["error"], "Registration has been disabled")
|
||||||
self.assertEquals(channel.json_body["errcode"], "M_FORBIDDEN")
|
self.assertEqual(channel.json_body["errcode"], "M_FORBIDDEN")
|
||||||
|
|
||||||
def test_POST_guest_registration(self):
|
def test_POST_guest_registration(self):
|
||||||
self.hs.config.key.macaroon_secret_key = "test"
|
self.hs.config.key.macaroon_secret_key = "test"
|
||||||
@ -152,7 +152,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(b"POST", self.url + b"?kind=guest", b"{}")
|
channel = self.make_request(b"POST", self.url + b"?kind=guest", b"{}")
|
||||||
|
|
||||||
det_data = {"home_server": self.hs.hostname, "device_id": "guest_device"}
|
det_data = {"home_server": self.hs.hostname, "device_id": "guest_device"}
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
self.assertDictContainsSubset(det_data, channel.json_body)
|
self.assertDictContainsSubset(det_data, channel.json_body)
|
||||||
|
|
||||||
def test_POST_disabled_guest_registration(self):
|
def test_POST_disabled_guest_registration(self):
|
||||||
@ -160,8 +160,8 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
channel = self.make_request(b"POST", self.url + b"?kind=guest", b"{}")
|
channel = self.make_request(b"POST", self.url + b"?kind=guest", b"{}")
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"403", channel.result)
|
self.assertEqual(channel.result["code"], b"403", channel.result)
|
||||||
self.assertEquals(channel.json_body["error"], "Guest access is disabled")
|
self.assertEqual(channel.json_body["error"], "Guest access is disabled")
|
||||||
|
|
||||||
@override_config({"rc_registration": {"per_second": 0.17, "burst_count": 5}})
|
@override_config({"rc_registration": {"per_second": 0.17, "burst_count": 5}})
|
||||||
def test_POST_ratelimiting_guest(self):
|
def test_POST_ratelimiting_guest(self):
|
||||||
@ -170,16 +170,16 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(b"POST", url, b"{}")
|
channel = self.make_request(b"POST", url, b"{}")
|
||||||
|
|
||||||
if i == 5:
|
if i == 5:
|
||||||
self.assertEquals(channel.result["code"], b"429", channel.result)
|
self.assertEqual(channel.result["code"], b"429", channel.result)
|
||||||
retry_after_ms = int(channel.json_body["retry_after_ms"])
|
retry_after_ms = int(channel.json_body["retry_after_ms"])
|
||||||
else:
|
else:
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
self.reactor.advance(retry_after_ms / 1000.0 + 1.0)
|
self.reactor.advance(retry_after_ms / 1000.0 + 1.0)
|
||||||
|
|
||||||
channel = self.make_request(b"POST", self.url + b"?kind=guest", b"{}")
|
channel = self.make_request(b"POST", self.url + b"?kind=guest", b"{}")
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
@override_config({"rc_registration": {"per_second": 0.17, "burst_count": 5}})
|
@override_config({"rc_registration": {"per_second": 0.17, "burst_count": 5}})
|
||||||
def test_POST_ratelimiting(self):
|
def test_POST_ratelimiting(self):
|
||||||
@ -194,16 +194,16 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(b"POST", self.url, request_data)
|
channel = self.make_request(b"POST", self.url, request_data)
|
||||||
|
|
||||||
if i == 5:
|
if i == 5:
|
||||||
self.assertEquals(channel.result["code"], b"429", channel.result)
|
self.assertEqual(channel.result["code"], b"429", channel.result)
|
||||||
retry_after_ms = int(channel.json_body["retry_after_ms"])
|
retry_after_ms = int(channel.json_body["retry_after_ms"])
|
||||||
else:
|
else:
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
self.reactor.advance(retry_after_ms / 1000.0 + 1.0)
|
self.reactor.advance(retry_after_ms / 1000.0 + 1.0)
|
||||||
|
|
||||||
channel = self.make_request(b"POST", self.url + b"?kind=guest", b"{}")
|
channel = self.make_request(b"POST", self.url + b"?kind=guest", b"{}")
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
@override_config({"registration_requires_token": True})
|
@override_config({"registration_requires_token": True})
|
||||||
def test_POST_registration_requires_token(self):
|
def test_POST_registration_requires_token(self):
|
||||||
@ -231,7 +231,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# Request without auth to get flows and session
|
# Request without auth to get flows and session
|
||||||
channel = self.make_request(b"POST", self.url, json.dumps(params))
|
channel = self.make_request(b"POST", self.url, json.dumps(params))
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
flows = channel.json_body["flows"]
|
flows = channel.json_body["flows"]
|
||||||
# Synapse adds a dummy stage to differentiate flows where otherwise one
|
# Synapse adds a dummy stage to differentiate flows where otherwise one
|
||||||
# flow would be a subset of another flow.
|
# flow would be a subset of another flow.
|
||||||
@ -249,7 +249,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
}
|
}
|
||||||
request_data = json.dumps(params)
|
request_data = json.dumps(params)
|
||||||
channel = self.make_request(b"POST", self.url, request_data)
|
channel = self.make_request(b"POST", self.url, request_data)
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
completed = channel.json_body["completed"]
|
completed = channel.json_body["completed"]
|
||||||
self.assertCountEqual([LoginType.REGISTRATION_TOKEN], completed)
|
self.assertCountEqual([LoginType.REGISTRATION_TOKEN], completed)
|
||||||
|
|
||||||
@ -265,7 +265,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
"home_server": self.hs.hostname,
|
"home_server": self.hs.hostname,
|
||||||
"device_id": device_id,
|
"device_id": device_id,
|
||||||
}
|
}
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
self.assertDictContainsSubset(det_data, channel.json_body)
|
self.assertDictContainsSubset(det_data, channel.json_body)
|
||||||
|
|
||||||
# Check the `completed` counter has been incremented and pending is 0
|
# Check the `completed` counter has been incremented and pending is 0
|
||||||
@ -276,8 +276,8 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
retcols=["pending", "completed"],
|
retcols=["pending", "completed"],
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(res["completed"], 1)
|
self.assertEqual(res["completed"], 1)
|
||||||
self.assertEquals(res["pending"], 0)
|
self.assertEqual(res["pending"], 0)
|
||||||
|
|
||||||
@override_config({"registration_requires_token": True})
|
@override_config({"registration_requires_token": True})
|
||||||
def test_POST_registration_token_invalid(self):
|
def test_POST_registration_token_invalid(self):
|
||||||
@ -295,23 +295,23 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
"session": session,
|
"session": session,
|
||||||
}
|
}
|
||||||
channel = self.make_request(b"POST", self.url, json.dumps(params))
|
channel = self.make_request(b"POST", self.url, json.dumps(params))
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
self.assertEquals(channel.json_body["errcode"], Codes.MISSING_PARAM)
|
self.assertEqual(channel.json_body["errcode"], Codes.MISSING_PARAM)
|
||||||
self.assertEquals(channel.json_body["completed"], [])
|
self.assertEqual(channel.json_body["completed"], [])
|
||||||
|
|
||||||
# Test with non-string (invalid)
|
# Test with non-string (invalid)
|
||||||
params["auth"]["token"] = 1234
|
params["auth"]["token"] = 1234
|
||||||
channel = self.make_request(b"POST", self.url, json.dumps(params))
|
channel = self.make_request(b"POST", self.url, json.dumps(params))
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
self.assertEquals(channel.json_body["errcode"], Codes.INVALID_PARAM)
|
self.assertEqual(channel.json_body["errcode"], Codes.INVALID_PARAM)
|
||||||
self.assertEquals(channel.json_body["completed"], [])
|
self.assertEqual(channel.json_body["completed"], [])
|
||||||
|
|
||||||
# Test with unknown token (invalid)
|
# Test with unknown token (invalid)
|
||||||
params["auth"]["token"] = "1234"
|
params["auth"]["token"] = "1234"
|
||||||
channel = self.make_request(b"POST", self.url, json.dumps(params))
|
channel = self.make_request(b"POST", self.url, json.dumps(params))
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
self.assertEquals(channel.json_body["errcode"], Codes.UNAUTHORIZED)
|
self.assertEqual(channel.json_body["errcode"], Codes.UNAUTHORIZED)
|
||||||
self.assertEquals(channel.json_body["completed"], [])
|
self.assertEqual(channel.json_body["completed"], [])
|
||||||
|
|
||||||
@override_config({"registration_requires_token": True})
|
@override_config({"registration_requires_token": True})
|
||||||
def test_POST_registration_token_limit_uses(self):
|
def test_POST_registration_token_limit_uses(self):
|
||||||
@ -354,7 +354,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
retcol="pending",
|
retcol="pending",
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(pending, 1)
|
self.assertEqual(pending, 1)
|
||||||
|
|
||||||
# Check auth fails when using token with session2
|
# Check auth fails when using token with session2
|
||||||
params2["auth"] = {
|
params2["auth"] = {
|
||||||
@ -363,9 +363,9 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
"session": session2,
|
"session": session2,
|
||||||
}
|
}
|
||||||
channel = self.make_request(b"POST", self.url, json.dumps(params2))
|
channel = self.make_request(b"POST", self.url, json.dumps(params2))
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
self.assertEquals(channel.json_body["errcode"], Codes.UNAUTHORIZED)
|
self.assertEqual(channel.json_body["errcode"], Codes.UNAUTHORIZED)
|
||||||
self.assertEquals(channel.json_body["completed"], [])
|
self.assertEqual(channel.json_body["completed"], [])
|
||||||
|
|
||||||
# Complete registration with session1
|
# Complete registration with session1
|
||||||
params1["auth"]["type"] = LoginType.DUMMY
|
params1["auth"]["type"] = LoginType.DUMMY
|
||||||
@ -378,14 +378,14 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
retcols=["pending", "completed"],
|
retcols=["pending", "completed"],
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(res["pending"], 0)
|
self.assertEqual(res["pending"], 0)
|
||||||
self.assertEquals(res["completed"], 1)
|
self.assertEqual(res["completed"], 1)
|
||||||
|
|
||||||
# Check auth still fails when using token with session2
|
# Check auth still fails when using token with session2
|
||||||
channel = self.make_request(b"POST", self.url, json.dumps(params2))
|
channel = self.make_request(b"POST", self.url, json.dumps(params2))
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
self.assertEquals(channel.json_body["errcode"], Codes.UNAUTHORIZED)
|
self.assertEqual(channel.json_body["errcode"], Codes.UNAUTHORIZED)
|
||||||
self.assertEquals(channel.json_body["completed"], [])
|
self.assertEqual(channel.json_body["completed"], [])
|
||||||
|
|
||||||
@override_config({"registration_requires_token": True})
|
@override_config({"registration_requires_token": True})
|
||||||
def test_POST_registration_token_expiry(self):
|
def test_POST_registration_token_expiry(self):
|
||||||
@ -417,9 +417,9 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
"session": session,
|
"session": session,
|
||||||
}
|
}
|
||||||
channel = self.make_request(b"POST", self.url, json.dumps(params))
|
channel = self.make_request(b"POST", self.url, json.dumps(params))
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
self.assertEquals(channel.json_body["errcode"], Codes.UNAUTHORIZED)
|
self.assertEqual(channel.json_body["errcode"], Codes.UNAUTHORIZED)
|
||||||
self.assertEquals(channel.json_body["completed"], [])
|
self.assertEqual(channel.json_body["completed"], [])
|
||||||
|
|
||||||
# Update token so it expires tomorrow
|
# Update token so it expires tomorrow
|
||||||
self.get_success(
|
self.get_success(
|
||||||
@ -504,7 +504,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
retcol="result",
|
retcol="result",
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(db_to_json(result2), token)
|
self.assertEqual(db_to_json(result2), token)
|
||||||
|
|
||||||
# Delete both sessions (mimics expiry)
|
# Delete both sessions (mimics expiry)
|
||||||
self.get_success(
|
self.get_success(
|
||||||
@ -519,7 +519,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
retcol="pending",
|
retcol="pending",
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(pending, 0)
|
self.assertEqual(pending, 0)
|
||||||
|
|
||||||
@override_config({"registration_requires_token": True})
|
@override_config({"registration_requires_token": True})
|
||||||
def test_POST_registration_token_session_expiry_deleted_token(self):
|
def test_POST_registration_token_session_expiry_deleted_token(self):
|
||||||
@ -572,7 +572,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
def test_advertised_flows(self):
|
def test_advertised_flows(self):
|
||||||
channel = self.make_request(b"POST", self.url, b"{}")
|
channel = self.make_request(b"POST", self.url, b"{}")
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
flows = channel.json_body["flows"]
|
flows = channel.json_body["flows"]
|
||||||
|
|
||||||
# with the stock config, we only expect the dummy flow
|
# with the stock config, we only expect the dummy flow
|
||||||
@ -595,7 +595,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
def test_advertised_flows_captcha_and_terms_and_3pids(self):
|
def test_advertised_flows_captcha_and_terms_and_3pids(self):
|
||||||
channel = self.make_request(b"POST", self.url, b"{}")
|
channel = self.make_request(b"POST", self.url, b"{}")
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
flows = channel.json_body["flows"]
|
flows = channel.json_body["flows"]
|
||||||
|
|
||||||
self.assertCountEqual(
|
self.assertCountEqual(
|
||||||
@ -627,7 +627,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
def test_advertised_flows_no_msisdn_email_required(self):
|
def test_advertised_flows_no_msisdn_email_required(self):
|
||||||
channel = self.make_request(b"POST", self.url, b"{}")
|
channel = self.make_request(b"POST", self.url, b"{}")
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
flows = channel.json_body["flows"]
|
flows = channel.json_body["flows"]
|
||||||
|
|
||||||
# with the stock config, we expect all four combinations of 3pid
|
# with the stock config, we expect all four combinations of 3pid
|
||||||
@ -671,7 +671,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"register/email/requestToken",
|
b"register/email/requestToken",
|
||||||
{"client_secret": "foobar", "email": email, "send_attempt": 1},
|
{"client_secret": "foobar", "email": email, "send_attempt": 1},
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
self.assertIsNotNone(channel.json_body.get("sid"))
|
self.assertIsNotNone(channel.json_body.get("sid"))
|
||||||
|
|
||||||
@ -694,9 +694,9 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"register/email/requestToken",
|
b"register/email/requestToken",
|
||||||
{"client_secret": "foobar", "email": "email@@email", "send_attempt": 1},
|
{"client_secret": "foobar", "email": "email@@email", "send_attempt": 1},
|
||||||
)
|
)
|
||||||
self.assertEquals(400, channel.code, channel.result)
|
self.assertEqual(400, channel.code, channel.result)
|
||||||
# Check error to ensure that we're not erroring due to a bug in the test.
|
# Check error to ensure that we're not erroring due to a bug in the test.
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
channel.json_body,
|
channel.json_body,
|
||||||
{"errcode": "M_UNKNOWN", "error": "Unable to parse email address"},
|
{"errcode": "M_UNKNOWN", "error": "Unable to parse email address"},
|
||||||
)
|
)
|
||||||
@ -707,8 +707,8 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"register/email/requestToken",
|
b"register/email/requestToken",
|
||||||
{"client_secret": "foobar", "email": "email", "send_attempt": 1},
|
{"client_secret": "foobar", "email": "email", "send_attempt": 1},
|
||||||
)
|
)
|
||||||
self.assertEquals(400, channel.code, channel.result)
|
self.assertEqual(400, channel.code, channel.result)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
channel.json_body,
|
channel.json_body,
|
||||||
{"errcode": "M_UNKNOWN", "error": "Unable to parse email address"},
|
{"errcode": "M_UNKNOWN", "error": "Unable to parse email address"},
|
||||||
)
|
)
|
||||||
@ -720,8 +720,8 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"register/email/requestToken",
|
b"register/email/requestToken",
|
||||||
{"client_secret": "foobar", "email": email, "send_attempt": 1},
|
{"client_secret": "foobar", "email": email, "send_attempt": 1},
|
||||||
)
|
)
|
||||||
self.assertEquals(400, channel.code, channel.result)
|
self.assertEqual(400, channel.code, channel.result)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
channel.json_body,
|
channel.json_body,
|
||||||
{"errcode": "M_UNKNOWN", "error": "Unable to parse email address"},
|
{"errcode": "M_UNKNOWN", "error": "Unable to parse email address"},
|
||||||
)
|
)
|
||||||
@ -745,7 +745,7 @@ class RegisterRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
# Check that /available correctly ignores the username provided despite the
|
# Check that /available correctly ignores the username provided despite the
|
||||||
# username being already registered.
|
# username being already registered.
|
||||||
channel = self.make_request("GET", "register/available?username=" + username)
|
channel = self.make_request("GET", "register/available?username=" + username)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
# Test that when starting a UIA registration flow the request doesn't fail because
|
# Test that when starting a UIA registration flow the request doesn't fail because
|
||||||
# of a conflicting username
|
# of a conflicting username
|
||||||
@ -799,14 +799,14 @@ class AccountValidityTestCase(unittest.HomeserverTestCase):
|
|||||||
# endpoint.
|
# endpoint.
|
||||||
channel = self.make_request(b"GET", "/sync", access_token=tok)
|
channel = self.make_request(b"GET", "/sync", access_token=tok)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
self.reactor.advance(datetime.timedelta(weeks=1).total_seconds())
|
self.reactor.advance(datetime.timedelta(weeks=1).total_seconds())
|
||||||
|
|
||||||
channel = self.make_request(b"GET", "/sync", access_token=tok)
|
channel = self.make_request(b"GET", "/sync", access_token=tok)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"403", channel.result)
|
self.assertEqual(channel.result["code"], b"403", channel.result)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
channel.json_body["errcode"], Codes.EXPIRED_ACCOUNT, channel.result
|
channel.json_body["errcode"], Codes.EXPIRED_ACCOUNT, channel.result
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -826,12 +826,12 @@ class AccountValidityTestCase(unittest.HomeserverTestCase):
|
|||||||
params = {"user_id": user_id}
|
params = {"user_id": user_id}
|
||||||
request_data = json.dumps(params)
|
request_data = json.dumps(params)
|
||||||
channel = self.make_request(b"POST", url, request_data, access_token=admin_tok)
|
channel = self.make_request(b"POST", url, request_data, access_token=admin_tok)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
# The specific endpoint doesn't matter, all we need is an authenticated
|
# The specific endpoint doesn't matter, all we need is an authenticated
|
||||||
# endpoint.
|
# endpoint.
|
||||||
channel = self.make_request(b"GET", "/sync", access_token=tok)
|
channel = self.make_request(b"GET", "/sync", access_token=tok)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
def test_manual_expire(self):
|
def test_manual_expire(self):
|
||||||
user_id = self.register_user("kermit", "monkey")
|
user_id = self.register_user("kermit", "monkey")
|
||||||
@ -848,13 +848,13 @@ class AccountValidityTestCase(unittest.HomeserverTestCase):
|
|||||||
}
|
}
|
||||||
request_data = json.dumps(params)
|
request_data = json.dumps(params)
|
||||||
channel = self.make_request(b"POST", url, request_data, access_token=admin_tok)
|
channel = self.make_request(b"POST", url, request_data, access_token=admin_tok)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
# The specific endpoint doesn't matter, all we need is an authenticated
|
# The specific endpoint doesn't matter, all we need is an authenticated
|
||||||
# endpoint.
|
# endpoint.
|
||||||
channel = self.make_request(b"GET", "/sync", access_token=tok)
|
channel = self.make_request(b"GET", "/sync", access_token=tok)
|
||||||
self.assertEquals(channel.result["code"], b"403", channel.result)
|
self.assertEqual(channel.result["code"], b"403", channel.result)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
channel.json_body["errcode"], Codes.EXPIRED_ACCOUNT, channel.result
|
channel.json_body["errcode"], Codes.EXPIRED_ACCOUNT, channel.result
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -873,18 +873,18 @@ class AccountValidityTestCase(unittest.HomeserverTestCase):
|
|||||||
}
|
}
|
||||||
request_data = json.dumps(params)
|
request_data = json.dumps(params)
|
||||||
channel = self.make_request(b"POST", url, request_data, access_token=admin_tok)
|
channel = self.make_request(b"POST", url, request_data, access_token=admin_tok)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
# Try to log the user out
|
# Try to log the user out
|
||||||
channel = self.make_request(b"POST", "/logout", access_token=tok)
|
channel = self.make_request(b"POST", "/logout", access_token=tok)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
# Log the user in again (allowed for expired accounts)
|
# Log the user in again (allowed for expired accounts)
|
||||||
tok = self.login("kermit", "monkey")
|
tok = self.login("kermit", "monkey")
|
||||||
|
|
||||||
# Try to log out all of the user's sessions
|
# Try to log out all of the user's sessions
|
||||||
channel = self.make_request(b"POST", "/logout/all", access_token=tok)
|
channel = self.make_request(b"POST", "/logout/all", access_token=tok)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
|
|
||||||
class AccountValidityRenewalByEmailTestCase(unittest.HomeserverTestCase):
|
class AccountValidityRenewalByEmailTestCase(unittest.HomeserverTestCase):
|
||||||
@ -959,7 +959,7 @@ class AccountValidityRenewalByEmailTestCase(unittest.HomeserverTestCase):
|
|||||||
renewal_token = self.get_success(self.store.get_renewal_token_for_user(user_id))
|
renewal_token = self.get_success(self.store.get_renewal_token_for_user(user_id))
|
||||||
url = "/_matrix/client/unstable/account_validity/renew?token=%s" % renewal_token
|
url = "/_matrix/client/unstable/account_validity/renew?token=%s" % renewal_token
|
||||||
channel = self.make_request(b"GET", url)
|
channel = self.make_request(b"GET", url)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
# Check that we're getting HTML back.
|
# Check that we're getting HTML back.
|
||||||
content_type = channel.headers.getRawHeaders(b"Content-Type")
|
content_type = channel.headers.getRawHeaders(b"Content-Type")
|
||||||
@ -977,7 +977,7 @@ class AccountValidityRenewalByEmailTestCase(unittest.HomeserverTestCase):
|
|||||||
# Move 1 day forward. Try to renew with the same token again.
|
# Move 1 day forward. Try to renew with the same token again.
|
||||||
url = "/_matrix/client/unstable/account_validity/renew?token=%s" % renewal_token
|
url = "/_matrix/client/unstable/account_validity/renew?token=%s" % renewal_token
|
||||||
channel = self.make_request(b"GET", url)
|
channel = self.make_request(b"GET", url)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
# Check that we're getting HTML back.
|
# Check that we're getting HTML back.
|
||||||
content_type = channel.headers.getRawHeaders(b"Content-Type")
|
content_type = channel.headers.getRawHeaders(b"Content-Type")
|
||||||
@ -997,14 +997,14 @@ class AccountValidityRenewalByEmailTestCase(unittest.HomeserverTestCase):
|
|||||||
# succeed.
|
# succeed.
|
||||||
self.reactor.advance(datetime.timedelta(days=3).total_seconds())
|
self.reactor.advance(datetime.timedelta(days=3).total_seconds())
|
||||||
channel = self.make_request(b"GET", "/sync", access_token=tok)
|
channel = self.make_request(b"GET", "/sync", access_token=tok)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
def test_renewal_invalid_token(self):
|
def test_renewal_invalid_token(self):
|
||||||
# Hit the renewal endpoint with an invalid token and check that it behaves as
|
# Hit the renewal endpoint with an invalid token and check that it behaves as
|
||||||
# expected, i.e. that it responds with 404 Not Found and the correct HTML.
|
# expected, i.e. that it responds with 404 Not Found and the correct HTML.
|
||||||
url = "/_matrix/client/unstable/account_validity/renew?token=123"
|
url = "/_matrix/client/unstable/account_validity/renew?token=123"
|
||||||
channel = self.make_request(b"GET", url)
|
channel = self.make_request(b"GET", url)
|
||||||
self.assertEquals(channel.result["code"], b"404", channel.result)
|
self.assertEqual(channel.result["code"], b"404", channel.result)
|
||||||
|
|
||||||
# Check that we're getting HTML back.
|
# Check that we're getting HTML back.
|
||||||
content_type = channel.headers.getRawHeaders(b"Content-Type")
|
content_type = channel.headers.getRawHeaders(b"Content-Type")
|
||||||
@ -1028,7 +1028,7 @@ class AccountValidityRenewalByEmailTestCase(unittest.HomeserverTestCase):
|
|||||||
"/_matrix/client/unstable/account_validity/send_mail",
|
"/_matrix/client/unstable/account_validity/send_mail",
|
||||||
access_token=tok,
|
access_token=tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
self.assertEqual(len(self.email_attempts), 1)
|
self.assertEqual(len(self.email_attempts), 1)
|
||||||
|
|
||||||
@ -1103,7 +1103,7 @@ class AccountValidityRenewalByEmailTestCase(unittest.HomeserverTestCase):
|
|||||||
"/_matrix/client/unstable/account_validity/send_mail",
|
"/_matrix/client/unstable/account_validity/send_mail",
|
||||||
access_token=tok,
|
access_token=tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
self.assertEqual(len(self.email_attempts), 1)
|
self.assertEqual(len(self.email_attempts), 1)
|
||||||
|
|
||||||
@ -1183,8 +1183,8 @@ class RegistrationTokenValidityRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"GET",
|
b"GET",
|
||||||
f"{self.url}?token={token}",
|
f"{self.url}?token={token}",
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
self.assertEquals(channel.json_body["valid"], True)
|
self.assertEqual(channel.json_body["valid"], True)
|
||||||
|
|
||||||
def test_GET_token_invalid(self):
|
def test_GET_token_invalid(self):
|
||||||
token = "1234"
|
token = "1234"
|
||||||
@ -1192,8 +1192,8 @@ class RegistrationTokenValidityRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"GET",
|
b"GET",
|
||||||
f"{self.url}?token={token}",
|
f"{self.url}?token={token}",
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
self.assertEquals(channel.json_body["valid"], False)
|
self.assertEqual(channel.json_body["valid"], False)
|
||||||
|
|
||||||
@override_config(
|
@override_config(
|
||||||
{"rc_registration_token_validity": {"per_second": 0.1, "burst_count": 5}}
|
{"rc_registration_token_validity": {"per_second": 0.1, "burst_count": 5}}
|
||||||
@ -1208,10 +1208,10 @@ class RegistrationTokenValidityRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
if i == 5:
|
if i == 5:
|
||||||
self.assertEquals(channel.result["code"], b"429", channel.result)
|
self.assertEqual(channel.result["code"], b"429", channel.result)
|
||||||
retry_after_ms = int(channel.json_body["retry_after_ms"])
|
retry_after_ms = int(channel.json_body["retry_after_ms"])
|
||||||
else:
|
else:
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
self.reactor.advance(retry_after_ms / 1000.0 + 1.0)
|
self.reactor.advance(retry_after_ms / 1000.0 + 1.0)
|
||||||
|
|
||||||
@ -1219,4 +1219,4 @@ class RegistrationTokenValidityRestServletTestCase(unittest.HomeserverTestCase):
|
|||||||
b"GET",
|
b"GET",
|
||||||
f"{self.url}?token={token}",
|
f"{self.url}?token={token}",
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
@ -69,7 +69,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", key="👍")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", key="👍")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
event_id = channel.json_body["event_id"]
|
event_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
@ -78,7 +78,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"/rooms/%s/event/%s" % (self.room, event_id),
|
"/rooms/%s/event/%s" % (self.room, event_id),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
self.assert_dict(
|
self.assert_dict(
|
||||||
{
|
{
|
||||||
@ -103,7 +103,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
parent_id="foo",
|
parent_id="foo",
|
||||||
content={"body": "foo", "msgtype": "m.text"},
|
content={"body": "foo", "msgtype": "m.text"},
|
||||||
)
|
)
|
||||||
self.assertEquals(400, channel.code, channel.json_body)
|
self.assertEqual(400, channel.code, channel.json_body)
|
||||||
|
|
||||||
# Unless that event is referenced from another event!
|
# Unless that event is referenced from another event!
|
||||||
self.get_success(
|
self.get_success(
|
||||||
@ -123,7 +123,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
parent_id="foo",
|
parent_id="foo",
|
||||||
content={"body": "foo", "msgtype": "m.text"},
|
content={"body": "foo", "msgtype": "m.text"},
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
def test_deny_invalid_room(self):
|
def test_deny_invalid_room(self):
|
||||||
"""Test that we deny relations on non-existant events"""
|
"""Test that we deny relations on non-existant events"""
|
||||||
@ -136,15 +136,15 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self._send_relation(
|
channel = self._send_relation(
|
||||||
RelationTypes.ANNOTATION, "m.reaction", parent_id=parent_id, key="A"
|
RelationTypes.ANNOTATION, "m.reaction", parent_id=parent_id, key="A"
|
||||||
)
|
)
|
||||||
self.assertEquals(400, channel.code, channel.json_body)
|
self.assertEqual(400, channel.code, channel.json_body)
|
||||||
|
|
||||||
def test_deny_double_react(self):
|
def test_deny_double_react(self):
|
||||||
"""Test that we deny relations on membership events"""
|
"""Test that we deny relations on membership events"""
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", key="a")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", key="a")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "a")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "a")
|
||||||
self.assertEquals(400, channel.code, channel.json_body)
|
self.assertEqual(400, channel.code, channel.json_body)
|
||||||
|
|
||||||
def test_deny_forked_thread(self):
|
def test_deny_forked_thread(self):
|
||||||
"""It is invalid to start a thread off a thread."""
|
"""It is invalid to start a thread off a thread."""
|
||||||
@ -154,7 +154,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
content={"msgtype": "m.text", "body": "foo"},
|
content={"msgtype": "m.text", "body": "foo"},
|
||||||
parent_id=self.parent_id,
|
parent_id=self.parent_id,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
parent_id = channel.json_body["event_id"]
|
parent_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
channel = self._send_relation(
|
channel = self._send_relation(
|
||||||
@ -163,16 +163,16 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
content={"msgtype": "m.text", "body": "foo"},
|
content={"msgtype": "m.text", "body": "foo"},
|
||||||
parent_id=parent_id,
|
parent_id=parent_id,
|
||||||
)
|
)
|
||||||
self.assertEquals(400, channel.code, channel.json_body)
|
self.assertEqual(400, channel.code, channel.json_body)
|
||||||
|
|
||||||
def test_basic_paginate_relations(self):
|
def test_basic_paginate_relations(self):
|
||||||
"""Tests that calling pagination API correctly the latest relations."""
|
"""Tests that calling pagination API correctly the latest relations."""
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "a")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "a")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
first_annotation_id = channel.json_body["event_id"]
|
first_annotation_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "b")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "b")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
second_annotation_id = channel.json_body["event_id"]
|
second_annotation_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -180,11 +180,11 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?limit=1",
|
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?limit=1",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# We expect to get back a single pagination result, which is the latest
|
# We expect to get back a single pagination result, which is the latest
|
||||||
# full relation event we sent above.
|
# full relation event we sent above.
|
||||||
self.assertEquals(len(channel.json_body["chunk"]), 1, channel.json_body)
|
self.assertEqual(len(channel.json_body["chunk"]), 1, channel.json_body)
|
||||||
self.assert_dict(
|
self.assert_dict(
|
||||||
{
|
{
|
||||||
"event_id": second_annotation_id,
|
"event_id": second_annotation_id,
|
||||||
@ -195,7 +195,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
# We also expect to get the original event (the id of which is self.parent_id)
|
# We also expect to get the original event (the id of which is self.parent_id)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
channel.json_body["original_event"]["event_id"], self.parent_id
|
channel.json_body["original_event"]["event_id"], self.parent_id
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -212,11 +212,11 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/{self.parent_id}?limit=1&org.matrix.msc3715.dir=f",
|
f"/{self.parent_id}?limit=1&org.matrix.msc3715.dir=f",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# We expect to get back a single pagination result, which is the earliest
|
# We expect to get back a single pagination result, which is the earliest
|
||||||
# full relation event we sent above.
|
# full relation event we sent above.
|
||||||
self.assertEquals(len(channel.json_body["chunk"]), 1, channel.json_body)
|
self.assertEqual(len(channel.json_body["chunk"]), 1, channel.json_body)
|
||||||
self.assert_dict(
|
self.assert_dict(
|
||||||
{
|
{
|
||||||
"event_id": first_annotation_id,
|
"event_id": first_annotation_id,
|
||||||
@ -245,7 +245,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self._send_relation(
|
channel = self._send_relation(
|
||||||
RelationTypes.ANNOTATION, "m.reaction", chr(ord("a") + idx)
|
RelationTypes.ANNOTATION, "m.reaction", chr(ord("a") + idx)
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
expected_event_ids.append(channel.json_body["event_id"])
|
expected_event_ids.append(channel.json_body["event_id"])
|
||||||
|
|
||||||
prev_token = ""
|
prev_token = ""
|
||||||
@ -260,12 +260,12 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?limit=1{from_token}",
|
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?limit=1{from_token}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
found_event_ids.extend(e["event_id"] for e in channel.json_body["chunk"])
|
found_event_ids.extend(e["event_id"] for e in channel.json_body["chunk"])
|
||||||
next_batch = channel.json_body.get("next_batch")
|
next_batch = channel.json_body.get("next_batch")
|
||||||
|
|
||||||
self.assertNotEquals(prev_token, next_batch)
|
self.assertNotEqual(prev_token, next_batch)
|
||||||
prev_token = next_batch
|
prev_token = next_batch
|
||||||
|
|
||||||
if not prev_token:
|
if not prev_token:
|
||||||
@ -273,7 +273,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# We paginated backwards, so reverse
|
# We paginated backwards, so reverse
|
||||||
found_event_ids.reverse()
|
found_event_ids.reverse()
|
||||||
self.assertEquals(found_event_ids, expected_event_ids)
|
self.assertEqual(found_event_ids, expected_event_ids)
|
||||||
|
|
||||||
# Reset and try again, but convert the tokens to the legacy format.
|
# Reset and try again, but convert the tokens to the legacy format.
|
||||||
prev_token = ""
|
prev_token = ""
|
||||||
@ -288,12 +288,12 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?limit=1{from_token}",
|
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?limit=1{from_token}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
found_event_ids.extend(e["event_id"] for e in channel.json_body["chunk"])
|
found_event_ids.extend(e["event_id"] for e in channel.json_body["chunk"])
|
||||||
next_batch = channel.json_body.get("next_batch")
|
next_batch = channel.json_body.get("next_batch")
|
||||||
|
|
||||||
self.assertNotEquals(prev_token, next_batch)
|
self.assertNotEqual(prev_token, next_batch)
|
||||||
prev_token = next_batch
|
prev_token = next_batch
|
||||||
|
|
||||||
if not prev_token:
|
if not prev_token:
|
||||||
@ -301,12 +301,12 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# We paginated backwards, so reverse
|
# We paginated backwards, so reverse
|
||||||
found_event_ids.reverse()
|
found_event_ids.reverse()
|
||||||
self.assertEquals(found_event_ids, expected_event_ids)
|
self.assertEqual(found_event_ids, expected_event_ids)
|
||||||
|
|
||||||
def test_pagination_from_sync_and_messages(self):
|
def test_pagination_from_sync_and_messages(self):
|
||||||
"""Pagination tokens from /sync and /messages can be used to paginate /relations."""
|
"""Pagination tokens from /sync and /messages can be used to paginate /relations."""
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "A")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "A")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
annotation_id = channel.json_body["event_id"]
|
annotation_id = channel.json_body["event_id"]
|
||||||
# Send an event after the relation events.
|
# Send an event after the relation events.
|
||||||
self.helper.send(self.room, body="Latest event", tok=self.user_token)
|
self.helper.send(self.room, body="Latest event", tok=self.user_token)
|
||||||
@ -319,7 +319,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET", f"/sync?filter={filter}", access_token=self.user_token
|
"GET", f"/sync?filter={filter}", access_token=self.user_token
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
room_timeline = channel.json_body["rooms"]["join"][self.room]["timeline"]
|
room_timeline = channel.json_body["rooms"]["join"][self.room]["timeline"]
|
||||||
sync_prev_batch = room_timeline["prev_batch"]
|
sync_prev_batch = room_timeline["prev_batch"]
|
||||||
self.assertIsNotNone(sync_prev_batch)
|
self.assertIsNotNone(sync_prev_batch)
|
||||||
@ -335,7 +335,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/rooms/{self.room}/messages?dir=b&limit=1",
|
f"/rooms/{self.room}/messages?dir=b&limit=1",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
messages_end = channel.json_body["end"]
|
messages_end = channel.json_body["end"]
|
||||||
self.assertIsNotNone(messages_end)
|
self.assertIsNotNone(messages_end)
|
||||||
# Ensure the relation event is not in the chunk returned from /messages.
|
# Ensure the relation event is not in the chunk returned from /messages.
|
||||||
@ -355,7 +355,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?from={from_token}",
|
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?from={from_token}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# The relation should be in the returned chunk.
|
# The relation should be in the returned chunk.
|
||||||
self.assertIn(
|
self.assertIn(
|
||||||
@ -386,7 +386,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
key=key,
|
key=key,
|
||||||
access_token=access_tokens[idx],
|
access_token=access_tokens[idx],
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
idx += 1
|
idx += 1
|
||||||
idx %= len(access_tokens)
|
idx %= len(access_tokens)
|
||||||
@ -404,7 +404,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
% (self.room, self.parent_id, from_token),
|
% (self.room, self.parent_id, from_token),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
self.assertEqual(len(channel.json_body["chunk"]), 1, channel.json_body)
|
self.assertEqual(len(channel.json_body["chunk"]), 1, channel.json_body)
|
||||||
|
|
||||||
@ -419,13 +419,13 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
next_batch = channel.json_body.get("next_batch")
|
next_batch = channel.json_body.get("next_batch")
|
||||||
|
|
||||||
self.assertNotEquals(prev_token, next_batch)
|
self.assertNotEqual(prev_token, next_batch)
|
||||||
prev_token = next_batch
|
prev_token = next_batch
|
||||||
|
|
||||||
if not prev_token:
|
if not prev_token:
|
||||||
break
|
break
|
||||||
|
|
||||||
self.assertEquals(sent_groups, found_groups)
|
self.assertEqual(sent_groups, found_groups)
|
||||||
|
|
||||||
def test_aggregation_pagination_within_group(self):
|
def test_aggregation_pagination_within_group(self):
|
||||||
"""Test that we can paginate within an annotation group."""
|
"""Test that we can paginate within an annotation group."""
|
||||||
@ -449,14 +449,14 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
key="👍",
|
key="👍",
|
||||||
access_token=access_tokens[idx],
|
access_token=access_tokens[idx],
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
expected_event_ids.append(channel.json_body["event_id"])
|
expected_event_ids.append(channel.json_body["event_id"])
|
||||||
|
|
||||||
idx += 1
|
idx += 1
|
||||||
|
|
||||||
# Also send a different type of reaction so that we test we don't see it
|
# Also send a different type of reaction so that we test we don't see it
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", key="a")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", key="a")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
prev_token = ""
|
prev_token = ""
|
||||||
found_event_ids: List[str] = []
|
found_event_ids: List[str] = []
|
||||||
@ -473,7 +473,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/m.reaction/{encoded_key}?limit=1{from_token}",
|
f"/m.reaction/{encoded_key}?limit=1{from_token}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
self.assertEqual(len(channel.json_body["chunk"]), 1, channel.json_body)
|
self.assertEqual(len(channel.json_body["chunk"]), 1, channel.json_body)
|
||||||
|
|
||||||
@ -481,7 +481,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
next_batch = channel.json_body.get("next_batch")
|
next_batch = channel.json_body.get("next_batch")
|
||||||
|
|
||||||
self.assertNotEquals(prev_token, next_batch)
|
self.assertNotEqual(prev_token, next_batch)
|
||||||
prev_token = next_batch
|
prev_token = next_batch
|
||||||
|
|
||||||
if not prev_token:
|
if not prev_token:
|
||||||
@ -489,7 +489,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# We paginated backwards, so reverse
|
# We paginated backwards, so reverse
|
||||||
found_event_ids.reverse()
|
found_event_ids.reverse()
|
||||||
self.assertEquals(found_event_ids, expected_event_ids)
|
self.assertEqual(found_event_ids, expected_event_ids)
|
||||||
|
|
||||||
# Reset and try again, but convert the tokens to the legacy format.
|
# Reset and try again, but convert the tokens to the legacy format.
|
||||||
prev_token = ""
|
prev_token = ""
|
||||||
@ -506,7 +506,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/m.reaction/{encoded_key}?limit=1{from_token}",
|
f"/m.reaction/{encoded_key}?limit=1{from_token}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
self.assertEqual(len(channel.json_body["chunk"]), 1, channel.json_body)
|
self.assertEqual(len(channel.json_body["chunk"]), 1, channel.json_body)
|
||||||
|
|
||||||
@ -514,7 +514,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
next_batch = channel.json_body.get("next_batch")
|
next_batch = channel.json_body.get("next_batch")
|
||||||
|
|
||||||
self.assertNotEquals(prev_token, next_batch)
|
self.assertNotEqual(prev_token, next_batch)
|
||||||
prev_token = next_batch
|
prev_token = next_batch
|
||||||
|
|
||||||
if not prev_token:
|
if not prev_token:
|
||||||
@ -522,21 +522,21 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# We paginated backwards, so reverse
|
# We paginated backwards, so reverse
|
||||||
found_event_ids.reverse()
|
found_event_ids.reverse()
|
||||||
self.assertEquals(found_event_ids, expected_event_ids)
|
self.assertEqual(found_event_ids, expected_event_ids)
|
||||||
|
|
||||||
def test_aggregation(self):
|
def test_aggregation(self):
|
||||||
"""Test that annotations get correctly aggregated."""
|
"""Test that annotations get correctly aggregated."""
|
||||||
|
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "a")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "a")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
channel = self._send_relation(
|
channel = self._send_relation(
|
||||||
RelationTypes.ANNOTATION, "m.reaction", "a", access_token=self.user2_token
|
RelationTypes.ANNOTATION, "m.reaction", "a", access_token=self.user2_token
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "b")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "b")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET",
|
"GET",
|
||||||
@ -544,9 +544,9 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
% (self.room, self.parent_id),
|
% (self.room, self.parent_id),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
channel.json_body,
|
channel.json_body,
|
||||||
{
|
{
|
||||||
"chunk": [
|
"chunk": [
|
||||||
@ -560,13 +560,13 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"""Test that annotations get correctly aggregated after a redaction."""
|
"""Test that annotations get correctly aggregated after a redaction."""
|
||||||
|
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "a")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "a")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
to_redact_event_id = channel.json_body["event_id"]
|
to_redact_event_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
channel = self._send_relation(
|
channel = self._send_relation(
|
||||||
RelationTypes.ANNOTATION, "m.reaction", "a", access_token=self.user2_token
|
RelationTypes.ANNOTATION, "m.reaction", "a", access_token=self.user2_token
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# Now lets redact one of the 'a' reactions
|
# Now lets redact one of the 'a' reactions
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -575,7 +575,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
content={},
|
content={},
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET",
|
"GET",
|
||||||
@ -583,9 +583,9 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
% (self.room, self.parent_id),
|
% (self.room, self.parent_id),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
channel.json_body,
|
channel.json_body,
|
||||||
{"chunk": [{"type": "m.reaction", "key": "a", "count": 1}]},
|
{"chunk": [{"type": "m.reaction", "key": "a", "count": 1}]},
|
||||||
)
|
)
|
||||||
@ -599,7 +599,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
% (self.room, self.parent_id, RelationTypes.REPLACE),
|
% (self.room, self.parent_id, RelationTypes.REPLACE),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(400, channel.code, channel.json_body)
|
self.assertEqual(400, channel.code, channel.json_body)
|
||||||
|
|
||||||
@unittest.override_config(
|
@unittest.override_config(
|
||||||
{"experimental_features": {"msc3440_enabled": True, "msc3666_enabled": True}}
|
{"experimental_features": {"msc3440_enabled": True, "msc3666_enabled": True}}
|
||||||
@ -615,29 +615,29 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"""
|
"""
|
||||||
# Setup by sending a variety of relations.
|
# Setup by sending a variety of relations.
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "a")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "a")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
channel = self._send_relation(
|
channel = self._send_relation(
|
||||||
RelationTypes.ANNOTATION, "m.reaction", "a", access_token=self.user2_token
|
RelationTypes.ANNOTATION, "m.reaction", "a", access_token=self.user2_token
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "b")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "b")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
channel = self._send_relation(RelationTypes.REFERENCE, "m.room.test")
|
channel = self._send_relation(RelationTypes.REFERENCE, "m.room.test")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
reply_1 = channel.json_body["event_id"]
|
reply_1 = channel.json_body["event_id"]
|
||||||
|
|
||||||
channel = self._send_relation(RelationTypes.REFERENCE, "m.room.test")
|
channel = self._send_relation(RelationTypes.REFERENCE, "m.room.test")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
reply_2 = channel.json_body["event_id"]
|
reply_2 = channel.json_body["event_id"]
|
||||||
|
|
||||||
channel = self._send_relation(RelationTypes.THREAD, "m.room.test")
|
channel = self._send_relation(RelationTypes.THREAD, "m.room.test")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
channel = self._send_relation(RelationTypes.THREAD, "m.room.test")
|
channel = self._send_relation(RelationTypes.THREAD, "m.room.test")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
thread_2 = channel.json_body["event_id"]
|
thread_2 = channel.json_body["event_id"]
|
||||||
|
|
||||||
def assert_bundle(event_json: JsonDict) -> None:
|
def assert_bundle(event_json: JsonDict) -> None:
|
||||||
@ -655,7 +655,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
# Check the values of each field.
|
# Check the values of each field.
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
{
|
{
|
||||||
"chunk": [
|
"chunk": [
|
||||||
{"type": "m.reaction", "key": "a", "count": 2},
|
{"type": "m.reaction", "key": "a", "count": 2},
|
||||||
@ -665,12 +665,12 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
relations_dict[RelationTypes.ANNOTATION],
|
relations_dict[RelationTypes.ANNOTATION],
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
{"chunk": [{"event_id": reply_1}, {"event_id": reply_2}]},
|
{"chunk": [{"event_id": reply_1}, {"event_id": reply_2}]},
|
||||||
relations_dict[RelationTypes.REFERENCE],
|
relations_dict[RelationTypes.REFERENCE],
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
2,
|
2,
|
||||||
relations_dict[RelationTypes.THREAD].get("count"),
|
relations_dict[RelationTypes.THREAD].get("count"),
|
||||||
)
|
)
|
||||||
@ -701,7 +701,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/rooms/{self.room}/event/{self.parent_id}",
|
f"/rooms/{self.room}/event/{self.parent_id}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
assert_bundle(channel.json_body)
|
assert_bundle(channel.json_body)
|
||||||
|
|
||||||
# Request the room messages.
|
# Request the room messages.
|
||||||
@ -710,7 +710,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/rooms/{self.room}/messages?dir=b",
|
f"/rooms/{self.room}/messages?dir=b",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
assert_bundle(self._find_event_in_chunk(channel.json_body["chunk"]))
|
assert_bundle(self._find_event_in_chunk(channel.json_body["chunk"]))
|
||||||
|
|
||||||
# Request the room context.
|
# Request the room context.
|
||||||
@ -719,12 +719,12 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/rooms/{self.room}/context/{self.parent_id}",
|
f"/rooms/{self.room}/context/{self.parent_id}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
assert_bundle(channel.json_body["event"])
|
assert_bundle(channel.json_body["event"])
|
||||||
|
|
||||||
# Request sync.
|
# Request sync.
|
||||||
channel = self.make_request("GET", "/sync", access_token=self.user_token)
|
channel = self.make_request("GET", "/sync", access_token=self.user_token)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
room_timeline = channel.json_body["rooms"]["join"][self.room]["timeline"]
|
room_timeline = channel.json_body["rooms"]["join"][self.room]["timeline"]
|
||||||
self.assertTrue(room_timeline["limited"])
|
self.assertTrue(room_timeline["limited"])
|
||||||
assert_bundle(self._find_event_in_chunk(room_timeline["events"]))
|
assert_bundle(self._find_event_in_chunk(room_timeline["events"]))
|
||||||
@ -737,7 +737,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
content={"search_categories": {"room_events": {"search_term": "Hi"}}},
|
content={"search_categories": {"room_events": {"search_term": "Hi"}}},
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
chunk = [
|
chunk = [
|
||||||
result["result"]
|
result["result"]
|
||||||
for result in channel.json_body["search_categories"]["room_events"][
|
for result in channel.json_body["search_categories"]["room_events"][
|
||||||
@ -751,42 +751,42 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
when directly requested.
|
when directly requested.
|
||||||
"""
|
"""
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "a")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", "a")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
annotation_id = channel.json_body["event_id"]
|
annotation_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
# Annotate the annotation.
|
# Annotate the annotation.
|
||||||
channel = self._send_relation(
|
channel = self._send_relation(
|
||||||
RelationTypes.ANNOTATION, "m.reaction", "a", parent_id=annotation_id
|
RelationTypes.ANNOTATION, "m.reaction", "a", parent_id=annotation_id
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET",
|
"GET",
|
||||||
f"/rooms/{self.room}/event/{annotation_id}",
|
f"/rooms/{self.room}/event/{annotation_id}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
self.assertIsNone(channel.json_body["unsigned"].get("m.relations"))
|
self.assertIsNone(channel.json_body["unsigned"].get("m.relations"))
|
||||||
|
|
||||||
def test_aggregation_get_event_for_thread(self):
|
def test_aggregation_get_event_for_thread(self):
|
||||||
"""Test that threads get bundled aggregations included when directly requested."""
|
"""Test that threads get bundled aggregations included when directly requested."""
|
||||||
channel = self._send_relation(RelationTypes.THREAD, "m.room.test")
|
channel = self._send_relation(RelationTypes.THREAD, "m.room.test")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
thread_id = channel.json_body["event_id"]
|
thread_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
# Annotate the annotation.
|
# Annotate the annotation.
|
||||||
channel = self._send_relation(
|
channel = self._send_relation(
|
||||||
RelationTypes.ANNOTATION, "m.reaction", "a", parent_id=thread_id
|
RelationTypes.ANNOTATION, "m.reaction", "a", parent_id=thread_id
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET",
|
"GET",
|
||||||
f"/rooms/{self.room}/event/{thread_id}",
|
f"/rooms/{self.room}/event/{thread_id}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
channel.json_body["unsigned"].get("m.relations"),
|
channel.json_body["unsigned"].get("m.relations"),
|
||||||
{
|
{
|
||||||
RelationTypes.ANNOTATION: {
|
RelationTypes.ANNOTATION: {
|
||||||
@ -801,11 +801,11 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?limit=1",
|
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?limit=1",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
self.assertEqual(len(channel.json_body["chunk"]), 1)
|
self.assertEqual(len(channel.json_body["chunk"]), 1)
|
||||||
|
|
||||||
thread_message = channel.json_body["chunk"][0]
|
thread_message = channel.json_body["chunk"][0]
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
thread_message["unsigned"].get("m.relations"),
|
thread_message["unsigned"].get("m.relations"),
|
||||||
{
|
{
|
||||||
RelationTypes.ANNOTATION: {
|
RelationTypes.ANNOTATION: {
|
||||||
@ -905,7 +905,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/_matrix/client/unstable/rooms/{room2}/relations/{parent_id}",
|
f"/_matrix/client/unstable/rooms/{room2}/relations/{parent_id}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
self.assertEqual(channel.json_body["chunk"], [])
|
self.assertEqual(channel.json_body["chunk"], [])
|
||||||
|
|
||||||
# And when fetching aggregations.
|
# And when fetching aggregations.
|
||||||
@ -914,7 +914,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/_matrix/client/unstable/rooms/{room2}/aggregations/{parent_id}",
|
f"/_matrix/client/unstable/rooms/{room2}/aggregations/{parent_id}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
self.assertEqual(channel.json_body["chunk"], [])
|
self.assertEqual(channel.json_body["chunk"], [])
|
||||||
|
|
||||||
# And for bundled aggregations.
|
# And for bundled aggregations.
|
||||||
@ -923,7 +923,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/rooms/{room2}/event/{parent_id}",
|
f"/rooms/{room2}/event/{parent_id}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
self.assertNotIn("m.relations", channel.json_body["unsigned"])
|
self.assertNotIn("m.relations", channel.json_body["unsigned"])
|
||||||
|
|
||||||
@unittest.override_config({"experimental_features": {"msc3666_enabled": True}})
|
@unittest.override_config({"experimental_features": {"msc3666_enabled": True}})
|
||||||
@ -936,7 +936,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"m.room.message",
|
"m.room.message",
|
||||||
content={"msgtype": "m.text", "body": "foo", "m.new_content": new_body},
|
content={"msgtype": "m.text", "body": "foo", "m.new_content": new_body},
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
edit_event_id = channel.json_body["event_id"]
|
edit_event_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
@ -958,8 +958,8 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/rooms/{self.room}/event/{self.parent_id}",
|
f"/rooms/{self.room}/event/{self.parent_id}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
self.assertEquals(channel.json_body["content"], new_body)
|
self.assertEqual(channel.json_body["content"], new_body)
|
||||||
assert_bundle(channel.json_body)
|
assert_bundle(channel.json_body)
|
||||||
|
|
||||||
# Request the room messages.
|
# Request the room messages.
|
||||||
@ -968,7 +968,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/rooms/{self.room}/messages?dir=b",
|
f"/rooms/{self.room}/messages?dir=b",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
assert_bundle(self._find_event_in_chunk(channel.json_body["chunk"]))
|
assert_bundle(self._find_event_in_chunk(channel.json_body["chunk"]))
|
||||||
|
|
||||||
# Request the room context.
|
# Request the room context.
|
||||||
@ -977,7 +977,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/rooms/{self.room}/context/{self.parent_id}",
|
f"/rooms/{self.room}/context/{self.parent_id}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
assert_bundle(channel.json_body["event"])
|
assert_bundle(channel.json_body["event"])
|
||||||
|
|
||||||
# Request sync, but limit the timeline so it becomes limited (and includes
|
# Request sync, but limit the timeline so it becomes limited (and includes
|
||||||
@ -988,7 +988,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET", f"/sync?filter={filter}", access_token=self.user_token
|
"GET", f"/sync?filter={filter}", access_token=self.user_token
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
room_timeline = channel.json_body["rooms"]["join"][self.room]["timeline"]
|
room_timeline = channel.json_body["rooms"]["join"][self.room]["timeline"]
|
||||||
self.assertTrue(room_timeline["limited"])
|
self.assertTrue(room_timeline["limited"])
|
||||||
assert_bundle(self._find_event_in_chunk(room_timeline["events"]))
|
assert_bundle(self._find_event_in_chunk(room_timeline["events"]))
|
||||||
@ -1001,7 +1001,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
content={"search_categories": {"room_events": {"search_term": "Hi"}}},
|
content={"search_categories": {"room_events": {"search_term": "Hi"}}},
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
chunk = [
|
chunk = [
|
||||||
result["result"]
|
result["result"]
|
||||||
for result in channel.json_body["search_categories"]["room_events"][
|
for result in channel.json_body["search_categories"]["room_events"][
|
||||||
@ -1024,7 +1024,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"m.new_content": {"msgtype": "m.text", "body": "First edit"},
|
"m.new_content": {"msgtype": "m.text", "body": "First edit"},
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
new_body = {"msgtype": "m.text", "body": "I've been edited!"}
|
new_body = {"msgtype": "m.text", "body": "I've been edited!"}
|
||||||
channel = self._send_relation(
|
channel = self._send_relation(
|
||||||
@ -1032,7 +1032,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"m.room.message",
|
"m.room.message",
|
||||||
content={"msgtype": "m.text", "body": "foo", "m.new_content": new_body},
|
content={"msgtype": "m.text", "body": "foo", "m.new_content": new_body},
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
edit_event_id = channel.json_body["event_id"]
|
edit_event_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
@ -1045,16 +1045,16 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"m.new_content": {"msgtype": "m.text", "body": "Edit, but wrong type"},
|
"m.new_content": {"msgtype": "m.text", "body": "Edit, but wrong type"},
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET",
|
"GET",
|
||||||
"/rooms/%s/event/%s" % (self.room, self.parent_id),
|
"/rooms/%s/event/%s" % (self.room, self.parent_id),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
self.assertEquals(channel.json_body["content"], new_body)
|
self.assertEqual(channel.json_body["content"], new_body)
|
||||||
|
|
||||||
relations_dict = channel.json_body["unsigned"].get("m.relations")
|
relations_dict = channel.json_body["unsigned"].get("m.relations")
|
||||||
self.assertIn(RelationTypes.REPLACE, relations_dict)
|
self.assertIn(RelationTypes.REPLACE, relations_dict)
|
||||||
@ -1076,7 +1076,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"m.room.message",
|
"m.room.message",
|
||||||
content={"msgtype": "m.text", "body": "A reply!"},
|
content={"msgtype": "m.text", "body": "A reply!"},
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
reply = channel.json_body["event_id"]
|
reply = channel.json_body["event_id"]
|
||||||
|
|
||||||
new_body = {"msgtype": "m.text", "body": "I've been edited!"}
|
new_body = {"msgtype": "m.text", "body": "I've been edited!"}
|
||||||
@ -1086,7 +1086,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
content={"msgtype": "m.text", "body": "foo", "m.new_content": new_body},
|
content={"msgtype": "m.text", "body": "foo", "m.new_content": new_body},
|
||||||
parent_id=reply,
|
parent_id=reply,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
edit_event_id = channel.json_body["event_id"]
|
edit_event_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
@ -1095,7 +1095,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"/rooms/%s/event/%s" % (self.room, reply),
|
"/rooms/%s/event/%s" % (self.room, reply),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# We expect to see the new body in the dict, as well as the reference
|
# We expect to see the new body in the dict, as well as the reference
|
||||||
# metadata sill intact.
|
# metadata sill intact.
|
||||||
@ -1133,7 +1133,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"m.room.message",
|
"m.room.message",
|
||||||
content={"msgtype": "m.text", "body": "A threaded reply!"},
|
content={"msgtype": "m.text", "body": "A threaded reply!"},
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
threaded_event_id = channel.json_body["event_id"]
|
threaded_event_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
new_body = {"msgtype": "m.text", "body": "I've been edited!"}
|
new_body = {"msgtype": "m.text", "body": "I've been edited!"}
|
||||||
@ -1143,7 +1143,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
content={"msgtype": "m.text", "body": "foo", "m.new_content": new_body},
|
content={"msgtype": "m.text", "body": "foo", "m.new_content": new_body},
|
||||||
parent_id=threaded_event_id,
|
parent_id=threaded_event_id,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# Fetch the thread root, to get the bundled aggregation for the thread.
|
# Fetch the thread root, to get the bundled aggregation for the thread.
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -1151,7 +1151,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/rooms/{self.room}/event/{self.parent_id}",
|
f"/rooms/{self.room}/event/{self.parent_id}",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# We expect that the edit message appears in the thread summary in the
|
# We expect that the edit message appears in the thread summary in the
|
||||||
# unsigned relations section.
|
# unsigned relations section.
|
||||||
@ -1161,9 +1161,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
thread_summary = relations_dict[RelationTypes.THREAD]
|
thread_summary = relations_dict[RelationTypes.THREAD]
|
||||||
self.assertIn("latest_event", thread_summary)
|
self.assertIn("latest_event", thread_summary)
|
||||||
latest_event_in_thread = thread_summary["latest_event"]
|
latest_event_in_thread = thread_summary["latest_event"]
|
||||||
self.assertEquals(
|
self.assertEqual(latest_event_in_thread["content"]["body"], "I've been edited!")
|
||||||
latest_event_in_thread["content"]["body"], "I've been edited!"
|
|
||||||
)
|
|
||||||
|
|
||||||
def test_edit_edit(self):
|
def test_edit_edit(self):
|
||||||
"""Test that an edit cannot be edited."""
|
"""Test that an edit cannot be edited."""
|
||||||
@ -1177,7 +1175,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"m.new_content": new_body,
|
"m.new_content": new_body,
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
edit_event_id = channel.json_body["event_id"]
|
edit_event_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
# Edit the edit event.
|
# Edit the edit event.
|
||||||
@ -1191,7 +1189,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
},
|
},
|
||||||
parent_id=edit_event_id,
|
parent_id=edit_event_id,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# Request the original event.
|
# Request the original event.
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -1199,9 +1197,9 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"/rooms/%s/event/%s" % (self.room, self.parent_id),
|
"/rooms/%s/event/%s" % (self.room, self.parent_id),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
# The edit to the edit should be ignored.
|
# The edit to the edit should be ignored.
|
||||||
self.assertEquals(channel.json_body["content"], new_body)
|
self.assertEqual(channel.json_body["content"], new_body)
|
||||||
|
|
||||||
# The relations information should not include the edit to the edit.
|
# The relations information should not include the edit to the edit.
|
||||||
relations_dict = channel.json_body["unsigned"].get("m.relations")
|
relations_dict = channel.json_body["unsigned"].get("m.relations")
|
||||||
@ -1234,7 +1232,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"m.new_content": {"msgtype": "m.text", "body": "First edit"},
|
"m.new_content": {"msgtype": "m.text", "body": "First edit"},
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# Check the relation is returned
|
# Check the relation is returned
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -1243,10 +1241,10 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
% (self.room, original_event_id),
|
% (self.room, original_event_id),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
self.assertIn("chunk", channel.json_body)
|
self.assertIn("chunk", channel.json_body)
|
||||||
self.assertEquals(len(channel.json_body["chunk"]), 1)
|
self.assertEqual(len(channel.json_body["chunk"]), 1)
|
||||||
|
|
||||||
# Redact the original event
|
# Redact the original event
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -1256,7 +1254,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
content="{}",
|
content="{}",
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# Try to check for remaining m.replace relations
|
# Try to check for remaining m.replace relations
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -1265,11 +1263,11 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
% (self.room, original_event_id),
|
% (self.room, original_event_id),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# Check that no relations are returned
|
# Check that no relations are returned
|
||||||
self.assertIn("chunk", channel.json_body)
|
self.assertIn("chunk", channel.json_body)
|
||||||
self.assertEquals(channel.json_body["chunk"], [])
|
self.assertEqual(channel.json_body["chunk"], [])
|
||||||
|
|
||||||
def test_aggregations_redaction_prevents_access_to_aggregations(self):
|
def test_aggregations_redaction_prevents_access_to_aggregations(self):
|
||||||
"""Test that annotations of an event are redacted when the original event
|
"""Test that annotations of an event are redacted when the original event
|
||||||
@ -1283,7 +1281,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
channel = self._send_relation(
|
channel = self._send_relation(
|
||||||
RelationTypes.ANNOTATION, "m.reaction", key="👍", parent_id=original_event_id
|
RelationTypes.ANNOTATION, "m.reaction", key="👍", parent_id=original_event_id
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# Redact the original
|
# Redact the original
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -1297,7 +1295,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
content="{}",
|
content="{}",
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# Check that aggregations returns zero
|
# Check that aggregations returns zero
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -1306,15 +1304,15 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
% (self.room, original_event_id),
|
% (self.room, original_event_id),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
self.assertIn("chunk", channel.json_body)
|
self.assertIn("chunk", channel.json_body)
|
||||||
self.assertEquals(channel.json_body["chunk"], [])
|
self.assertEqual(channel.json_body["chunk"], [])
|
||||||
|
|
||||||
def test_unknown_relations(self):
|
def test_unknown_relations(self):
|
||||||
"""Unknown relations should be accepted."""
|
"""Unknown relations should be accepted."""
|
||||||
channel = self._send_relation("m.relation.test", "m.room.test")
|
channel = self._send_relation("m.relation.test", "m.room.test")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
event_id = channel.json_body["event_id"]
|
event_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -1323,18 +1321,18 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
% (self.room, self.parent_id),
|
% (self.room, self.parent_id),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
|
|
||||||
# We expect to get back a single pagination result, which is the full
|
# We expect to get back a single pagination result, which is the full
|
||||||
# relation event we sent above.
|
# relation event we sent above.
|
||||||
self.assertEquals(len(channel.json_body["chunk"]), 1, channel.json_body)
|
self.assertEqual(len(channel.json_body["chunk"]), 1, channel.json_body)
|
||||||
self.assert_dict(
|
self.assert_dict(
|
||||||
{"event_id": event_id, "sender": self.user_id, "type": "m.room.test"},
|
{"event_id": event_id, "sender": self.user_id, "type": "m.room.test"},
|
||||||
channel.json_body["chunk"][0],
|
channel.json_body["chunk"][0],
|
||||||
)
|
)
|
||||||
|
|
||||||
# We also expect to get the original event (the id of which is self.parent_id)
|
# We also expect to get the original event (the id of which is self.parent_id)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
channel.json_body["original_event"]["event_id"], self.parent_id
|
channel.json_body["original_event"]["event_id"], self.parent_id
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -1344,7 +1342,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
"/rooms/%s/event/%s" % (self.room, self.parent_id),
|
"/rooms/%s/event/%s" % (self.room, self.parent_id),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
self.assertNotIn("m.relations", channel.json_body["unsigned"])
|
self.assertNotIn("m.relations", channel.json_body["unsigned"])
|
||||||
|
|
||||||
# But unknown relations can be directly queried.
|
# But unknown relations can be directly queried.
|
||||||
@ -1354,8 +1352,8 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
% (self.room, self.parent_id),
|
% (self.room, self.parent_id),
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
self.assertEquals(channel.json_body["chunk"], [])
|
self.assertEqual(channel.json_body["chunk"], [])
|
||||||
|
|
||||||
def _find_event_in_chunk(self, events: List[JsonDict]) -> JsonDict:
|
def _find_event_in_chunk(self, events: List[JsonDict]) -> JsonDict:
|
||||||
"""
|
"""
|
||||||
@ -1422,15 +1420,15 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
def test_background_update(self):
|
def test_background_update(self):
|
||||||
"""Test the event_arbitrary_relations background update."""
|
"""Test the event_arbitrary_relations background update."""
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", key="👍")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", key="👍")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
annotation_event_id_good = channel.json_body["event_id"]
|
annotation_event_id_good = channel.json_body["event_id"]
|
||||||
|
|
||||||
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", key="A")
|
channel = self._send_relation(RelationTypes.ANNOTATION, "m.reaction", key="A")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
annotation_event_id_bad = channel.json_body["event_id"]
|
annotation_event_id_bad = channel.json_body["event_id"]
|
||||||
|
|
||||||
channel = self._send_relation(RelationTypes.THREAD, "m.room.test")
|
channel = self._send_relation(RelationTypes.THREAD, "m.room.test")
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
thread_event_id = channel.json_body["event_id"]
|
thread_event_id = channel.json_body["event_id"]
|
||||||
|
|
||||||
# Clean-up the table as if the inserts did not happen during event creation.
|
# Clean-up the table as if the inserts did not happen during event creation.
|
||||||
@ -1450,8 +1448,8 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?limit=10",
|
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?limit=10",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
[ev["event_id"] for ev in channel.json_body["chunk"]],
|
[ev["event_id"] for ev in channel.json_body["chunk"]],
|
||||||
[annotation_event_id_good],
|
[annotation_event_id_good],
|
||||||
)
|
)
|
||||||
@ -1475,7 +1473,7 @@ class RelationsTestCase(unittest.HomeserverTestCase):
|
|||||||
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?limit=10",
|
f"/_matrix/client/unstable/rooms/{self.room}/relations/{self.parent_id}?limit=10",
|
||||||
access_token=self.user_token,
|
access_token=self.user_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.json_body)
|
self.assertEqual(200, channel.code, channel.json_body)
|
||||||
self.assertCountEqual(
|
self.assertCountEqual(
|
||||||
[ev["event_id"] for ev in channel.json_body["chunk"]],
|
[ev["event_id"] for ev in channel.json_body["chunk"]],
|
||||||
[annotation_event_id_good, thread_event_id],
|
[annotation_event_id_good, thread_event_id],
|
||||||
|
@ -95,7 +95,7 @@ class RoomPermissionsTestCase(RoomBase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"PUT", self.created_rmid_msg_path, b'{"msgtype":"m.text","body":"test msg"}'
|
"PUT", self.created_rmid_msg_path, b'{"msgtype":"m.text","body":"test msg"}'
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
# set topic for public room
|
# set topic for public room
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -103,7 +103,7 @@ class RoomPermissionsTestCase(RoomBase):
|
|||||||
("rooms/%s/state/m.room.topic" % self.created_public_rmid).encode("ascii"),
|
("rooms/%s/state/m.room.topic" % self.created_public_rmid).encode("ascii"),
|
||||||
b'{"topic":"Public Room Topic"}',
|
b'{"topic":"Public Room Topic"}',
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
# auth as user_id now
|
# auth as user_id now
|
||||||
self.helper.auth_user_id = self.user_id
|
self.helper.auth_user_id = self.user_id
|
||||||
@ -125,28 +125,28 @@ class RoomPermissionsTestCase(RoomBase):
|
|||||||
"/rooms/%s/send/m.room.message/mid2" % (self.uncreated_rmid,),
|
"/rooms/%s/send/m.room.message/mid2" % (self.uncreated_rmid,),
|
||||||
msg_content,
|
msg_content,
|
||||||
)
|
)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# send message in created room not joined (no state), expect 403
|
# send message in created room not joined (no state), expect 403
|
||||||
channel = self.make_request("PUT", send_msg_path(), msg_content)
|
channel = self.make_request("PUT", send_msg_path(), msg_content)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# send message in created room and invited, expect 403
|
# send message in created room and invited, expect 403
|
||||||
self.helper.invite(
|
self.helper.invite(
|
||||||
room=self.created_rmid, src=self.rmcreator_id, targ=self.user_id
|
room=self.created_rmid, src=self.rmcreator_id, targ=self.user_id
|
||||||
)
|
)
|
||||||
channel = self.make_request("PUT", send_msg_path(), msg_content)
|
channel = self.make_request("PUT", send_msg_path(), msg_content)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# send message in created room and joined, expect 200
|
# send message in created room and joined, expect 200
|
||||||
self.helper.join(room=self.created_rmid, user=self.user_id)
|
self.helper.join(room=self.created_rmid, user=self.user_id)
|
||||||
channel = self.make_request("PUT", send_msg_path(), msg_content)
|
channel = self.make_request("PUT", send_msg_path(), msg_content)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# send message in created room and left, expect 403
|
# send message in created room and left, expect 403
|
||||||
self.helper.leave(room=self.created_rmid, user=self.user_id)
|
self.helper.leave(room=self.created_rmid, user=self.user_id)
|
||||||
channel = self.make_request("PUT", send_msg_path(), msg_content)
|
channel = self.make_request("PUT", send_msg_path(), msg_content)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
def test_topic_perms(self):
|
def test_topic_perms(self):
|
||||||
topic_content = b'{"topic":"My Topic Name"}'
|
topic_content = b'{"topic":"My Topic Name"}'
|
||||||
@ -156,28 +156,28 @@ class RoomPermissionsTestCase(RoomBase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"PUT", "/rooms/%s/state/m.room.topic" % self.uncreated_rmid, topic_content
|
"PUT", "/rooms/%s/state/m.room.topic" % self.uncreated_rmid, topic_content
|
||||||
)
|
)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET", "/rooms/%s/state/m.room.topic" % self.uncreated_rmid
|
"GET", "/rooms/%s/state/m.room.topic" % self.uncreated_rmid
|
||||||
)
|
)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# set/get topic in created PRIVATE room not joined, expect 403
|
# set/get topic in created PRIVATE room not joined, expect 403
|
||||||
channel = self.make_request("PUT", topic_path, topic_content)
|
channel = self.make_request("PUT", topic_path, topic_content)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
channel = self.make_request("GET", topic_path)
|
channel = self.make_request("GET", topic_path)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# set topic in created PRIVATE room and invited, expect 403
|
# set topic in created PRIVATE room and invited, expect 403
|
||||||
self.helper.invite(
|
self.helper.invite(
|
||||||
room=self.created_rmid, src=self.rmcreator_id, targ=self.user_id
|
room=self.created_rmid, src=self.rmcreator_id, targ=self.user_id
|
||||||
)
|
)
|
||||||
channel = self.make_request("PUT", topic_path, topic_content)
|
channel = self.make_request("PUT", topic_path, topic_content)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# get topic in created PRIVATE room and invited, expect 403
|
# get topic in created PRIVATE room and invited, expect 403
|
||||||
channel = self.make_request("GET", topic_path)
|
channel = self.make_request("GET", topic_path)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# set/get topic in created PRIVATE room and joined, expect 200
|
# set/get topic in created PRIVATE room and joined, expect 200
|
||||||
self.helper.join(room=self.created_rmid, user=self.user_id)
|
self.helper.join(room=self.created_rmid, user=self.user_id)
|
||||||
@ -185,25 +185,25 @@ class RoomPermissionsTestCase(RoomBase):
|
|||||||
# Only room ops can set topic by default
|
# Only room ops can set topic by default
|
||||||
self.helper.auth_user_id = self.rmcreator_id
|
self.helper.auth_user_id = self.rmcreator_id
|
||||||
channel = self.make_request("PUT", topic_path, topic_content)
|
channel = self.make_request("PUT", topic_path, topic_content)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
self.helper.auth_user_id = self.user_id
|
self.helper.auth_user_id = self.user_id
|
||||||
|
|
||||||
channel = self.make_request("GET", topic_path)
|
channel = self.make_request("GET", topic_path)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
self.assert_dict(json.loads(topic_content.decode("utf8")), channel.json_body)
|
self.assert_dict(json.loads(topic_content.decode("utf8")), channel.json_body)
|
||||||
|
|
||||||
# set/get topic in created PRIVATE room and left, expect 403
|
# set/get topic in created PRIVATE room and left, expect 403
|
||||||
self.helper.leave(room=self.created_rmid, user=self.user_id)
|
self.helper.leave(room=self.created_rmid, user=self.user_id)
|
||||||
channel = self.make_request("PUT", topic_path, topic_content)
|
channel = self.make_request("PUT", topic_path, topic_content)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
channel = self.make_request("GET", topic_path)
|
channel = self.make_request("GET", topic_path)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# get topic in PUBLIC room, not joined, expect 403
|
# get topic in PUBLIC room, not joined, expect 403
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET", "/rooms/%s/state/m.room.topic" % self.created_public_rmid
|
"GET", "/rooms/%s/state/m.room.topic" % self.created_public_rmid
|
||||||
)
|
)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# set topic in PUBLIC room, not joined, expect 403
|
# set topic in PUBLIC room, not joined, expect 403
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -211,7 +211,7 @@ class RoomPermissionsTestCase(RoomBase):
|
|||||||
"/rooms/%s/state/m.room.topic" % self.created_public_rmid,
|
"/rooms/%s/state/m.room.topic" % self.created_public_rmid,
|
||||||
topic_content,
|
topic_content,
|
||||||
)
|
)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
def _test_get_membership(
|
def _test_get_membership(
|
||||||
self, room=None, members: Iterable = frozenset(), expect_code=None
|
self, room=None, members: Iterable = frozenset(), expect_code=None
|
||||||
@ -219,7 +219,7 @@ class RoomPermissionsTestCase(RoomBase):
|
|||||||
for member in members:
|
for member in members:
|
||||||
path = "/rooms/%s/state/m.room.member/%s" % (room, member)
|
path = "/rooms/%s/state/m.room.member/%s" % (room, member)
|
||||||
channel = self.make_request("GET", path)
|
channel = self.make_request("GET", path)
|
||||||
self.assertEquals(expect_code, channel.code)
|
self.assertEqual(expect_code, channel.code)
|
||||||
|
|
||||||
def test_membership_basic_room_perms(self):
|
def test_membership_basic_room_perms(self):
|
||||||
# === room does not exist ===
|
# === room does not exist ===
|
||||||
@ -478,16 +478,16 @@ class RoomsMemberListTestCase(RoomBase):
|
|||||||
def test_get_member_list(self):
|
def test_get_member_list(self):
|
||||||
room_id = self.helper.create_room_as(self.user_id)
|
room_id = self.helper.create_room_as(self.user_id)
|
||||||
channel = self.make_request("GET", "/rooms/%s/members" % room_id)
|
channel = self.make_request("GET", "/rooms/%s/members" % room_id)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
def test_get_member_list_no_room(self):
|
def test_get_member_list_no_room(self):
|
||||||
channel = self.make_request("GET", "/rooms/roomdoesnotexist/members")
|
channel = self.make_request("GET", "/rooms/roomdoesnotexist/members")
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
def test_get_member_list_no_permission(self):
|
def test_get_member_list_no_permission(self):
|
||||||
room_id = self.helper.create_room_as("@some_other_guy:red")
|
room_id = self.helper.create_room_as("@some_other_guy:red")
|
||||||
channel = self.make_request("GET", "/rooms/%s/members" % room_id)
|
channel = self.make_request("GET", "/rooms/%s/members" % room_id)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
def test_get_member_list_no_permission_with_at_token(self):
|
def test_get_member_list_no_permission_with_at_token(self):
|
||||||
"""
|
"""
|
||||||
@ -498,7 +498,7 @@ class RoomsMemberListTestCase(RoomBase):
|
|||||||
|
|
||||||
# first sync to get an at token
|
# first sync to get an at token
|
||||||
channel = self.make_request("GET", "/sync")
|
channel = self.make_request("GET", "/sync")
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
sync_token = channel.json_body["next_batch"]
|
sync_token = channel.json_body["next_batch"]
|
||||||
|
|
||||||
# check that permission is denied for @sid1:red to get the
|
# check that permission is denied for @sid1:red to get the
|
||||||
@ -507,7 +507,7 @@ class RoomsMemberListTestCase(RoomBase):
|
|||||||
"GET",
|
"GET",
|
||||||
f"/rooms/{room_id}/members?at={sync_token}",
|
f"/rooms/{room_id}/members?at={sync_token}",
|
||||||
)
|
)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
def test_get_member_list_no_permission_former_member(self):
|
def test_get_member_list_no_permission_former_member(self):
|
||||||
"""
|
"""
|
||||||
@ -520,14 +520,14 @@ class RoomsMemberListTestCase(RoomBase):
|
|||||||
|
|
||||||
# check that the user can see the member list to start with
|
# check that the user can see the member list to start with
|
||||||
channel = self.make_request("GET", "/rooms/%s/members" % room_id)
|
channel = self.make_request("GET", "/rooms/%s/members" % room_id)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# ban the user
|
# ban the user
|
||||||
self.helper.change_membership(room_id, "@alice:red", self.user_id, "ban")
|
self.helper.change_membership(room_id, "@alice:red", self.user_id, "ban")
|
||||||
|
|
||||||
# check the user can no longer see the member list
|
# check the user can no longer see the member list
|
||||||
channel = self.make_request("GET", "/rooms/%s/members" % room_id)
|
channel = self.make_request("GET", "/rooms/%s/members" % room_id)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
def test_get_member_list_no_permission_former_member_with_at_token(self):
|
def test_get_member_list_no_permission_former_member_with_at_token(self):
|
||||||
"""
|
"""
|
||||||
@ -541,14 +541,14 @@ class RoomsMemberListTestCase(RoomBase):
|
|||||||
|
|
||||||
# sync to get an at token
|
# sync to get an at token
|
||||||
channel = self.make_request("GET", "/sync")
|
channel = self.make_request("GET", "/sync")
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
sync_token = channel.json_body["next_batch"]
|
sync_token = channel.json_body["next_batch"]
|
||||||
|
|
||||||
# check that the user can see the member list to start with
|
# check that the user can see the member list to start with
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET", "/rooms/%s/members?at=%s" % (room_id, sync_token)
|
"GET", "/rooms/%s/members?at=%s" % (room_id, sync_token)
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# ban the user (Note: the user is actually allowed to see this event and
|
# ban the user (Note: the user is actually allowed to see this event and
|
||||||
# state so that they know they're banned!)
|
# state so that they know they're banned!)
|
||||||
@ -560,14 +560,14 @@ class RoomsMemberListTestCase(RoomBase):
|
|||||||
|
|
||||||
# now, with the original user, sync again to get a new at token
|
# now, with the original user, sync again to get a new at token
|
||||||
channel = self.make_request("GET", "/sync")
|
channel = self.make_request("GET", "/sync")
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
sync_token = channel.json_body["next_batch"]
|
sync_token = channel.json_body["next_batch"]
|
||||||
|
|
||||||
# check the user can no longer see the updated member list
|
# check the user can no longer see the updated member list
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET", "/rooms/%s/members?at=%s" % (room_id, sync_token)
|
"GET", "/rooms/%s/members?at=%s" % (room_id, sync_token)
|
||||||
)
|
)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
def test_get_member_list_mixed_memberships(self):
|
def test_get_member_list_mixed_memberships(self):
|
||||||
room_creator = "@some_other_guy:red"
|
room_creator = "@some_other_guy:red"
|
||||||
@ -576,17 +576,17 @@ class RoomsMemberListTestCase(RoomBase):
|
|||||||
self.helper.invite(room=room_id, src=room_creator, targ=self.user_id)
|
self.helper.invite(room=room_id, src=room_creator, targ=self.user_id)
|
||||||
# can't see list if you're just invited.
|
# can't see list if you're just invited.
|
||||||
channel = self.make_request("GET", room_path)
|
channel = self.make_request("GET", room_path)
|
||||||
self.assertEquals(403, channel.code, msg=channel.result["body"])
|
self.assertEqual(403, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
self.helper.join(room=room_id, user=self.user_id)
|
self.helper.join(room=room_id, user=self.user_id)
|
||||||
# can see list now joined
|
# can see list now joined
|
||||||
channel = self.make_request("GET", room_path)
|
channel = self.make_request("GET", room_path)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
self.helper.leave(room=room_id, user=self.user_id)
|
self.helper.leave(room=room_id, user=self.user_id)
|
||||||
# can see old list once left
|
# can see old list once left
|
||||||
channel = self.make_request("GET", room_path)
|
channel = self.make_request("GET", room_path)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
|
|
||||||
class RoomsCreateTestCase(RoomBase):
|
class RoomsCreateTestCase(RoomBase):
|
||||||
@ -598,19 +598,19 @@ class RoomsCreateTestCase(RoomBase):
|
|||||||
# POST with no config keys, expect new room id
|
# POST with no config keys, expect new room id
|
||||||
channel = self.make_request("POST", "/createRoom", "{}")
|
channel = self.make_request("POST", "/createRoom", "{}")
|
||||||
|
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertTrue("room_id" in channel.json_body)
|
self.assertTrue("room_id" in channel.json_body)
|
||||||
|
|
||||||
def test_post_room_visibility_key(self):
|
def test_post_room_visibility_key(self):
|
||||||
# POST with visibility config key, expect new room id
|
# POST with visibility config key, expect new room id
|
||||||
channel = self.make_request("POST", "/createRoom", b'{"visibility":"private"}')
|
channel = self.make_request("POST", "/createRoom", b'{"visibility":"private"}')
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
self.assertTrue("room_id" in channel.json_body)
|
self.assertTrue("room_id" in channel.json_body)
|
||||||
|
|
||||||
def test_post_room_custom_key(self):
|
def test_post_room_custom_key(self):
|
||||||
# POST with custom config keys, expect new room id
|
# POST with custom config keys, expect new room id
|
||||||
channel = self.make_request("POST", "/createRoom", b'{"custom":"stuff"}')
|
channel = self.make_request("POST", "/createRoom", b'{"custom":"stuff"}')
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
self.assertTrue("room_id" in channel.json_body)
|
self.assertTrue("room_id" in channel.json_body)
|
||||||
|
|
||||||
def test_post_room_known_and_unknown_keys(self):
|
def test_post_room_known_and_unknown_keys(self):
|
||||||
@ -618,16 +618,16 @@ class RoomsCreateTestCase(RoomBase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"POST", "/createRoom", b'{"visibility":"private","custom":"things"}'
|
"POST", "/createRoom", b'{"visibility":"private","custom":"things"}'
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
self.assertTrue("room_id" in channel.json_body)
|
self.assertTrue("room_id" in channel.json_body)
|
||||||
|
|
||||||
def test_post_room_invalid_content(self):
|
def test_post_room_invalid_content(self):
|
||||||
# POST with invalid content / paths, expect 400
|
# POST with invalid content / paths, expect 400
|
||||||
channel = self.make_request("POST", "/createRoom", b'{"visibili')
|
channel = self.make_request("POST", "/createRoom", b'{"visibili')
|
||||||
self.assertEquals(400, channel.code)
|
self.assertEqual(400, channel.code)
|
||||||
|
|
||||||
channel = self.make_request("POST", "/createRoom", b'["hello"]')
|
channel = self.make_request("POST", "/createRoom", b'["hello"]')
|
||||||
self.assertEquals(400, channel.code)
|
self.assertEqual(400, channel.code)
|
||||||
|
|
||||||
def test_post_room_invitees_invalid_mxid(self):
|
def test_post_room_invitees_invalid_mxid(self):
|
||||||
# POST with invalid invitee, see https://github.com/matrix-org/synapse/issues/4088
|
# POST with invalid invitee, see https://github.com/matrix-org/synapse/issues/4088
|
||||||
@ -635,7 +635,7 @@ class RoomsCreateTestCase(RoomBase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"POST", "/createRoom", b'{"invite":["@alice:example.com "]}'
|
"POST", "/createRoom", b'{"invite":["@alice:example.com "]}'
|
||||||
)
|
)
|
||||||
self.assertEquals(400, channel.code)
|
self.assertEqual(400, channel.code)
|
||||||
|
|
||||||
@unittest.override_config({"rc_invites": {"per_room": {"burst_count": 3}}})
|
@unittest.override_config({"rc_invites": {"per_room": {"burst_count": 3}}})
|
||||||
def test_post_room_invitees_ratelimit(self):
|
def test_post_room_invitees_ratelimit(self):
|
||||||
@ -694,9 +694,9 @@ class RoomsCreateTestCase(RoomBase):
|
|||||||
"/createRoom",
|
"/createRoom",
|
||||||
{},
|
{},
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.code, 200, channel.json_body)
|
self.assertEqual(channel.code, 200, channel.json_body)
|
||||||
|
|
||||||
self.assertEquals(join_mock.call_count, 0)
|
self.assertEqual(join_mock.call_count, 0)
|
||||||
|
|
||||||
|
|
||||||
class RoomTopicTestCase(RoomBase):
|
class RoomTopicTestCase(RoomBase):
|
||||||
@ -712,54 +712,54 @@ class RoomTopicTestCase(RoomBase):
|
|||||||
def test_invalid_puts(self):
|
def test_invalid_puts(self):
|
||||||
# missing keys or invalid json
|
# missing keys or invalid json
|
||||||
channel = self.make_request("PUT", self.path, "{}")
|
channel = self.make_request("PUT", self.path, "{}")
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", self.path, '{"_name":"bo"}')
|
channel = self.make_request("PUT", self.path, '{"_name":"bo"}')
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", self.path, '{"nao')
|
channel = self.make_request("PUT", self.path, '{"nao')
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"PUT", self.path, '[{"_name":"bo"},{"_name":"jill"}]'
|
"PUT", self.path, '[{"_name":"bo"},{"_name":"jill"}]'
|
||||||
)
|
)
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", self.path, "text only")
|
channel = self.make_request("PUT", self.path, "text only")
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", self.path, "")
|
channel = self.make_request("PUT", self.path, "")
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# valid key, wrong type
|
# valid key, wrong type
|
||||||
content = '{"topic":["Topic name"]}'
|
content = '{"topic":["Topic name"]}'
|
||||||
channel = self.make_request("PUT", self.path, content)
|
channel = self.make_request("PUT", self.path, content)
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
def test_rooms_topic(self):
|
def test_rooms_topic(self):
|
||||||
# nothing should be there
|
# nothing should be there
|
||||||
channel = self.make_request("GET", self.path)
|
channel = self.make_request("GET", self.path)
|
||||||
self.assertEquals(404, channel.code, msg=channel.result["body"])
|
self.assertEqual(404, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# valid put
|
# valid put
|
||||||
content = '{"topic":"Topic name"}'
|
content = '{"topic":"Topic name"}'
|
||||||
channel = self.make_request("PUT", self.path, content)
|
channel = self.make_request("PUT", self.path, content)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# valid get
|
# valid get
|
||||||
channel = self.make_request("GET", self.path)
|
channel = self.make_request("GET", self.path)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
self.assert_dict(json.loads(content), channel.json_body)
|
self.assert_dict(json.loads(content), channel.json_body)
|
||||||
|
|
||||||
def test_rooms_topic_with_extra_keys(self):
|
def test_rooms_topic_with_extra_keys(self):
|
||||||
# valid put with extra keys
|
# valid put with extra keys
|
||||||
content = '{"topic":"Seasons","subtopic":"Summer"}'
|
content = '{"topic":"Seasons","subtopic":"Summer"}'
|
||||||
channel = self.make_request("PUT", self.path, content)
|
channel = self.make_request("PUT", self.path, content)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# valid get
|
# valid get
|
||||||
channel = self.make_request("GET", self.path)
|
channel = self.make_request("GET", self.path)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
self.assert_dict(json.loads(content), channel.json_body)
|
self.assert_dict(json.loads(content), channel.json_body)
|
||||||
|
|
||||||
|
|
||||||
@ -775,22 +775,22 @@ class RoomMemberStateTestCase(RoomBase):
|
|||||||
path = "/rooms/%s/state/m.room.member/%s" % (self.room_id, self.user_id)
|
path = "/rooms/%s/state/m.room.member/%s" % (self.room_id, self.user_id)
|
||||||
# missing keys or invalid json
|
# missing keys or invalid json
|
||||||
channel = self.make_request("PUT", path, "{}")
|
channel = self.make_request("PUT", path, "{}")
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", path, '{"_name":"bo"}')
|
channel = self.make_request("PUT", path, '{"_name":"bo"}')
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", path, '{"nao')
|
channel = self.make_request("PUT", path, '{"nao')
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", path, b'[{"_name":"bo"},{"_name":"jill"}]')
|
channel = self.make_request("PUT", path, b'[{"_name":"bo"},{"_name":"jill"}]')
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", path, "text only")
|
channel = self.make_request("PUT", path, "text only")
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", path, "")
|
channel = self.make_request("PUT", path, "")
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# valid keys, wrong types
|
# valid keys, wrong types
|
||||||
content = '{"membership":["%s","%s","%s"]}' % (
|
content = '{"membership":["%s","%s","%s"]}' % (
|
||||||
@ -799,7 +799,7 @@ class RoomMemberStateTestCase(RoomBase):
|
|||||||
Membership.LEAVE,
|
Membership.LEAVE,
|
||||||
)
|
)
|
||||||
channel = self.make_request("PUT", path, content.encode("ascii"))
|
channel = self.make_request("PUT", path, content.encode("ascii"))
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
def test_rooms_members_self(self):
|
def test_rooms_members_self(self):
|
||||||
path = "/rooms/%s/state/m.room.member/%s" % (
|
path = "/rooms/%s/state/m.room.member/%s" % (
|
||||||
@ -810,13 +810,13 @@ class RoomMemberStateTestCase(RoomBase):
|
|||||||
# valid join message (NOOP since we made the room)
|
# valid join message (NOOP since we made the room)
|
||||||
content = '{"membership":"%s"}' % Membership.JOIN
|
content = '{"membership":"%s"}' % Membership.JOIN
|
||||||
channel = self.make_request("PUT", path, content.encode("ascii"))
|
channel = self.make_request("PUT", path, content.encode("ascii"))
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("GET", path, None)
|
channel = self.make_request("GET", path, None)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
expected_response = {"membership": Membership.JOIN}
|
expected_response = {"membership": Membership.JOIN}
|
||||||
self.assertEquals(expected_response, channel.json_body)
|
self.assertEqual(expected_response, channel.json_body)
|
||||||
|
|
||||||
def test_rooms_members_other(self):
|
def test_rooms_members_other(self):
|
||||||
self.other_id = "@zzsid1:red"
|
self.other_id = "@zzsid1:red"
|
||||||
@ -828,11 +828,11 @@ class RoomMemberStateTestCase(RoomBase):
|
|||||||
# valid invite message
|
# valid invite message
|
||||||
content = '{"membership":"%s"}' % Membership.INVITE
|
content = '{"membership":"%s"}' % Membership.INVITE
|
||||||
channel = self.make_request("PUT", path, content)
|
channel = self.make_request("PUT", path, content)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("GET", path, None)
|
channel = self.make_request("GET", path, None)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
self.assertEquals(json.loads(content), channel.json_body)
|
self.assertEqual(json.loads(content), channel.json_body)
|
||||||
|
|
||||||
def test_rooms_members_other_custom_keys(self):
|
def test_rooms_members_other_custom_keys(self):
|
||||||
self.other_id = "@zzsid1:red"
|
self.other_id = "@zzsid1:red"
|
||||||
@ -847,11 +847,11 @@ class RoomMemberStateTestCase(RoomBase):
|
|||||||
"Join us!",
|
"Join us!",
|
||||||
)
|
)
|
||||||
channel = self.make_request("PUT", path, content)
|
channel = self.make_request("PUT", path, content)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("GET", path, None)
|
channel = self.make_request("GET", path, None)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
self.assertEquals(json.loads(content), channel.json_body)
|
self.assertEqual(json.loads(content), channel.json_body)
|
||||||
|
|
||||||
|
|
||||||
class RoomInviteRatelimitTestCase(RoomBase):
|
class RoomInviteRatelimitTestCase(RoomBase):
|
||||||
@ -937,7 +937,7 @@ class RoomJoinTestCase(RoomBase):
|
|||||||
False,
|
False,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
callback_mock.call_args,
|
callback_mock.call_args,
|
||||||
expected_call_args,
|
expected_call_args,
|
||||||
callback_mock.call_args,
|
callback_mock.call_args,
|
||||||
@ -955,7 +955,7 @@ class RoomJoinTestCase(RoomBase):
|
|||||||
True,
|
True,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
callback_mock.call_args,
|
callback_mock.call_args,
|
||||||
expected_call_args,
|
expected_call_args,
|
||||||
callback_mock.call_args,
|
callback_mock.call_args,
|
||||||
@ -1013,7 +1013,7 @@ class RoomJoinRatelimitTestCase(RoomBase):
|
|||||||
# Update the display name for the user.
|
# Update the display name for the user.
|
||||||
path = "/_matrix/client/r0/profile/%s/displayname" % self.user_id
|
path = "/_matrix/client/r0/profile/%s/displayname" % self.user_id
|
||||||
channel = self.make_request("PUT", path, {"displayname": "John Doe"})
|
channel = self.make_request("PUT", path, {"displayname": "John Doe"})
|
||||||
self.assertEquals(channel.code, 200, channel.json_body)
|
self.assertEqual(channel.code, 200, channel.json_body)
|
||||||
|
|
||||||
# Check that all the rooms have been sent a profile update into.
|
# Check that all the rooms have been sent a profile update into.
|
||||||
for room_id in room_ids:
|
for room_id in room_ids:
|
||||||
@ -1023,10 +1023,10 @@ class RoomJoinRatelimitTestCase(RoomBase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
channel = self.make_request("GET", path)
|
channel = self.make_request("GET", path)
|
||||||
self.assertEquals(channel.code, 200)
|
self.assertEqual(channel.code, 200)
|
||||||
|
|
||||||
self.assertIn("displayname", channel.json_body)
|
self.assertIn("displayname", channel.json_body)
|
||||||
self.assertEquals(channel.json_body["displayname"], "John Doe")
|
self.assertEqual(channel.json_body["displayname"], "John Doe")
|
||||||
|
|
||||||
@unittest.override_config(
|
@unittest.override_config(
|
||||||
{"rc_joins": {"local": {"per_second": 0.5, "burst_count": 3}}}
|
{"rc_joins": {"local": {"per_second": 0.5, "burst_count": 3}}}
|
||||||
@ -1047,7 +1047,7 @@ class RoomJoinRatelimitTestCase(RoomBase):
|
|||||||
# if all of these requests ended up joining the user to a room.
|
# if all of these requests ended up joining the user to a room.
|
||||||
for _ in range(4):
|
for _ in range(4):
|
||||||
channel = self.make_request("POST", path % room_id, {})
|
channel = self.make_request("POST", path % room_id, {})
|
||||||
self.assertEquals(channel.code, 200)
|
self.assertEqual(channel.code, 200)
|
||||||
|
|
||||||
@unittest.override_config(
|
@unittest.override_config(
|
||||||
{
|
{
|
||||||
@ -1078,40 +1078,40 @@ class RoomMessagesTestCase(RoomBase):
|
|||||||
path = "/rooms/%s/send/m.room.message/mid1" % (urlparse.quote(self.room_id))
|
path = "/rooms/%s/send/m.room.message/mid1" % (urlparse.quote(self.room_id))
|
||||||
# missing keys or invalid json
|
# missing keys or invalid json
|
||||||
channel = self.make_request("PUT", path, b"{}")
|
channel = self.make_request("PUT", path, b"{}")
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", path, b'{"_name":"bo"}')
|
channel = self.make_request("PUT", path, b'{"_name":"bo"}')
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", path, b'{"nao')
|
channel = self.make_request("PUT", path, b'{"nao')
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", path, b'[{"_name":"bo"},{"_name":"jill"}]')
|
channel = self.make_request("PUT", path, b'[{"_name":"bo"},{"_name":"jill"}]')
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", path, b"text only")
|
channel = self.make_request("PUT", path, b"text only")
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
channel = self.make_request("PUT", path, b"")
|
channel = self.make_request("PUT", path, b"")
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
def test_rooms_messages_sent(self):
|
def test_rooms_messages_sent(self):
|
||||||
path = "/rooms/%s/send/m.room.message/mid1" % (urlparse.quote(self.room_id))
|
path = "/rooms/%s/send/m.room.message/mid1" % (urlparse.quote(self.room_id))
|
||||||
|
|
||||||
content = b'{"body":"test","msgtype":{"type":"a"}}'
|
content = b'{"body":"test","msgtype":{"type":"a"}}'
|
||||||
channel = self.make_request("PUT", path, content)
|
channel = self.make_request("PUT", path, content)
|
||||||
self.assertEquals(400, channel.code, msg=channel.result["body"])
|
self.assertEqual(400, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# custom message types
|
# custom message types
|
||||||
content = b'{"body":"test","msgtype":"test.custom.text"}'
|
content = b'{"body":"test","msgtype":"test.custom.text"}'
|
||||||
channel = self.make_request("PUT", path, content)
|
channel = self.make_request("PUT", path, content)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
# m.text message type
|
# m.text message type
|
||||||
path = "/rooms/%s/send/m.room.message/mid2" % (urlparse.quote(self.room_id))
|
path = "/rooms/%s/send/m.room.message/mid2" % (urlparse.quote(self.room_id))
|
||||||
content = b'{"body":"test2","msgtype":"m.text"}'
|
content = b'{"body":"test2","msgtype":"m.text"}'
|
||||||
channel = self.make_request("PUT", path, content)
|
channel = self.make_request("PUT", path, content)
|
||||||
self.assertEquals(200, channel.code, msg=channel.result["body"])
|
self.assertEqual(200, channel.code, msg=channel.result["body"])
|
||||||
|
|
||||||
|
|
||||||
class RoomInitialSyncTestCase(RoomBase):
|
class RoomInitialSyncTestCase(RoomBase):
|
||||||
@ -1125,10 +1125,10 @@ class RoomInitialSyncTestCase(RoomBase):
|
|||||||
|
|
||||||
def test_initial_sync(self):
|
def test_initial_sync(self):
|
||||||
channel = self.make_request("GET", "/rooms/%s/initialSync" % self.room_id)
|
channel = self.make_request("GET", "/rooms/%s/initialSync" % self.room_id)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
|
|
||||||
self.assertEquals(self.room_id, channel.json_body["room_id"])
|
self.assertEqual(self.room_id, channel.json_body["room_id"])
|
||||||
self.assertEquals("join", channel.json_body["membership"])
|
self.assertEqual("join", channel.json_body["membership"])
|
||||||
|
|
||||||
# Room state is easier to assert on if we unpack it into a dict
|
# Room state is easier to assert on if we unpack it into a dict
|
||||||
state = {}
|
state = {}
|
||||||
@ -1152,7 +1152,7 @@ class RoomInitialSyncTestCase(RoomBase):
|
|||||||
e["content"]["user_id"]: e for e in channel.json_body["presence"]
|
e["content"]["user_id"]: e for e in channel.json_body["presence"]
|
||||||
}
|
}
|
||||||
self.assertTrue(self.user_id in presence_by_user)
|
self.assertTrue(self.user_id in presence_by_user)
|
||||||
self.assertEquals("m.presence", presence_by_user[self.user_id]["type"])
|
self.assertEqual("m.presence", presence_by_user[self.user_id]["type"])
|
||||||
|
|
||||||
|
|
||||||
class RoomMessageListTestCase(RoomBase):
|
class RoomMessageListTestCase(RoomBase):
|
||||||
@ -1168,9 +1168,9 @@ class RoomMessageListTestCase(RoomBase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET", "/rooms/%s/messages?access_token=x&from=%s" % (self.room_id, token)
|
"GET", "/rooms/%s/messages?access_token=x&from=%s" % (self.room_id, token)
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
self.assertTrue("start" in channel.json_body)
|
self.assertTrue("start" in channel.json_body)
|
||||||
self.assertEquals(token, channel.json_body["start"])
|
self.assertEqual(token, channel.json_body["start"])
|
||||||
self.assertTrue("chunk" in channel.json_body)
|
self.assertTrue("chunk" in channel.json_body)
|
||||||
self.assertTrue("end" in channel.json_body)
|
self.assertTrue("end" in channel.json_body)
|
||||||
|
|
||||||
@ -1179,9 +1179,9 @@ class RoomMessageListTestCase(RoomBase):
|
|||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"GET", "/rooms/%s/messages?access_token=x&from=%s" % (self.room_id, token)
|
"GET", "/rooms/%s/messages?access_token=x&from=%s" % (self.room_id, token)
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
self.assertTrue("start" in channel.json_body)
|
self.assertTrue("start" in channel.json_body)
|
||||||
self.assertEquals(token, channel.json_body["start"])
|
self.assertEqual(token, channel.json_body["start"])
|
||||||
self.assertTrue("chunk" in channel.json_body)
|
self.assertTrue("chunk" in channel.json_body)
|
||||||
self.assertTrue("end" in channel.json_body)
|
self.assertTrue("end" in channel.json_body)
|
||||||
|
|
||||||
@ -2614,7 +2614,7 @@ class ThreepidInviteTestCase(unittest.HomeserverTestCase):
|
|||||||
},
|
},
|
||||||
access_token=self.tok,
|
access_token=self.tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.code, 200)
|
self.assertEqual(channel.code, 200)
|
||||||
|
|
||||||
# Check that the callback was called with the right params.
|
# Check that the callback was called with the right params.
|
||||||
mock.assert_called_with(self.user_id, "email", email_to_invite, self.room_id)
|
mock.assert_called_with(self.user_id, "email", email_to_invite, self.room_id)
|
||||||
@ -2636,7 +2636,7 @@ class ThreepidInviteTestCase(unittest.HomeserverTestCase):
|
|||||||
},
|
},
|
||||||
access_token=self.tok,
|
access_token=self.tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.code, 403)
|
self.assertEqual(channel.code, 403)
|
||||||
|
|
||||||
# Also check that it stopped before calling _make_and_store_3pid_invite.
|
# Also check that it stopped before calling _make_and_store_3pid_invite.
|
||||||
make_invite_mock.assert_called_once()
|
make_invite_mock.assert_called_once()
|
||||||
|
@ -96,7 +96,7 @@ class RoomTestCase(_ShadowBannedBase):
|
|||||||
{"id_server": "test", "medium": "email", "address": "test@test.test"},
|
{"id_server": "test", "medium": "email", "address": "test@test.test"},
|
||||||
access_token=self.banned_access_token,
|
access_token=self.banned_access_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
# This should have raised an error earlier, but double check this wasn't called.
|
# This should have raised an error earlier, but double check this wasn't called.
|
||||||
identity_handler.lookup_3pid.assert_not_called()
|
identity_handler.lookup_3pid.assert_not_called()
|
||||||
@ -110,7 +110,7 @@ class RoomTestCase(_ShadowBannedBase):
|
|||||||
{"visibility": "public", "invite": [self.other_user_id]},
|
{"visibility": "public", "invite": [self.other_user_id]},
|
||||||
access_token=self.banned_access_token,
|
access_token=self.banned_access_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
room_id = channel.json_body["room_id"]
|
room_id = channel.json_body["room_id"]
|
||||||
|
|
||||||
# But the user wasn't actually invited.
|
# But the user wasn't actually invited.
|
||||||
@ -165,7 +165,7 @@ class RoomTestCase(_ShadowBannedBase):
|
|||||||
{"new_version": "6"},
|
{"new_version": "6"},
|
||||||
access_token=self.banned_access_token,
|
access_token=self.banned_access_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
# A new room_id should be returned.
|
# A new room_id should be returned.
|
||||||
self.assertIn("replacement_room", channel.json_body)
|
self.assertIn("replacement_room", channel.json_body)
|
||||||
|
|
||||||
@ -190,11 +190,11 @@ class RoomTestCase(_ShadowBannedBase):
|
|||||||
{"typing": True, "timeout": 30000},
|
{"typing": True, "timeout": 30000},
|
||||||
access_token=self.banned_access_token,
|
access_token=self.banned_access_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
|
|
||||||
# There should be no typing events.
|
# There should be no typing events.
|
||||||
event_source = self.hs.get_event_sources().sources.typing
|
event_source = self.hs.get_event_sources().sources.typing
|
||||||
self.assertEquals(event_source.get_current_key(), 0)
|
self.assertEqual(event_source.get_current_key(), 0)
|
||||||
|
|
||||||
# The other user can join and send typing events.
|
# The other user can join and send typing events.
|
||||||
self.helper.join(room_id, self.other_user_id, tok=self.other_access_token)
|
self.helper.join(room_id, self.other_user_id, tok=self.other_access_token)
|
||||||
@ -205,10 +205,10 @@ class RoomTestCase(_ShadowBannedBase):
|
|||||||
{"typing": True, "timeout": 30000},
|
{"typing": True, "timeout": 30000},
|
||||||
access_token=self.other_access_token,
|
access_token=self.other_access_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
|
|
||||||
# These appear in the room.
|
# These appear in the room.
|
||||||
self.assertEquals(event_source.get_current_key(), 1)
|
self.assertEqual(event_source.get_current_key(), 1)
|
||||||
events = self.get_success(
|
events = self.get_success(
|
||||||
event_source.get_new_events(
|
event_source.get_new_events(
|
||||||
user=UserID.from_string(self.other_user_id),
|
user=UserID.from_string(self.other_user_id),
|
||||||
@ -218,7 +218,7 @@ class RoomTestCase(_ShadowBannedBase):
|
|||||||
is_guest=False,
|
is_guest=False,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
events[0],
|
events[0],
|
||||||
[
|
[
|
||||||
{
|
{
|
||||||
@ -257,7 +257,7 @@ class ProfileTestCase(_ShadowBannedBase):
|
|||||||
{"displayname": new_display_name},
|
{"displayname": new_display_name},
|
||||||
access_token=self.banned_access_token,
|
access_token=self.banned_access_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertEqual(channel.json_body, {})
|
self.assertEqual(channel.json_body, {})
|
||||||
|
|
||||||
# The user's display name should be updated.
|
# The user's display name should be updated.
|
||||||
@ -299,7 +299,7 @@ class ProfileTestCase(_ShadowBannedBase):
|
|||||||
{"membership": "join", "displayname": new_display_name},
|
{"membership": "join", "displayname": new_display_name},
|
||||||
access_token=self.banned_access_token,
|
access_token=self.banned_access_token,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertIn("event_id", channel.json_body)
|
self.assertIn("event_id", channel.json_body)
|
||||||
|
|
||||||
# The display name in the room should not be changed.
|
# The display name in the room should not be changed.
|
||||||
|
@ -91,9 +91,9 @@ class UserSharedRoomsTest(unittest.HomeserverTestCase):
|
|||||||
# Check shared rooms from user1's perspective.
|
# Check shared rooms from user1's perspective.
|
||||||
# We should see the one room in common
|
# We should see the one room in common
|
||||||
channel = self._get_shared_rooms(u1_token, u2)
|
channel = self._get_shared_rooms(u1_token, u2)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertEquals(len(channel.json_body["joined"]), 1)
|
self.assertEqual(len(channel.json_body["joined"]), 1)
|
||||||
self.assertEquals(channel.json_body["joined"][0], room_id_one)
|
self.assertEqual(channel.json_body["joined"][0], room_id_one)
|
||||||
|
|
||||||
# Create another room and invite user2 to it
|
# Create another room and invite user2 to it
|
||||||
room_id_two = self.helper.create_room_as(
|
room_id_two = self.helper.create_room_as(
|
||||||
@ -104,8 +104,8 @@ class UserSharedRoomsTest(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# Check shared rooms again. We should now see both rooms.
|
# Check shared rooms again. We should now see both rooms.
|
||||||
channel = self._get_shared_rooms(u1_token, u2)
|
channel = self._get_shared_rooms(u1_token, u2)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertEquals(len(channel.json_body["joined"]), 2)
|
self.assertEqual(len(channel.json_body["joined"]), 2)
|
||||||
for room_id_id in channel.json_body["joined"]:
|
for room_id_id in channel.json_body["joined"]:
|
||||||
self.assertIn(room_id_id, [room_id_one, room_id_two])
|
self.assertIn(room_id_id, [room_id_one, room_id_two])
|
||||||
|
|
||||||
@ -125,18 +125,18 @@ class UserSharedRoomsTest(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# Assert user directory is not empty
|
# Assert user directory is not empty
|
||||||
channel = self._get_shared_rooms(u1_token, u2)
|
channel = self._get_shared_rooms(u1_token, u2)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertEquals(len(channel.json_body["joined"]), 1)
|
self.assertEqual(len(channel.json_body["joined"]), 1)
|
||||||
self.assertEquals(channel.json_body["joined"][0], room)
|
self.assertEqual(channel.json_body["joined"][0], room)
|
||||||
|
|
||||||
self.helper.leave(room, user=u1, tok=u1_token)
|
self.helper.leave(room, user=u1, tok=u1_token)
|
||||||
|
|
||||||
# Check user1's view of shared rooms with user2
|
# Check user1's view of shared rooms with user2
|
||||||
channel = self._get_shared_rooms(u1_token, u2)
|
channel = self._get_shared_rooms(u1_token, u2)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertEquals(len(channel.json_body["joined"]), 0)
|
self.assertEqual(len(channel.json_body["joined"]), 0)
|
||||||
|
|
||||||
# Check user2's view of shared rooms with user1
|
# Check user2's view of shared rooms with user1
|
||||||
channel = self._get_shared_rooms(u2_token, u1)
|
channel = self._get_shared_rooms(u2_token, u1)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertEquals(len(channel.json_body["joined"]), 0)
|
self.assertEqual(len(channel.json_body["joined"]), 0)
|
||||||
|
@ -237,10 +237,10 @@ class SyncTypingTests(unittest.HomeserverTestCase):
|
|||||||
typing_url % (room, other_user_id, other_access_token),
|
typing_url % (room, other_user_id, other_access_token),
|
||||||
b'{"typing": true, "timeout": 30000}',
|
b'{"typing": true, "timeout": 30000}',
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
|
|
||||||
channel = self.make_request("GET", "/sync?access_token=%s" % (access_token,))
|
channel = self.make_request("GET", "/sync?access_token=%s" % (access_token,))
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
next_batch = channel.json_body["next_batch"]
|
next_batch = channel.json_body["next_batch"]
|
||||||
|
|
||||||
# Stop typing.
|
# Stop typing.
|
||||||
@ -249,7 +249,7 @@ class SyncTypingTests(unittest.HomeserverTestCase):
|
|||||||
typing_url % (room, other_user_id, other_access_token),
|
typing_url % (room, other_user_id, other_access_token),
|
||||||
b'{"typing": false}',
|
b'{"typing": false}',
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
|
|
||||||
# Start typing.
|
# Start typing.
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -257,11 +257,11 @@ class SyncTypingTests(unittest.HomeserverTestCase):
|
|||||||
typing_url % (room, other_user_id, other_access_token),
|
typing_url % (room, other_user_id, other_access_token),
|
||||||
b'{"typing": true, "timeout": 30000}',
|
b'{"typing": true, "timeout": 30000}',
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
|
|
||||||
# Should return immediately
|
# Should return immediately
|
||||||
channel = self.make_request("GET", sync_url % (access_token, next_batch))
|
channel = self.make_request("GET", sync_url % (access_token, next_batch))
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
next_batch = channel.json_body["next_batch"]
|
next_batch = channel.json_body["next_batch"]
|
||||||
|
|
||||||
# Reset typing serial back to 0, as if the master had.
|
# Reset typing serial back to 0, as if the master had.
|
||||||
@ -273,7 +273,7 @@ class SyncTypingTests(unittest.HomeserverTestCase):
|
|||||||
self.helper.send(room, body="There!", tok=other_access_token)
|
self.helper.send(room, body="There!", tok=other_access_token)
|
||||||
|
|
||||||
channel = self.make_request("GET", sync_url % (access_token, next_batch))
|
channel = self.make_request("GET", sync_url % (access_token, next_batch))
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
next_batch = channel.json_body["next_batch"]
|
next_batch = channel.json_body["next_batch"]
|
||||||
|
|
||||||
# This should time out! But it does not, because our stream token is
|
# This should time out! But it does not, because our stream token is
|
||||||
@ -281,7 +281,7 @@ class SyncTypingTests(unittest.HomeserverTestCase):
|
|||||||
# already seen) is new, since it's got a token above our new, now-reset
|
# already seen) is new, since it's got a token above our new, now-reset
|
||||||
# stream token.
|
# stream token.
|
||||||
channel = self.make_request("GET", sync_url % (access_token, next_batch))
|
channel = self.make_request("GET", sync_url % (access_token, next_batch))
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
next_batch = channel.json_body["next_batch"]
|
next_batch = channel.json_body["next_batch"]
|
||||||
|
|
||||||
# Clear the typing information, so that it doesn't think everything is
|
# Clear the typing information, so that it doesn't think everything is
|
||||||
@ -351,7 +351,7 @@ class SyncKnockTestCase(
|
|||||||
b"{}",
|
b"{}",
|
||||||
self.knocker_tok,
|
self.knocker_tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
# We expect to see the knock event in the stripped room state later
|
# We expect to see the knock event in the stripped room state later
|
||||||
self.expected_room_state[EventTypes.Member] = {
|
self.expected_room_state[EventTypes.Member] = {
|
||||||
|
@ -139,7 +139,7 @@ class ThirdPartyRulesTestCase(unittest.FederatingHomeserverTestCase):
|
|||||||
{},
|
{},
|
||||||
access_token=self.tok,
|
access_token=self.tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
callback.assert_called_once()
|
callback.assert_called_once()
|
||||||
|
|
||||||
@ -157,7 +157,7 @@ class ThirdPartyRulesTestCase(unittest.FederatingHomeserverTestCase):
|
|||||||
{},
|
{},
|
||||||
access_token=self.tok,
|
access_token=self.tok,
|
||||||
)
|
)
|
||||||
self.assertEquals(channel.result["code"], b"403", channel.result)
|
self.assertEqual(channel.result["code"], b"403", channel.result)
|
||||||
|
|
||||||
def test_third_party_rules_workaround_synapse_errors_pass_through(self):
|
def test_third_party_rules_workaround_synapse_errors_pass_through(self):
|
||||||
"""
|
"""
|
||||||
@ -193,7 +193,7 @@ class ThirdPartyRulesTestCase(unittest.FederatingHomeserverTestCase):
|
|||||||
access_token=self.tok,
|
access_token=self.tok,
|
||||||
)
|
)
|
||||||
# Check the error code
|
# Check the error code
|
||||||
self.assertEquals(channel.result["code"], b"429", channel.result)
|
self.assertEqual(channel.result["code"], b"429", channel.result)
|
||||||
# Check the JSON body has had the `nasty` key injected
|
# Check the JSON body has had the `nasty` key injected
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
channel.json_body,
|
channel.json_body,
|
||||||
@ -329,10 +329,10 @@ class ThirdPartyRulesTestCase(unittest.FederatingHomeserverTestCase):
|
|||||||
self.hs.get_module_api().create_and_send_event_into_room(event_dict)
|
self.hs.get_module_api().create_and_send_event_into_room(event_dict)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(event.sender, self.user_id)
|
self.assertEqual(event.sender, self.user_id)
|
||||||
self.assertEquals(event.room_id, self.room_id)
|
self.assertEqual(event.room_id, self.room_id)
|
||||||
self.assertEquals(event.type, "m.room.message")
|
self.assertEqual(event.type, "m.room.message")
|
||||||
self.assertEquals(event.content, content)
|
self.assertEqual(event.content, content)
|
||||||
|
|
||||||
@unittest.override_config(
|
@unittest.override_config(
|
||||||
{
|
{
|
||||||
|
@ -72,9 +72,9 @@ class RoomTypingTestCase(unittest.HomeserverTestCase):
|
|||||||
"/rooms/%s/typing/%s" % (self.room_id, self.user_id),
|
"/rooms/%s/typing/%s" % (self.room_id, self.user_id),
|
||||||
b'{"typing": true, "timeout": 30000}',
|
b'{"typing": true, "timeout": 30000}',
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 1)
|
self.assertEqual(self.event_source.get_current_key(), 1)
|
||||||
events = self.get_success(
|
events = self.get_success(
|
||||||
self.event_source.get_new_events(
|
self.event_source.get_new_events(
|
||||||
user=UserID.from_string(self.user_id),
|
user=UserID.from_string(self.user_id),
|
||||||
@ -84,7 +84,7 @@ class RoomTypingTestCase(unittest.HomeserverTestCase):
|
|||||||
is_guest=False,
|
is_guest=False,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
events[0],
|
events[0],
|
||||||
[
|
[
|
||||||
{
|
{
|
||||||
@ -101,7 +101,7 @@ class RoomTypingTestCase(unittest.HomeserverTestCase):
|
|||||||
"/rooms/%s/typing/%s" % (self.room_id, self.user_id),
|
"/rooms/%s/typing/%s" % (self.room_id, self.user_id),
|
||||||
b'{"typing": false}',
|
b'{"typing": false}',
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
|
|
||||||
def test_typing_timeout(self):
|
def test_typing_timeout(self):
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
@ -109,19 +109,19 @@ class RoomTypingTestCase(unittest.HomeserverTestCase):
|
|||||||
"/rooms/%s/typing/%s" % (self.room_id, self.user_id),
|
"/rooms/%s/typing/%s" % (self.room_id, self.user_id),
|
||||||
b'{"typing": true, "timeout": 30000}',
|
b'{"typing": true, "timeout": 30000}',
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 1)
|
self.assertEqual(self.event_source.get_current_key(), 1)
|
||||||
|
|
||||||
self.reactor.advance(36)
|
self.reactor.advance(36)
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 2)
|
self.assertEqual(self.event_source.get_current_key(), 2)
|
||||||
|
|
||||||
channel = self.make_request(
|
channel = self.make_request(
|
||||||
"PUT",
|
"PUT",
|
||||||
"/rooms/%s/typing/%s" % (self.room_id, self.user_id),
|
"/rooms/%s/typing/%s" % (self.room_id, self.user_id),
|
||||||
b'{"typing": true, "timeout": 30000}',
|
b'{"typing": true, "timeout": 30000}',
|
||||||
)
|
)
|
||||||
self.assertEquals(200, channel.code)
|
self.assertEqual(200, channel.code)
|
||||||
|
|
||||||
self.assertEquals(self.event_source.get_current_key(), 3)
|
self.assertEqual(self.event_source.get_current_key(), 3)
|
||||||
|
@ -65,7 +65,7 @@ class UpgradeRoomTest(unittest.HomeserverTestCase):
|
|||||||
Upgrading a room should work fine.
|
Upgrading a room should work fine.
|
||||||
"""
|
"""
|
||||||
channel = self._upgrade_room()
|
channel = self._upgrade_room()
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertIn("replacement_room", channel.json_body)
|
self.assertIn("replacement_room", channel.json_body)
|
||||||
|
|
||||||
def test_not_in_room(self):
|
def test_not_in_room(self):
|
||||||
@ -77,7 +77,7 @@ class UpgradeRoomTest(unittest.HomeserverTestCase):
|
|||||||
roomless_token = self.login(roomless, "pass")
|
roomless_token = self.login(roomless, "pass")
|
||||||
|
|
||||||
channel = self._upgrade_room(roomless_token)
|
channel = self._upgrade_room(roomless_token)
|
||||||
self.assertEquals(403, channel.code, channel.result)
|
self.assertEqual(403, channel.code, channel.result)
|
||||||
|
|
||||||
def test_power_levels(self):
|
def test_power_levels(self):
|
||||||
"""
|
"""
|
||||||
@ -85,7 +85,7 @@ class UpgradeRoomTest(unittest.HomeserverTestCase):
|
|||||||
"""
|
"""
|
||||||
# The other user doesn't have the proper power level.
|
# The other user doesn't have the proper power level.
|
||||||
channel = self._upgrade_room(self.other_token)
|
channel = self._upgrade_room(self.other_token)
|
||||||
self.assertEquals(403, channel.code, channel.result)
|
self.assertEqual(403, channel.code, channel.result)
|
||||||
|
|
||||||
# Increase the power levels so that this user can upgrade.
|
# Increase the power levels so that this user can upgrade.
|
||||||
power_levels = self.helper.get_state(
|
power_levels = self.helper.get_state(
|
||||||
@ -103,7 +103,7 @@ class UpgradeRoomTest(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# The upgrade should succeed!
|
# The upgrade should succeed!
|
||||||
channel = self._upgrade_room(self.other_token)
|
channel = self._upgrade_room(self.other_token)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
def test_power_levels_user_default(self):
|
def test_power_levels_user_default(self):
|
||||||
"""
|
"""
|
||||||
@ -111,7 +111,7 @@ class UpgradeRoomTest(unittest.HomeserverTestCase):
|
|||||||
"""
|
"""
|
||||||
# The other user doesn't have the proper power level.
|
# The other user doesn't have the proper power level.
|
||||||
channel = self._upgrade_room(self.other_token)
|
channel = self._upgrade_room(self.other_token)
|
||||||
self.assertEquals(403, channel.code, channel.result)
|
self.assertEqual(403, channel.code, channel.result)
|
||||||
|
|
||||||
# Increase the power levels so that this user can upgrade.
|
# Increase the power levels so that this user can upgrade.
|
||||||
power_levels = self.helper.get_state(
|
power_levels = self.helper.get_state(
|
||||||
@ -129,7 +129,7 @@ class UpgradeRoomTest(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# The upgrade should succeed!
|
# The upgrade should succeed!
|
||||||
channel = self._upgrade_room(self.other_token)
|
channel = self._upgrade_room(self.other_token)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
def test_power_levels_tombstone(self):
|
def test_power_levels_tombstone(self):
|
||||||
"""
|
"""
|
||||||
@ -137,7 +137,7 @@ class UpgradeRoomTest(unittest.HomeserverTestCase):
|
|||||||
"""
|
"""
|
||||||
# The other user doesn't have the proper power level.
|
# The other user doesn't have the proper power level.
|
||||||
channel = self._upgrade_room(self.other_token)
|
channel = self._upgrade_room(self.other_token)
|
||||||
self.assertEquals(403, channel.code, channel.result)
|
self.assertEqual(403, channel.code, channel.result)
|
||||||
|
|
||||||
# Increase the power levels so that this user can upgrade.
|
# Increase the power levels so that this user can upgrade.
|
||||||
power_levels = self.helper.get_state(
|
power_levels = self.helper.get_state(
|
||||||
@ -155,7 +155,7 @@ class UpgradeRoomTest(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# The upgrade should succeed!
|
# The upgrade should succeed!
|
||||||
channel = self._upgrade_room(self.other_token)
|
channel = self._upgrade_room(self.other_token)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
|
|
||||||
power_levels = self.helper.get_state(
|
power_levels = self.helper.get_state(
|
||||||
self.room_id,
|
self.room_id,
|
||||||
@ -197,7 +197,7 @@ class UpgradeRoomTest(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# Upgrade the room!
|
# Upgrade the room!
|
||||||
channel = self._upgrade_room(room_id=space_id)
|
channel = self._upgrade_room(room_id=space_id)
|
||||||
self.assertEquals(200, channel.code, channel.result)
|
self.assertEqual(200, channel.code, channel.result)
|
||||||
self.assertIn("replacement_room", channel.json_body)
|
self.assertIn("replacement_room", channel.json_body)
|
||||||
|
|
||||||
new_space_id = channel.json_body["replacement_room"]
|
new_space_id = channel.json_body["replacement_room"]
|
||||||
|
@ -94,7 +94,7 @@ class MediaStorageTests(unittest.HomeserverTestCase):
|
|||||||
self.assertTrue(os.path.exists(local_path))
|
self.assertTrue(os.path.exists(local_path))
|
||||||
|
|
||||||
# Asserts the file is under the expected local cache directory
|
# Asserts the file is under the expected local cache directory
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
os.path.commonprefix([self.primary_base_path, local_path]),
|
os.path.commonprefix([self.primary_base_path, local_path]),
|
||||||
self.primary_base_path,
|
self.primary_base_path,
|
||||||
)
|
)
|
||||||
|
@ -88,18 +88,18 @@ class HaveSeenEventsTestCase(unittest.HomeserverTestCase):
|
|||||||
res = self.get_success(
|
res = self.get_success(
|
||||||
self.store.have_seen_events("room1", ["event10", "event19"])
|
self.store.have_seen_events("room1", ["event10", "event19"])
|
||||||
)
|
)
|
||||||
self.assertEquals(res, {"event10"})
|
self.assertEqual(res, {"event10"})
|
||||||
|
|
||||||
# that should result in a single db query
|
# that should result in a single db query
|
||||||
self.assertEquals(ctx.get_resource_usage().db_txn_count, 1)
|
self.assertEqual(ctx.get_resource_usage().db_txn_count, 1)
|
||||||
|
|
||||||
# a second lookup of the same events should cause no queries
|
# a second lookup of the same events should cause no queries
|
||||||
with LoggingContext(name="test") as ctx:
|
with LoggingContext(name="test") as ctx:
|
||||||
res = self.get_success(
|
res = self.get_success(
|
||||||
self.store.have_seen_events("room1", ["event10", "event19"])
|
self.store.have_seen_events("room1", ["event10", "event19"])
|
||||||
)
|
)
|
||||||
self.assertEquals(res, {"event10"})
|
self.assertEqual(res, {"event10"})
|
||||||
self.assertEquals(ctx.get_resource_usage().db_txn_count, 0)
|
self.assertEqual(ctx.get_resource_usage().db_txn_count, 0)
|
||||||
|
|
||||||
def test_query_via_event_cache(self):
|
def test_query_via_event_cache(self):
|
||||||
# fetch an event into the event cache
|
# fetch an event into the event cache
|
||||||
@ -108,8 +108,8 @@ class HaveSeenEventsTestCase(unittest.HomeserverTestCase):
|
|||||||
# looking it up should now cause no db hits
|
# looking it up should now cause no db hits
|
||||||
with LoggingContext(name="test") as ctx:
|
with LoggingContext(name="test") as ctx:
|
||||||
res = self.get_success(self.store.have_seen_events("room1", ["event10"]))
|
res = self.get_success(self.store.have_seen_events("room1", ["event10"]))
|
||||||
self.assertEquals(res, {"event10"})
|
self.assertEqual(res, {"event10"})
|
||||||
self.assertEquals(ctx.get_resource_usage().db_txn_count, 0)
|
self.assertEqual(ctx.get_resource_usage().db_txn_count, 0)
|
||||||
|
|
||||||
|
|
||||||
class EventCacheTestCase(unittest.HomeserverTestCase):
|
class EventCacheTestCase(unittest.HomeserverTestCase):
|
||||||
|
@ -88,21 +88,21 @@ class ApplicationServiceStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
def test_retrieve_unknown_service_token(self) -> None:
|
def test_retrieve_unknown_service_token(self) -> None:
|
||||||
service = self.store.get_app_service_by_token("invalid_token")
|
service = self.store.get_app_service_by_token("invalid_token")
|
||||||
self.assertEquals(service, None)
|
self.assertEqual(service, None)
|
||||||
|
|
||||||
def test_retrieval_of_service(self) -> None:
|
def test_retrieval_of_service(self) -> None:
|
||||||
stored_service = self.store.get_app_service_by_token(self.as_token)
|
stored_service = self.store.get_app_service_by_token(self.as_token)
|
||||||
assert stored_service is not None
|
assert stored_service is not None
|
||||||
self.assertEquals(stored_service.token, self.as_token)
|
self.assertEqual(stored_service.token, self.as_token)
|
||||||
self.assertEquals(stored_service.id, self.as_id)
|
self.assertEqual(stored_service.id, self.as_id)
|
||||||
self.assertEquals(stored_service.url, self.as_url)
|
self.assertEqual(stored_service.url, self.as_url)
|
||||||
self.assertEquals(stored_service.namespaces[ApplicationService.NS_ALIASES], [])
|
self.assertEqual(stored_service.namespaces[ApplicationService.NS_ALIASES], [])
|
||||||
self.assertEquals(stored_service.namespaces[ApplicationService.NS_ROOMS], [])
|
self.assertEqual(stored_service.namespaces[ApplicationService.NS_ROOMS], [])
|
||||||
self.assertEquals(stored_service.namespaces[ApplicationService.NS_USERS], [])
|
self.assertEqual(stored_service.namespaces[ApplicationService.NS_USERS], [])
|
||||||
|
|
||||||
def test_retrieval_of_all_services(self) -> None:
|
def test_retrieval_of_all_services(self) -> None:
|
||||||
services = self.store.get_app_services()
|
services = self.store.get_app_services()
|
||||||
self.assertEquals(len(services), 3)
|
self.assertEqual(len(services), 3)
|
||||||
|
|
||||||
|
|
||||||
class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
||||||
@ -182,7 +182,7 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
) -> None:
|
) -> None:
|
||||||
service = Mock(id="999")
|
service = Mock(id="999")
|
||||||
state = self.get_success(self.store.get_appservice_state(service))
|
state = self.get_success(self.store.get_appservice_state(service))
|
||||||
self.assertEquals(None, state)
|
self.assertEqual(None, state)
|
||||||
|
|
||||||
def test_get_appservice_state_up(
|
def test_get_appservice_state_up(
|
||||||
self,
|
self,
|
||||||
@ -194,7 +194,7 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
state = self.get_success(
|
state = self.get_success(
|
||||||
defer.ensureDeferred(self.store.get_appservice_state(service))
|
defer.ensureDeferred(self.store.get_appservice_state(service))
|
||||||
)
|
)
|
||||||
self.assertEquals(ApplicationServiceState.UP, state)
|
self.assertEqual(ApplicationServiceState.UP, state)
|
||||||
|
|
||||||
def test_get_appservice_state_down(
|
def test_get_appservice_state_down(
|
||||||
self,
|
self,
|
||||||
@ -210,7 +210,7 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
service = Mock(id=self.as_list[1]["id"])
|
service = Mock(id=self.as_list[1]["id"])
|
||||||
state = self.get_success(self.store.get_appservice_state(service))
|
state = self.get_success(self.store.get_appservice_state(service))
|
||||||
self.assertEquals(ApplicationServiceState.DOWN, state)
|
self.assertEqual(ApplicationServiceState.DOWN, state)
|
||||||
|
|
||||||
def test_get_appservices_by_state_none(
|
def test_get_appservices_by_state_none(
|
||||||
self,
|
self,
|
||||||
@ -218,7 +218,7 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
services = self.get_success(
|
services = self.get_success(
|
||||||
self.store.get_appservices_by_state(ApplicationServiceState.DOWN)
|
self.store.get_appservices_by_state(ApplicationServiceState.DOWN)
|
||||||
)
|
)
|
||||||
self.assertEquals(0, len(services))
|
self.assertEqual(0, len(services))
|
||||||
|
|
||||||
def test_set_appservices_state_down(
|
def test_set_appservices_state_down(
|
||||||
self,
|
self,
|
||||||
@ -235,7 +235,7 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
(ApplicationServiceState.DOWN.value,),
|
(ApplicationServiceState.DOWN.value,),
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(service.id, rows[0][0])
|
self.assertEqual(service.id, rows[0][0])
|
||||||
|
|
||||||
def test_set_appservices_state_multiple_up(
|
def test_set_appservices_state_multiple_up(
|
||||||
self,
|
self,
|
||||||
@ -258,7 +258,7 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
(ApplicationServiceState.UP.value,),
|
(ApplicationServiceState.UP.value,),
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(service.id, rows[0][0])
|
self.assertEqual(service.id, rows[0][0])
|
||||||
|
|
||||||
def test_create_appservice_txn_first(
|
def test_create_appservice_txn_first(
|
||||||
self,
|
self,
|
||||||
@ -270,9 +270,9 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
self.store.create_appservice_txn(service, events, [], [], {}, {})
|
self.store.create_appservice_txn(service, events, [], [], {}, {})
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(txn.id, 1)
|
self.assertEqual(txn.id, 1)
|
||||||
self.assertEquals(txn.events, events)
|
self.assertEqual(txn.events, events)
|
||||||
self.assertEquals(txn.service, service)
|
self.assertEqual(txn.service, service)
|
||||||
|
|
||||||
def test_create_appservice_txn_older_last_txn(
|
def test_create_appservice_txn_older_last_txn(
|
||||||
self,
|
self,
|
||||||
@ -285,9 +285,9 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
txn = self.get_success(
|
txn = self.get_success(
|
||||||
self.store.create_appservice_txn(service, events, [], [], {}, {})
|
self.store.create_appservice_txn(service, events, [], [], {}, {})
|
||||||
)
|
)
|
||||||
self.assertEquals(txn.id, 9646)
|
self.assertEqual(txn.id, 9646)
|
||||||
self.assertEquals(txn.events, events)
|
self.assertEqual(txn.events, events)
|
||||||
self.assertEquals(txn.service, service)
|
self.assertEqual(txn.service, service)
|
||||||
|
|
||||||
def test_create_appservice_txn_up_to_date_last_txn(
|
def test_create_appservice_txn_up_to_date_last_txn(
|
||||||
self,
|
self,
|
||||||
@ -298,9 +298,9 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
txn = self.get_success(
|
txn = self.get_success(
|
||||||
self.store.create_appservice_txn(service, events, [], [], {}, {})
|
self.store.create_appservice_txn(service, events, [], [], {}, {})
|
||||||
)
|
)
|
||||||
self.assertEquals(txn.id, 9644)
|
self.assertEqual(txn.id, 9644)
|
||||||
self.assertEquals(txn.events, events)
|
self.assertEqual(txn.events, events)
|
||||||
self.assertEquals(txn.service, service)
|
self.assertEqual(txn.service, service)
|
||||||
|
|
||||||
def test_create_appservice_txn_up_fuzzing(
|
def test_create_appservice_txn_up_fuzzing(
|
||||||
self,
|
self,
|
||||||
@ -322,9 +322,9 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
txn = self.get_success(
|
txn = self.get_success(
|
||||||
self.store.create_appservice_txn(service, events, [], [], {}, {})
|
self.store.create_appservice_txn(service, events, [], [], {}, {})
|
||||||
)
|
)
|
||||||
self.assertEquals(txn.id, 9644)
|
self.assertEqual(txn.id, 9644)
|
||||||
self.assertEquals(txn.events, events)
|
self.assertEqual(txn.events, events)
|
||||||
self.assertEquals(txn.service, service)
|
self.assertEqual(txn.service, service)
|
||||||
|
|
||||||
def test_complete_appservice_txn_first_txn(
|
def test_complete_appservice_txn_first_txn(
|
||||||
self,
|
self,
|
||||||
@ -346,8 +346,8 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
(service.id,),
|
(service.id,),
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(1, len(res))
|
self.assertEqual(1, len(res))
|
||||||
self.assertEquals(txn_id, res[0][0])
|
self.assertEqual(txn_id, res[0][0])
|
||||||
|
|
||||||
res = self.get_success(
|
res = self.get_success(
|
||||||
self.db_pool.runQuery(
|
self.db_pool.runQuery(
|
||||||
@ -357,7 +357,7 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
(txn_id,),
|
(txn_id,),
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(0, len(res))
|
self.assertEqual(0, len(res))
|
||||||
|
|
||||||
def test_complete_appservice_txn_existing_in_state_table(
|
def test_complete_appservice_txn_existing_in_state_table(
|
||||||
self,
|
self,
|
||||||
@ -379,9 +379,9 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
(service.id,),
|
(service.id,),
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(1, len(res))
|
self.assertEqual(1, len(res))
|
||||||
self.assertEquals(txn_id, res[0][0])
|
self.assertEqual(txn_id, res[0][0])
|
||||||
self.assertEquals(ApplicationServiceState.UP.value, res[0][1])
|
self.assertEqual(ApplicationServiceState.UP.value, res[0][1])
|
||||||
|
|
||||||
res = self.get_success(
|
res = self.get_success(
|
||||||
self.db_pool.runQuery(
|
self.db_pool.runQuery(
|
||||||
@ -391,7 +391,7 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
(txn_id,),
|
(txn_id,),
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(0, len(res))
|
self.assertEqual(0, len(res))
|
||||||
|
|
||||||
def test_get_oldest_unsent_txn_none(
|
def test_get_oldest_unsent_txn_none(
|
||||||
self,
|
self,
|
||||||
@ -399,7 +399,7 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
service = Mock(id=self.as_list[0]["id"])
|
service = Mock(id=self.as_list[0]["id"])
|
||||||
|
|
||||||
txn = self.get_success(self.store.get_oldest_unsent_txn(service))
|
txn = self.get_success(self.store.get_oldest_unsent_txn(service))
|
||||||
self.assertEquals(None, txn)
|
self.assertEqual(None, txn)
|
||||||
|
|
||||||
def test_get_oldest_unsent_txn(self) -> None:
|
def test_get_oldest_unsent_txn(self) -> None:
|
||||||
service = Mock(id=self.as_list[0]["id"])
|
service = Mock(id=self.as_list[0]["id"])
|
||||||
@ -416,9 +416,9 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
self.get_success(self._insert_txn(service.id, 12, other_events))
|
self.get_success(self._insert_txn(service.id, 12, other_events))
|
||||||
|
|
||||||
txn = self.get_success(self.store.get_oldest_unsent_txn(service))
|
txn = self.get_success(self.store.get_oldest_unsent_txn(service))
|
||||||
self.assertEquals(service, txn.service)
|
self.assertEqual(service, txn.service)
|
||||||
self.assertEquals(10, txn.id)
|
self.assertEqual(10, txn.id)
|
||||||
self.assertEquals(events, txn.events)
|
self.assertEqual(events, txn.events)
|
||||||
|
|
||||||
def test_get_appservices_by_state_single(
|
def test_get_appservices_by_state_single(
|
||||||
self,
|
self,
|
||||||
@ -433,8 +433,8 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
services = self.get_success(
|
services = self.get_success(
|
||||||
self.store.get_appservices_by_state(ApplicationServiceState.DOWN)
|
self.store.get_appservices_by_state(ApplicationServiceState.DOWN)
|
||||||
)
|
)
|
||||||
self.assertEquals(1, len(services))
|
self.assertEqual(1, len(services))
|
||||||
self.assertEquals(self.as_list[0]["id"], services[0].id)
|
self.assertEqual(self.as_list[0]["id"], services[0].id)
|
||||||
|
|
||||||
def test_get_appservices_by_state_multiple(
|
def test_get_appservices_by_state_multiple(
|
||||||
self,
|
self,
|
||||||
@ -455,8 +455,8 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
services = self.get_success(
|
services = self.get_success(
|
||||||
self.store.get_appservices_by_state(ApplicationServiceState.DOWN)
|
self.store.get_appservices_by_state(ApplicationServiceState.DOWN)
|
||||||
)
|
)
|
||||||
self.assertEquals(2, len(services))
|
self.assertEqual(2, len(services))
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
{self.as_list[2]["id"], self.as_list[0]["id"]},
|
{self.as_list[2]["id"], self.as_list[0]["id"]},
|
||||||
{services[0].id, services[1].id},
|
{services[0].id, services[1].id},
|
||||||
)
|
)
|
||||||
@ -476,12 +476,12 @@ class ApplicationServiceStoreTypeStreamIds(unittest.HomeserverTestCase):
|
|||||||
value = self.get_success(
|
value = self.get_success(
|
||||||
self.store.get_type_stream_id_for_appservice(self.service, "read_receipt")
|
self.store.get_type_stream_id_for_appservice(self.service, "read_receipt")
|
||||||
)
|
)
|
||||||
self.assertEquals(value, 0)
|
self.assertEqual(value, 0)
|
||||||
|
|
||||||
value = self.get_success(
|
value = self.get_success(
|
||||||
self.store.get_type_stream_id_for_appservice(self.service, "presence")
|
self.store.get_type_stream_id_for_appservice(self.service, "presence")
|
||||||
)
|
)
|
||||||
self.assertEquals(value, 0)
|
self.assertEqual(value, 0)
|
||||||
|
|
||||||
def test_get_type_stream_id_for_appservice_invalid_type(self) -> None:
|
def test_get_type_stream_id_for_appservice_invalid_type(self) -> None:
|
||||||
self.get_failure(
|
self.get_failure(
|
||||||
|
@ -103,7 +103,7 @@ class SQLBaseStoreTestCase(unittest.TestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals("Value", value)
|
self.assertEqual("Value", value)
|
||||||
self.mock_txn.execute.assert_called_with(
|
self.mock_txn.execute.assert_called_with(
|
||||||
"SELECT retcol FROM tablename WHERE keycol = ?", ["TheKey"]
|
"SELECT retcol FROM tablename WHERE keycol = ?", ["TheKey"]
|
||||||
)
|
)
|
||||||
@ -121,7 +121,7 @@ class SQLBaseStoreTestCase(unittest.TestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals({"colA": 1, "colB": 2, "colC": 3}, ret)
|
self.assertEqual({"colA": 1, "colB": 2, "colC": 3}, ret)
|
||||||
self.mock_txn.execute.assert_called_with(
|
self.mock_txn.execute.assert_called_with(
|
||||||
"SELECT colA, colB, colC FROM tablename WHERE keycol = ?", ["TheKey"]
|
"SELECT colA, colB, colC FROM tablename WHERE keycol = ?", ["TheKey"]
|
||||||
)
|
)
|
||||||
@ -154,7 +154,7 @@ class SQLBaseStoreTestCase(unittest.TestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals([{"colA": 1}, {"colA": 2}, {"colA": 3}], ret)
|
self.assertEqual([{"colA": 1}, {"colA": 2}, {"colA": 3}], ret)
|
||||||
self.mock_txn.execute.assert_called_with(
|
self.mock_txn.execute.assert_called_with(
|
||||||
"SELECT colA FROM tablename WHERE keycol = ?", ["A set"]
|
"SELECT colA FROM tablename WHERE keycol = ?", ["A set"]
|
||||||
)
|
)
|
||||||
|
@ -31,7 +31,7 @@ class DirectoryStoreTestCase(HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
["#my-room:test"],
|
["#my-room:test"],
|
||||||
(self.get_success(self.store.get_aliases_for_room(self.room.to_string()))),
|
(self.get_success(self.store.get_aliases_for_room(self.room.to_string()))),
|
||||||
)
|
)
|
||||||
|
@ -57,7 +57,7 @@ class EventPushActionsStoreTestCase(HomeserverTestCase):
|
|||||||
"", self.store._get_unread_counts_by_pos_txn, room_id, user_id, 0
|
"", self.store._get_unread_counts_by_pos_txn, room_id, user_id, 0
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
counts,
|
counts,
|
||||||
NotifCounts(
|
NotifCounts(
|
||||||
notify_count=noitf_count,
|
notify_count=noitf_count,
|
||||||
|
@ -38,12 +38,12 @@ class DataStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
self.store.get_users_paginate(0, 10, name="bc", guests=False)
|
self.store.get_users_paginate(0, 10, name="bc", guests=False)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(1, total)
|
self.assertEqual(1, total)
|
||||||
self.assertEquals(self.displayname, users.pop()["displayname"])
|
self.assertEqual(self.displayname, users.pop()["displayname"])
|
||||||
|
|
||||||
users, total = self.get_success(
|
users, total = self.get_success(
|
||||||
self.store.get_users_paginate(0, 10, name="BC", guests=False)
|
self.store.get_users_paginate(0, 10, name="BC", guests=False)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(1, total)
|
self.assertEqual(1, total)
|
||||||
self.assertEquals(self.displayname, users.pop()["displayname"])
|
self.assertEqual(self.displayname, users.pop()["displayname"])
|
||||||
|
@ -33,7 +33,7 @@ class ProfileStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
self.store.set_profile_displayname(self.u_frank.localpart, "Frank")
|
self.store.set_profile_displayname(self.u_frank.localpart, "Frank")
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
"Frank",
|
"Frank",
|
||||||
(
|
(
|
||||||
self.get_success(
|
self.get_success(
|
||||||
@ -60,7 +60,7 @@ class ProfileStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
"http://my.site/here",
|
"http://my.site/here",
|
||||||
(
|
(
|
||||||
self.get_success(
|
self.get_success(
|
||||||
|
@ -30,7 +30,7 @@ class RegistrationStoreTestCase(HomeserverTestCase):
|
|||||||
def test_register(self):
|
def test_register(self):
|
||||||
self.get_success(self.store.register_user(self.user_id, self.pwhash))
|
self.get_success(self.store.register_user(self.user_id, self.pwhash))
|
||||||
|
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
{
|
{
|
||||||
# TODO(paul): Surely this field should be 'user_id', not 'name'
|
# TODO(paul): Surely this field should be 'user_id', not 'name'
|
||||||
"name": self.user_id,
|
"name": self.user_id,
|
||||||
@ -131,7 +131,7 @@ class RegistrationStoreTestCase(HomeserverTestCase):
|
|||||||
),
|
),
|
||||||
ThreepidValidationError,
|
ThreepidValidationError,
|
||||||
)
|
)
|
||||||
self.assertEquals(e.value.msg, "Unknown session_id", e)
|
self.assertEqual(e.value.msg, "Unknown session_id", e)
|
||||||
|
|
||||||
# Set the config setting to true.
|
# Set the config setting to true.
|
||||||
self.store._ignore_unknown_session_error = True
|
self.store._ignore_unknown_session_error = True
|
||||||
@ -146,4 +146,4 @@ class RegistrationStoreTestCase(HomeserverTestCase):
|
|||||||
),
|
),
|
||||||
ThreepidValidationError,
|
ThreepidValidationError,
|
||||||
)
|
)
|
||||||
self.assertEquals(e.value.msg, "Validation token not found or has expired", e)
|
self.assertEqual(e.value.msg, "Validation token not found or has expired", e)
|
||||||
|
@ -104,7 +104,7 @@ class RoomEventsStoreTestCase(HomeserverTestCase):
|
|||||||
self.store.get_current_state(room_id=self.room.to_string())
|
self.store.get_current_state(room_id=self.room.to_string())
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(1, len(state))
|
self.assertEqual(1, len(state))
|
||||||
self.assertObjectHasAttributes(
|
self.assertObjectHasAttributes(
|
||||||
{"type": "m.room.name", "room_id": self.room.to_string(), "name": name},
|
{"type": "m.room.name", "room_id": self.room.to_string(), "name": name},
|
||||||
state[0],
|
state[0],
|
||||||
@ -121,7 +121,7 @@ class RoomEventsStoreTestCase(HomeserverTestCase):
|
|||||||
self.store.get_current_state(room_id=self.room.to_string())
|
self.store.get_current_state(room_id=self.room.to_string())
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals(1, len(state))
|
self.assertEqual(1, len(state))
|
||||||
self.assertObjectHasAttributes(
|
self.assertObjectHasAttributes(
|
||||||
{"type": "m.room.topic", "room_id": self.room.to_string(), "topic": topic},
|
{"type": "m.room.topic", "room_id": self.room.to_string(), "topic": topic},
|
||||||
state[0],
|
state[0],
|
||||||
|
@ -53,7 +53,7 @@ class EventSearchInsertionTest(HomeserverTestCase):
|
|||||||
result = self.get_success(
|
result = self.get_success(
|
||||||
store.search_msgs([room_id], "hi bob", ["content.body"])
|
store.search_msgs([room_id], "hi bob", ["content.body"])
|
||||||
)
|
)
|
||||||
self.assertEquals(result.get("count"), 1)
|
self.assertEqual(result.get("count"), 1)
|
||||||
if isinstance(store.database_engine, PostgresEngine):
|
if isinstance(store.database_engine, PostgresEngine):
|
||||||
self.assertIn("hi", result.get("highlights"))
|
self.assertIn("hi", result.get("highlights"))
|
||||||
self.assertIn("bob", result.get("highlights"))
|
self.assertIn("bob", result.get("highlights"))
|
||||||
@ -62,14 +62,14 @@ class EventSearchInsertionTest(HomeserverTestCase):
|
|||||||
result = self.get_success(
|
result = self.get_success(
|
||||||
store.search_msgs([room_id], "another", ["content.body"])
|
store.search_msgs([room_id], "another", ["content.body"])
|
||||||
)
|
)
|
||||||
self.assertEquals(result.get("count"), 1)
|
self.assertEqual(result.get("count"), 1)
|
||||||
if isinstance(store.database_engine, PostgresEngine):
|
if isinstance(store.database_engine, PostgresEngine):
|
||||||
self.assertIn("another", result.get("highlights"))
|
self.assertIn("another", result.get("highlights"))
|
||||||
|
|
||||||
# Check that search works for a search term that overlaps with the message
|
# Check that search works for a search term that overlaps with the message
|
||||||
# containing a null byte and an unrelated message.
|
# containing a null byte and an unrelated message.
|
||||||
result = self.get_success(store.search_msgs([room_id], "hi", ["content.body"]))
|
result = self.get_success(store.search_msgs([room_id], "hi", ["content.body"]))
|
||||||
self.assertEquals(result.get("count"), 2)
|
self.assertEqual(result.get("count"), 2)
|
||||||
result = self.get_success(
|
result = self.get_success(
|
||||||
store.search_msgs([room_id], "hi alice", ["content.body"])
|
store.search_msgs([room_id], "hi alice", ["content.body"])
|
||||||
)
|
)
|
||||||
|
@ -55,7 +55,7 @@ class RoomMemberStoreTestCase(unittest.HomeserverTestCase):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.assertEquals([self.room], [m.room_id for m in rooms_for_user])
|
self.assertEqual([self.room], [m.room_id for m in rooms_for_user])
|
||||||
|
|
||||||
def test_count_known_servers(self):
|
def test_count_known_servers(self):
|
||||||
"""
|
"""
|
||||||
|
@ -48,7 +48,7 @@ class DistributorTestCase(unittest.TestCase):
|
|||||||
observers[0].assert_called_once_with("Go")
|
observers[0].assert_called_once_with("Go")
|
||||||
observers[1].assert_called_once_with("Go")
|
observers[1].assert_called_once_with("Go")
|
||||||
|
|
||||||
self.assertEquals(mock_logger.warning.call_count, 1)
|
self.assertEqual(mock_logger.warning.call_count, 1)
|
||||||
self.assertIsInstance(mock_logger.warning.call_args[0][0], str)
|
self.assertIsInstance(mock_logger.warning.call_args[0][0], str)
|
||||||
|
|
||||||
def test_signal_prereg(self):
|
def test_signal_prereg(self):
|
||||||
|
@ -54,7 +54,7 @@ class TermsTestCase(unittest.HomeserverTestCase):
|
|||||||
request_data = json.dumps({"username": "kermit", "password": "monkey"})
|
request_data = json.dumps({"username": "kermit", "password": "monkey"})
|
||||||
channel = self.make_request(b"POST", self.url, request_data)
|
channel = self.make_request(b"POST", self.url, request_data)
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
|
|
||||||
self.assertTrue(channel.json_body is not None)
|
self.assertTrue(channel.json_body is not None)
|
||||||
self.assertIsInstance(channel.json_body["session"], str)
|
self.assertIsInstance(channel.json_body["session"], str)
|
||||||
@ -99,7 +99,7 @@ class TermsTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
# We don't bother checking that the response is correct - we'll leave that to
|
# We don't bother checking that the response is correct - we'll leave that to
|
||||||
# other tests. We just want to make sure we're on the right path.
|
# other tests. We just want to make sure we're on the right path.
|
||||||
self.assertEquals(channel.result["code"], b"401", channel.result)
|
self.assertEqual(channel.result["code"], b"401", channel.result)
|
||||||
|
|
||||||
# Finish the UI auth for terms
|
# Finish the UI auth for terms
|
||||||
request_data = json.dumps(
|
request_data = json.dumps(
|
||||||
@ -117,7 +117,7 @@ class TermsTestCase(unittest.HomeserverTestCase):
|
|||||||
# We're interested in getting a response that looks like a successful
|
# We're interested in getting a response that looks like a successful
|
||||||
# registration, not so much that the details are exactly what we want.
|
# registration, not so much that the details are exactly what we want.
|
||||||
|
|
||||||
self.assertEquals(channel.result["code"], b"200", channel.result)
|
self.assertEqual(channel.result["code"], b"200", channel.result)
|
||||||
|
|
||||||
self.assertTrue(channel.json_body is not None)
|
self.assertTrue(channel.json_body is not None)
|
||||||
self.assertIsInstance(channel.json_body["user_id"], str)
|
self.assertIsInstance(channel.json_body["user_id"], str)
|
||||||
|
@ -25,7 +25,7 @@ class MockClockTestCase(unittest.TestCase):
|
|||||||
|
|
||||||
self.clock.advance_time(20)
|
self.clock.advance_time(20)
|
||||||
|
|
||||||
self.assertEquals(20, self.clock.time() - start_time)
|
self.assertEqual(20, self.clock.time() - start_time)
|
||||||
|
|
||||||
def test_later(self):
|
def test_later(self):
|
||||||
invoked = [0, 0]
|
invoked = [0, 0]
|
||||||
|
@ -22,9 +22,9 @@ class UserIDTestCase(unittest.HomeserverTestCase):
|
|||||||
def test_parse(self):
|
def test_parse(self):
|
||||||
user = UserID.from_string("@1234abcd:test")
|
user = UserID.from_string("@1234abcd:test")
|
||||||
|
|
||||||
self.assertEquals("1234abcd", user.localpart)
|
self.assertEqual("1234abcd", user.localpart)
|
||||||
self.assertEquals("test", user.domain)
|
self.assertEqual("test", user.domain)
|
||||||
self.assertEquals(True, self.hs.is_mine(user))
|
self.assertEqual(True, self.hs.is_mine(user))
|
||||||
|
|
||||||
def test_pase_empty(self):
|
def test_pase_empty(self):
|
||||||
with self.assertRaises(SynapseError):
|
with self.assertRaises(SynapseError):
|
||||||
@ -33,7 +33,7 @@ class UserIDTestCase(unittest.HomeserverTestCase):
|
|||||||
def test_build(self):
|
def test_build(self):
|
||||||
user = UserID("5678efgh", "my.domain")
|
user = UserID("5678efgh", "my.domain")
|
||||||
|
|
||||||
self.assertEquals(user.to_string(), "@5678efgh:my.domain")
|
self.assertEqual(user.to_string(), "@5678efgh:my.domain")
|
||||||
|
|
||||||
def test_compare(self):
|
def test_compare(self):
|
||||||
userA = UserID.from_string("@userA:my.domain")
|
userA = UserID.from_string("@userA:my.domain")
|
||||||
@ -48,14 +48,14 @@ class RoomAliasTestCase(unittest.HomeserverTestCase):
|
|||||||
def test_parse(self):
|
def test_parse(self):
|
||||||
room = RoomAlias.from_string("#channel:test")
|
room = RoomAlias.from_string("#channel:test")
|
||||||
|
|
||||||
self.assertEquals("channel", room.localpart)
|
self.assertEqual("channel", room.localpart)
|
||||||
self.assertEquals("test", room.domain)
|
self.assertEqual("test", room.domain)
|
||||||
self.assertEquals(True, self.hs.is_mine(room))
|
self.assertEqual(True, self.hs.is_mine(room))
|
||||||
|
|
||||||
def test_build(self):
|
def test_build(self):
|
||||||
room = RoomAlias("channel", "my.domain")
|
room = RoomAlias("channel", "my.domain")
|
||||||
|
|
||||||
self.assertEquals(room.to_string(), "#channel:my.domain")
|
self.assertEqual(room.to_string(), "#channel:my.domain")
|
||||||
|
|
||||||
def test_validate(self):
|
def test_validate(self):
|
||||||
id_string = "#test:domain,test"
|
id_string = "#test:domain,test"
|
||||||
|
@ -152,12 +152,12 @@ class TestCase(unittest.TestCase):
|
|||||||
|
|
||||||
def assertObjectHasAttributes(self, attrs, obj):
|
def assertObjectHasAttributes(self, attrs, obj):
|
||||||
"""Asserts that the given object has each of the attributes given, and
|
"""Asserts that the given object has each of the attributes given, and
|
||||||
that the value of each matches according to assertEquals."""
|
that the value of each matches according to assertEqual."""
|
||||||
for key in attrs.keys():
|
for key in attrs.keys():
|
||||||
if not hasattr(obj, key):
|
if not hasattr(obj, key):
|
||||||
raise AssertionError("Expected obj to have a '.%s'" % key)
|
raise AssertionError("Expected obj to have a '.%s'" % key)
|
||||||
try:
|
try:
|
||||||
self.assertEquals(attrs[key], getattr(obj, key))
|
self.assertEqual(attrs[key], getattr(obj, key))
|
||||||
except AssertionError as e:
|
except AssertionError as e:
|
||||||
raise (type(e))(f"Assert error for '.{key}':") from e
|
raise (type(e))(f"Assert error for '.{key}':") from e
|
||||||
|
|
||||||
@ -169,7 +169,7 @@ class TestCase(unittest.TestCase):
|
|||||||
actual (dict): The test result. Extra keys will not be checked.
|
actual (dict): The test result. Extra keys will not be checked.
|
||||||
"""
|
"""
|
||||||
for key in required:
|
for key in required:
|
||||||
self.assertEquals(
|
self.assertEqual(
|
||||||
required[key], actual[key], msg="%s mismatch. %s" % (key, actual)
|
required[key], actual[key], msg="%s mismatch. %s" % (key, actual)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -31,7 +31,7 @@ class DeferredCacheTestCase(TestCase):
|
|||||||
cache = DeferredCache("test")
|
cache = DeferredCache("test")
|
||||||
cache.prefill("foo", 123)
|
cache.prefill("foo", 123)
|
||||||
|
|
||||||
self.assertEquals(self.successResultOf(cache.get("foo")), 123)
|
self.assertEqual(self.successResultOf(cache.get("foo")), 123)
|
||||||
|
|
||||||
def test_hit_deferred(self):
|
def test_hit_deferred(self):
|
||||||
cache = DeferredCache("test")
|
cache = DeferredCache("test")
|
||||||
|
@ -434,8 +434,8 @@ class CacheDecoratorTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
a = A()
|
a = A()
|
||||||
|
|
||||||
self.assertEquals((yield a.func("foo")), "foo")
|
self.assertEqual((yield a.func("foo")), "foo")
|
||||||
self.assertEquals((yield a.func("bar")), "bar")
|
self.assertEqual((yield a.func("bar")), "bar")
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def test_hit(self):
|
def test_hit(self):
|
||||||
@ -450,10 +450,10 @@ class CacheDecoratorTestCase(unittest.HomeserverTestCase):
|
|||||||
a = A()
|
a = A()
|
||||||
yield a.func("foo")
|
yield a.func("foo")
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 1)
|
self.assertEqual(callcount[0], 1)
|
||||||
|
|
||||||
self.assertEquals((yield a.func("foo")), "foo")
|
self.assertEqual((yield a.func("foo")), "foo")
|
||||||
self.assertEquals(callcount[0], 1)
|
self.assertEqual(callcount[0], 1)
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def test_invalidate(self):
|
def test_invalidate(self):
|
||||||
@ -468,13 +468,13 @@ class CacheDecoratorTestCase(unittest.HomeserverTestCase):
|
|||||||
a = A()
|
a = A()
|
||||||
yield a.func("foo")
|
yield a.func("foo")
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 1)
|
self.assertEqual(callcount[0], 1)
|
||||||
|
|
||||||
a.func.invalidate(("foo",))
|
a.func.invalidate(("foo",))
|
||||||
|
|
||||||
yield a.func("foo")
|
yield a.func("foo")
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 2)
|
self.assertEqual(callcount[0], 2)
|
||||||
|
|
||||||
def test_invalidate_missing(self):
|
def test_invalidate_missing(self):
|
||||||
class A:
|
class A:
|
||||||
@ -499,7 +499,7 @@ class CacheDecoratorTestCase(unittest.HomeserverTestCase):
|
|||||||
for k in range(0, 12):
|
for k in range(0, 12):
|
||||||
yield a.func(k)
|
yield a.func(k)
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 12)
|
self.assertEqual(callcount[0], 12)
|
||||||
|
|
||||||
# There must have been at least 2 evictions, meaning if we calculate
|
# There must have been at least 2 evictions, meaning if we calculate
|
||||||
# all 12 values again, we must get called at least 2 more times
|
# all 12 values again, we must get called at least 2 more times
|
||||||
@ -525,8 +525,8 @@ class CacheDecoratorTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
a.func.prefill(("foo",), 456)
|
a.func.prefill(("foo",), 456)
|
||||||
|
|
||||||
self.assertEquals(a.func("foo").result, 456)
|
self.assertEqual(a.func("foo").result, 456)
|
||||||
self.assertEquals(callcount[0], 0)
|
self.assertEqual(callcount[0], 0)
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def test_invalidate_context(self):
|
def test_invalidate_context(self):
|
||||||
@ -547,19 +547,19 @@ class CacheDecoratorTestCase(unittest.HomeserverTestCase):
|
|||||||
a = A()
|
a = A()
|
||||||
yield a.func2("foo")
|
yield a.func2("foo")
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 1)
|
self.assertEqual(callcount[0], 1)
|
||||||
self.assertEquals(callcount2[0], 1)
|
self.assertEqual(callcount2[0], 1)
|
||||||
|
|
||||||
a.func.invalidate(("foo",))
|
a.func.invalidate(("foo",))
|
||||||
yield a.func("foo")
|
yield a.func("foo")
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 2)
|
self.assertEqual(callcount[0], 2)
|
||||||
self.assertEquals(callcount2[0], 1)
|
self.assertEqual(callcount2[0], 1)
|
||||||
|
|
||||||
yield a.func2("foo")
|
yield a.func2("foo")
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 2)
|
self.assertEqual(callcount[0], 2)
|
||||||
self.assertEquals(callcount2[0], 2)
|
self.assertEqual(callcount2[0], 2)
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def test_eviction_context(self):
|
def test_eviction_context(self):
|
||||||
@ -581,22 +581,22 @@ class CacheDecoratorTestCase(unittest.HomeserverTestCase):
|
|||||||
yield a.func2("foo")
|
yield a.func2("foo")
|
||||||
yield a.func2("foo2")
|
yield a.func2("foo2")
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 2)
|
self.assertEqual(callcount[0], 2)
|
||||||
self.assertEquals(callcount2[0], 2)
|
self.assertEqual(callcount2[0], 2)
|
||||||
|
|
||||||
yield a.func2("foo")
|
yield a.func2("foo")
|
||||||
self.assertEquals(callcount[0], 2)
|
self.assertEqual(callcount[0], 2)
|
||||||
self.assertEquals(callcount2[0], 2)
|
self.assertEqual(callcount2[0], 2)
|
||||||
|
|
||||||
yield a.func("foo3")
|
yield a.func("foo3")
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 3)
|
self.assertEqual(callcount[0], 3)
|
||||||
self.assertEquals(callcount2[0], 2)
|
self.assertEqual(callcount2[0], 2)
|
||||||
|
|
||||||
yield a.func2("foo")
|
yield a.func2("foo")
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 4)
|
self.assertEqual(callcount[0], 4)
|
||||||
self.assertEquals(callcount2[0], 3)
|
self.assertEqual(callcount2[0], 3)
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def test_double_get(self):
|
def test_double_get(self):
|
||||||
@ -619,30 +619,30 @@ class CacheDecoratorTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
yield a.func2("foo")
|
yield a.func2("foo")
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 1)
|
self.assertEqual(callcount[0], 1)
|
||||||
self.assertEquals(callcount2[0], 1)
|
self.assertEqual(callcount2[0], 1)
|
||||||
|
|
||||||
a.func2.invalidate(("foo",))
|
a.func2.invalidate(("foo",))
|
||||||
self.assertEquals(a.func2.cache.cache.del_multi.call_count, 1)
|
self.assertEqual(a.func2.cache.cache.del_multi.call_count, 1)
|
||||||
|
|
||||||
yield a.func2("foo")
|
yield a.func2("foo")
|
||||||
a.func2.invalidate(("foo",))
|
a.func2.invalidate(("foo",))
|
||||||
self.assertEquals(a.func2.cache.cache.del_multi.call_count, 2)
|
self.assertEqual(a.func2.cache.cache.del_multi.call_count, 2)
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 1)
|
self.assertEqual(callcount[0], 1)
|
||||||
self.assertEquals(callcount2[0], 2)
|
self.assertEqual(callcount2[0], 2)
|
||||||
|
|
||||||
a.func.invalidate(("foo",))
|
a.func.invalidate(("foo",))
|
||||||
self.assertEquals(a.func2.cache.cache.del_multi.call_count, 3)
|
self.assertEqual(a.func2.cache.cache.del_multi.call_count, 3)
|
||||||
yield a.func("foo")
|
yield a.func("foo")
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 2)
|
self.assertEqual(callcount[0], 2)
|
||||||
self.assertEquals(callcount2[0], 2)
|
self.assertEqual(callcount2[0], 2)
|
||||||
|
|
||||||
yield a.func2("foo")
|
yield a.func2("foo")
|
||||||
|
|
||||||
self.assertEquals(callcount[0], 2)
|
self.assertEqual(callcount[0], 2)
|
||||||
self.assertEquals(callcount2[0], 3)
|
self.assertEqual(callcount2[0], 3)
|
||||||
|
|
||||||
|
|
||||||
class CachedListDescriptorTestCase(unittest.TestCase):
|
class CachedListDescriptorTestCase(unittest.TestCase):
|
||||||
|
@ -26,8 +26,8 @@ class ExpiringCacheTestCase(unittest.HomeserverTestCase):
|
|||||||
cache = ExpiringCache("test", clock, max_len=1)
|
cache = ExpiringCache("test", clock, max_len=1)
|
||||||
|
|
||||||
cache["key"] = "value"
|
cache["key"] = "value"
|
||||||
self.assertEquals(cache.get("key"), "value")
|
self.assertEqual(cache.get("key"), "value")
|
||||||
self.assertEquals(cache["key"], "value")
|
self.assertEqual(cache["key"], "value")
|
||||||
|
|
||||||
def test_eviction(self):
|
def test_eviction(self):
|
||||||
clock = MockClock()
|
clock = MockClock()
|
||||||
@ -35,13 +35,13 @@ class ExpiringCacheTestCase(unittest.HomeserverTestCase):
|
|||||||
|
|
||||||
cache["key"] = "value"
|
cache["key"] = "value"
|
||||||
cache["key2"] = "value2"
|
cache["key2"] = "value2"
|
||||||
self.assertEquals(cache.get("key"), "value")
|
self.assertEqual(cache.get("key"), "value")
|
||||||
self.assertEquals(cache.get("key2"), "value2")
|
self.assertEqual(cache.get("key2"), "value2")
|
||||||
|
|
||||||
cache["key3"] = "value3"
|
cache["key3"] = "value3"
|
||||||
self.assertEquals(cache.get("key"), None)
|
self.assertEqual(cache.get("key"), None)
|
||||||
self.assertEquals(cache.get("key2"), "value2")
|
self.assertEqual(cache.get("key2"), "value2")
|
||||||
self.assertEquals(cache.get("key3"), "value3")
|
self.assertEqual(cache.get("key3"), "value3")
|
||||||
|
|
||||||
def test_iterable_eviction(self):
|
def test_iterable_eviction(self):
|
||||||
clock = MockClock()
|
clock = MockClock()
|
||||||
@ -51,15 +51,15 @@ class ExpiringCacheTestCase(unittest.HomeserverTestCase):
|
|||||||
cache["key2"] = [2, 3]
|
cache["key2"] = [2, 3]
|
||||||
cache["key3"] = [4, 5]
|
cache["key3"] = [4, 5]
|
||||||
|
|
||||||
self.assertEquals(cache.get("key"), [1])
|
self.assertEqual(cache.get("key"), [1])
|
||||||
self.assertEquals(cache.get("key2"), [2, 3])
|
self.assertEqual(cache.get("key2"), [2, 3])
|
||||||
self.assertEquals(cache.get("key3"), [4, 5])
|
self.assertEqual(cache.get("key3"), [4, 5])
|
||||||
|
|
||||||
cache["key4"] = [6, 7]
|
cache["key4"] = [6, 7]
|
||||||
self.assertEquals(cache.get("key"), None)
|
self.assertEqual(cache.get("key"), None)
|
||||||
self.assertEquals(cache.get("key2"), None)
|
self.assertEqual(cache.get("key2"), None)
|
||||||
self.assertEquals(cache.get("key3"), [4, 5])
|
self.assertEqual(cache.get("key3"), [4, 5])
|
||||||
self.assertEquals(cache.get("key4"), [6, 7])
|
self.assertEqual(cache.get("key4"), [6, 7])
|
||||||
|
|
||||||
def test_time_eviction(self):
|
def test_time_eviction(self):
|
||||||
clock = MockClock()
|
clock = MockClock()
|
||||||
@ -69,13 +69,13 @@ class ExpiringCacheTestCase(unittest.HomeserverTestCase):
|
|||||||
clock.advance_time(0.5)
|
clock.advance_time(0.5)
|
||||||
cache["key2"] = 2
|
cache["key2"] = 2
|
||||||
|
|
||||||
self.assertEquals(cache.get("key"), 1)
|
self.assertEqual(cache.get("key"), 1)
|
||||||
self.assertEquals(cache.get("key2"), 2)
|
self.assertEqual(cache.get("key2"), 2)
|
||||||
|
|
||||||
clock.advance_time(0.9)
|
clock.advance_time(0.9)
|
||||||
self.assertEquals(cache.get("key"), None)
|
self.assertEqual(cache.get("key"), None)
|
||||||
self.assertEquals(cache.get("key2"), 2)
|
self.assertEqual(cache.get("key2"), 2)
|
||||||
|
|
||||||
clock.advance_time(1)
|
clock.advance_time(1)
|
||||||
self.assertEquals(cache.get("key"), None)
|
self.assertEqual(cache.get("key"), None)
|
||||||
self.assertEquals(cache.get("key2"), None)
|
self.assertEqual(cache.get("key2"), None)
|
||||||
|
@ -17,7 +17,7 @@ from .. import unittest
|
|||||||
|
|
||||||
class LoggingContextTestCase(unittest.TestCase):
|
class LoggingContextTestCase(unittest.TestCase):
|
||||||
def _check_test_key(self, value):
|
def _check_test_key(self, value):
|
||||||
self.assertEquals(current_context().name, value)
|
self.assertEqual(current_context().name, value)
|
||||||
|
|
||||||
def test_with_context(self):
|
def test_with_context(self):
|
||||||
with LoggingContext("test"):
|
with LoggingContext("test"):
|
||||||
|
@ -27,37 +27,37 @@ class LruCacheTestCase(unittest.HomeserverTestCase):
|
|||||||
def test_get_set(self):
|
def test_get_set(self):
|
||||||
cache = LruCache(1)
|
cache = LruCache(1)
|
||||||
cache["key"] = "value"
|
cache["key"] = "value"
|
||||||
self.assertEquals(cache.get("key"), "value")
|
self.assertEqual(cache.get("key"), "value")
|
||||||
self.assertEquals(cache["key"], "value")
|
self.assertEqual(cache["key"], "value")
|
||||||
|
|
||||||
def test_eviction(self):
|
def test_eviction(self):
|
||||||
cache = LruCache(2)
|
cache = LruCache(2)
|
||||||
cache[1] = 1
|
cache[1] = 1
|
||||||
cache[2] = 2
|
cache[2] = 2
|
||||||
|
|
||||||
self.assertEquals(cache.get(1), 1)
|
self.assertEqual(cache.get(1), 1)
|
||||||
self.assertEquals(cache.get(2), 2)
|
self.assertEqual(cache.get(2), 2)
|
||||||
|
|
||||||
cache[3] = 3
|
cache[3] = 3
|
||||||
|
|
||||||
self.assertEquals(cache.get(1), None)
|
self.assertEqual(cache.get(1), None)
|
||||||
self.assertEquals(cache.get(2), 2)
|
self.assertEqual(cache.get(2), 2)
|
||||||
self.assertEquals(cache.get(3), 3)
|
self.assertEqual(cache.get(3), 3)
|
||||||
|
|
||||||
def test_setdefault(self):
|
def test_setdefault(self):
|
||||||
cache = LruCache(1)
|
cache = LruCache(1)
|
||||||
self.assertEquals(cache.setdefault("key", 1), 1)
|
self.assertEqual(cache.setdefault("key", 1), 1)
|
||||||
self.assertEquals(cache.get("key"), 1)
|
self.assertEqual(cache.get("key"), 1)
|
||||||
self.assertEquals(cache.setdefault("key", 2), 1)
|
self.assertEqual(cache.setdefault("key", 2), 1)
|
||||||
self.assertEquals(cache.get("key"), 1)
|
self.assertEqual(cache.get("key"), 1)
|
||||||
cache["key"] = 2 # Make sure overriding works.
|
cache["key"] = 2 # Make sure overriding works.
|
||||||
self.assertEquals(cache.get("key"), 2)
|
self.assertEqual(cache.get("key"), 2)
|
||||||
|
|
||||||
def test_pop(self):
|
def test_pop(self):
|
||||||
cache = LruCache(1)
|
cache = LruCache(1)
|
||||||
cache["key"] = 1
|
cache["key"] = 1
|
||||||
self.assertEquals(cache.pop("key"), 1)
|
self.assertEqual(cache.pop("key"), 1)
|
||||||
self.assertEquals(cache.pop("key"), None)
|
self.assertEqual(cache.pop("key"), None)
|
||||||
|
|
||||||
def test_del_multi(self):
|
def test_del_multi(self):
|
||||||
cache = LruCache(4, cache_type=TreeCache)
|
cache = LruCache(4, cache_type=TreeCache)
|
||||||
@ -66,23 +66,23 @@ class LruCacheTestCase(unittest.HomeserverTestCase):
|
|||||||
cache[("vehicles", "car")] = "vroom"
|
cache[("vehicles", "car")] = "vroom"
|
||||||
cache[("vehicles", "train")] = "chuff"
|
cache[("vehicles", "train")] = "chuff"
|
||||||
|
|
||||||
self.assertEquals(len(cache), 4)
|
self.assertEqual(len(cache), 4)
|
||||||
|
|
||||||
self.assertEquals(cache.get(("animal", "cat")), "mew")
|
self.assertEqual(cache.get(("animal", "cat")), "mew")
|
||||||
self.assertEquals(cache.get(("vehicles", "car")), "vroom")
|
self.assertEqual(cache.get(("vehicles", "car")), "vroom")
|
||||||
cache.del_multi(("animal",))
|
cache.del_multi(("animal",))
|
||||||
self.assertEquals(len(cache), 2)
|
self.assertEqual(len(cache), 2)
|
||||||
self.assertEquals(cache.get(("animal", "cat")), None)
|
self.assertEqual(cache.get(("animal", "cat")), None)
|
||||||
self.assertEquals(cache.get(("animal", "dog")), None)
|
self.assertEqual(cache.get(("animal", "dog")), None)
|
||||||
self.assertEquals(cache.get(("vehicles", "car")), "vroom")
|
self.assertEqual(cache.get(("vehicles", "car")), "vroom")
|
||||||
self.assertEquals(cache.get(("vehicles", "train")), "chuff")
|
self.assertEqual(cache.get(("vehicles", "train")), "chuff")
|
||||||
# Man from del_multi say "Yes".
|
# Man from del_multi say "Yes".
|
||||||
|
|
||||||
def test_clear(self):
|
def test_clear(self):
|
||||||
cache = LruCache(1)
|
cache = LruCache(1)
|
||||||
cache["key"] = 1
|
cache["key"] = 1
|
||||||
cache.clear()
|
cache.clear()
|
||||||
self.assertEquals(len(cache), 0)
|
self.assertEqual(len(cache), 0)
|
||||||
|
|
||||||
@override_config({"caches": {"per_cache_factors": {"mycache": 10}}})
|
@override_config({"caches": {"per_cache_factors": {"mycache": 10}}})
|
||||||
def test_special_size(self):
|
def test_special_size(self):
|
||||||
@ -105,10 +105,10 @@ class LruCacheCallbacksTestCase(unittest.HomeserverTestCase):
|
|||||||
self.assertFalse(m.called)
|
self.assertFalse(m.called)
|
||||||
|
|
||||||
cache.set("key", "value2")
|
cache.set("key", "value2")
|
||||||
self.assertEquals(m.call_count, 1)
|
self.assertEqual(m.call_count, 1)
|
||||||
|
|
||||||
cache.set("key", "value")
|
cache.set("key", "value")
|
||||||
self.assertEquals(m.call_count, 1)
|
self.assertEqual(m.call_count, 1)
|
||||||
|
|
||||||
def test_multi_get(self):
|
def test_multi_get(self):
|
||||||
m = Mock()
|
m = Mock()
|
||||||
@ -124,10 +124,10 @@ class LruCacheCallbacksTestCase(unittest.HomeserverTestCase):
|
|||||||
self.assertFalse(m.called)
|
self.assertFalse(m.called)
|
||||||
|
|
||||||
cache.set("key", "value2")
|
cache.set("key", "value2")
|
||||||
self.assertEquals(m.call_count, 1)
|
self.assertEqual(m.call_count, 1)
|
||||||
|
|
||||||
cache.set("key", "value")
|
cache.set("key", "value")
|
||||||
self.assertEquals(m.call_count, 1)
|
self.assertEqual(m.call_count, 1)
|
||||||
|
|
||||||
def test_set(self):
|
def test_set(self):
|
||||||
m = Mock()
|
m = Mock()
|
||||||
@ -140,10 +140,10 @@ class LruCacheCallbacksTestCase(unittest.HomeserverTestCase):
|
|||||||
self.assertFalse(m.called)
|
self.assertFalse(m.called)
|
||||||
|
|
||||||
cache.set("key", "value2")
|
cache.set("key", "value2")
|
||||||
self.assertEquals(m.call_count, 1)
|
self.assertEqual(m.call_count, 1)
|
||||||
|
|
||||||
cache.set("key", "value")
|
cache.set("key", "value")
|
||||||
self.assertEquals(m.call_count, 1)
|
self.assertEqual(m.call_count, 1)
|
||||||
|
|
||||||
def test_pop(self):
|
def test_pop(self):
|
||||||
m = Mock()
|
m = Mock()
|
||||||
@ -153,13 +153,13 @@ class LruCacheCallbacksTestCase(unittest.HomeserverTestCase):
|
|||||||
self.assertFalse(m.called)
|
self.assertFalse(m.called)
|
||||||
|
|
||||||
cache.pop("key")
|
cache.pop("key")
|
||||||
self.assertEquals(m.call_count, 1)
|
self.assertEqual(m.call_count, 1)
|
||||||
|
|
||||||
cache.set("key", "value")
|
cache.set("key", "value")
|
||||||
self.assertEquals(m.call_count, 1)
|
self.assertEqual(m.call_count, 1)
|
||||||
|
|
||||||
cache.pop("key")
|
cache.pop("key")
|
||||||
self.assertEquals(m.call_count, 1)
|
self.assertEqual(m.call_count, 1)
|
||||||
|
|
||||||
def test_del_multi(self):
|
def test_del_multi(self):
|
||||||
m1 = Mock()
|
m1 = Mock()
|
||||||
@ -173,17 +173,17 @@ class LruCacheCallbacksTestCase(unittest.HomeserverTestCase):
|
|||||||
cache.set(("b", "1"), "value", callbacks=[m3])
|
cache.set(("b", "1"), "value", callbacks=[m3])
|
||||||
cache.set(("b", "2"), "value", callbacks=[m4])
|
cache.set(("b", "2"), "value", callbacks=[m4])
|
||||||
|
|
||||||
self.assertEquals(m1.call_count, 0)
|
self.assertEqual(m1.call_count, 0)
|
||||||
self.assertEquals(m2.call_count, 0)
|
self.assertEqual(m2.call_count, 0)
|
||||||
self.assertEquals(m3.call_count, 0)
|
self.assertEqual(m3.call_count, 0)
|
||||||
self.assertEquals(m4.call_count, 0)
|
self.assertEqual(m4.call_count, 0)
|
||||||
|
|
||||||
cache.del_multi(("a",))
|
cache.del_multi(("a",))
|
||||||
|
|
||||||
self.assertEquals(m1.call_count, 1)
|
self.assertEqual(m1.call_count, 1)
|
||||||
self.assertEquals(m2.call_count, 1)
|
self.assertEqual(m2.call_count, 1)
|
||||||
self.assertEquals(m3.call_count, 0)
|
self.assertEqual(m3.call_count, 0)
|
||||||
self.assertEquals(m4.call_count, 0)
|
self.assertEqual(m4.call_count, 0)
|
||||||
|
|
||||||
def test_clear(self):
|
def test_clear(self):
|
||||||
m1 = Mock()
|
m1 = Mock()
|
||||||
@ -193,13 +193,13 @@ class LruCacheCallbacksTestCase(unittest.HomeserverTestCase):
|
|||||||
cache.set("key1", "value", callbacks=[m1])
|
cache.set("key1", "value", callbacks=[m1])
|
||||||
cache.set("key2", "value", callbacks=[m2])
|
cache.set("key2", "value", callbacks=[m2])
|
||||||
|
|
||||||
self.assertEquals(m1.call_count, 0)
|
self.assertEqual(m1.call_count, 0)
|
||||||
self.assertEquals(m2.call_count, 0)
|
self.assertEqual(m2.call_count, 0)
|
||||||
|
|
||||||
cache.clear()
|
cache.clear()
|
||||||
|
|
||||||
self.assertEquals(m1.call_count, 1)
|
self.assertEqual(m1.call_count, 1)
|
||||||
self.assertEquals(m2.call_count, 1)
|
self.assertEqual(m2.call_count, 1)
|
||||||
|
|
||||||
def test_eviction(self):
|
def test_eviction(self):
|
||||||
m1 = Mock(name="m1")
|
m1 = Mock(name="m1")
|
||||||
@ -210,33 +210,33 @@ class LruCacheCallbacksTestCase(unittest.HomeserverTestCase):
|
|||||||
cache.set("key1", "value", callbacks=[m1])
|
cache.set("key1", "value", callbacks=[m1])
|
||||||
cache.set("key2", "value", callbacks=[m2])
|
cache.set("key2", "value", callbacks=[m2])
|
||||||
|
|
||||||
self.assertEquals(m1.call_count, 0)
|
self.assertEqual(m1.call_count, 0)
|
||||||
self.assertEquals(m2.call_count, 0)
|
self.assertEqual(m2.call_count, 0)
|
||||||
self.assertEquals(m3.call_count, 0)
|
self.assertEqual(m3.call_count, 0)
|
||||||
|
|
||||||
cache.set("key3", "value", callbacks=[m3])
|
cache.set("key3", "value", callbacks=[m3])
|
||||||
|
|
||||||
self.assertEquals(m1.call_count, 1)
|
self.assertEqual(m1.call_count, 1)
|
||||||
self.assertEquals(m2.call_count, 0)
|
self.assertEqual(m2.call_count, 0)
|
||||||
self.assertEquals(m3.call_count, 0)
|
self.assertEqual(m3.call_count, 0)
|
||||||
|
|
||||||
cache.set("key3", "value")
|
cache.set("key3", "value")
|
||||||
|
|
||||||
self.assertEquals(m1.call_count, 1)
|
self.assertEqual(m1.call_count, 1)
|
||||||
self.assertEquals(m2.call_count, 0)
|
self.assertEqual(m2.call_count, 0)
|
||||||
self.assertEquals(m3.call_count, 0)
|
self.assertEqual(m3.call_count, 0)
|
||||||
|
|
||||||
cache.get("key2")
|
cache.get("key2")
|
||||||
|
|
||||||
self.assertEquals(m1.call_count, 1)
|
self.assertEqual(m1.call_count, 1)
|
||||||
self.assertEquals(m2.call_count, 0)
|
self.assertEqual(m2.call_count, 0)
|
||||||
self.assertEquals(m3.call_count, 0)
|
self.assertEqual(m3.call_count, 0)
|
||||||
|
|
||||||
cache.set("key1", "value", callbacks=[m1])
|
cache.set("key1", "value", callbacks=[m1])
|
||||||
|
|
||||||
self.assertEquals(m1.call_count, 1)
|
self.assertEqual(m1.call_count, 1)
|
||||||
self.assertEquals(m2.call_count, 0)
|
self.assertEqual(m2.call_count, 0)
|
||||||
self.assertEquals(m3.call_count, 1)
|
self.assertEqual(m3.call_count, 1)
|
||||||
|
|
||||||
|
|
||||||
class LruCacheSizedTestCase(unittest.HomeserverTestCase):
|
class LruCacheSizedTestCase(unittest.HomeserverTestCase):
|
||||||
@ -247,20 +247,20 @@ class LruCacheSizedTestCase(unittest.HomeserverTestCase):
|
|||||||
cache["key3"] = [3]
|
cache["key3"] = [3]
|
||||||
cache["key4"] = [4]
|
cache["key4"] = [4]
|
||||||
|
|
||||||
self.assertEquals(cache["key1"], [0])
|
self.assertEqual(cache["key1"], [0])
|
||||||
self.assertEquals(cache["key2"], [1, 2])
|
self.assertEqual(cache["key2"], [1, 2])
|
||||||
self.assertEquals(cache["key3"], [3])
|
self.assertEqual(cache["key3"], [3])
|
||||||
self.assertEquals(cache["key4"], [4])
|
self.assertEqual(cache["key4"], [4])
|
||||||
self.assertEquals(len(cache), 5)
|
self.assertEqual(len(cache), 5)
|
||||||
|
|
||||||
cache["key5"] = [5, 6]
|
cache["key5"] = [5, 6]
|
||||||
|
|
||||||
self.assertEquals(len(cache), 4)
|
self.assertEqual(len(cache), 4)
|
||||||
self.assertEquals(cache.get("key1"), None)
|
self.assertEqual(cache.get("key1"), None)
|
||||||
self.assertEquals(cache.get("key2"), None)
|
self.assertEqual(cache.get("key2"), None)
|
||||||
self.assertEquals(cache["key3"], [3])
|
self.assertEqual(cache["key3"], [3])
|
||||||
self.assertEquals(cache["key4"], [4])
|
self.assertEqual(cache["key4"], [4])
|
||||||
self.assertEquals(cache["key5"], [5, 6])
|
self.assertEqual(cache["key5"], [5, 6])
|
||||||
|
|
||||||
def test_zero_size_drop_from_cache(self) -> None:
|
def test_zero_size_drop_from_cache(self) -> None:
|
||||||
"""Test that `drop_from_cache` works correctly with 0-sized entries."""
|
"""Test that `drop_from_cache` works correctly with 0-sized entries."""
|
||||||
|
@ -23,61 +23,61 @@ class TreeCacheTestCase(unittest.TestCase):
|
|||||||
cache = TreeCache()
|
cache = TreeCache()
|
||||||
cache[("a",)] = "A"
|
cache[("a",)] = "A"
|
||||||
cache[("b",)] = "B"
|
cache[("b",)] = "B"
|
||||||
self.assertEquals(cache.get(("a",)), "A")
|
self.assertEqual(cache.get(("a",)), "A")
|
||||||
self.assertEquals(cache.get(("b",)), "B")
|
self.assertEqual(cache.get(("b",)), "B")
|
||||||
self.assertEquals(len(cache), 2)
|
self.assertEqual(len(cache), 2)
|
||||||
|
|
||||||
def test_pop_onelevel(self):
|
def test_pop_onelevel(self):
|
||||||
cache = TreeCache()
|
cache = TreeCache()
|
||||||
cache[("a",)] = "A"
|
cache[("a",)] = "A"
|
||||||
cache[("b",)] = "B"
|
cache[("b",)] = "B"
|
||||||
self.assertEquals(cache.pop(("a",)), "A")
|
self.assertEqual(cache.pop(("a",)), "A")
|
||||||
self.assertEquals(cache.pop(("a",)), None)
|
self.assertEqual(cache.pop(("a",)), None)
|
||||||
self.assertEquals(cache.get(("b",)), "B")
|
self.assertEqual(cache.get(("b",)), "B")
|
||||||
self.assertEquals(len(cache), 1)
|
self.assertEqual(len(cache), 1)
|
||||||
|
|
||||||
def test_get_set_twolevel(self):
|
def test_get_set_twolevel(self):
|
||||||
cache = TreeCache()
|
cache = TreeCache()
|
||||||
cache[("a", "a")] = "AA"
|
cache[("a", "a")] = "AA"
|
||||||
cache[("a", "b")] = "AB"
|
cache[("a", "b")] = "AB"
|
||||||
cache[("b", "a")] = "BA"
|
cache[("b", "a")] = "BA"
|
||||||
self.assertEquals(cache.get(("a", "a")), "AA")
|
self.assertEqual(cache.get(("a", "a")), "AA")
|
||||||
self.assertEquals(cache.get(("a", "b")), "AB")
|
self.assertEqual(cache.get(("a", "b")), "AB")
|
||||||
self.assertEquals(cache.get(("b", "a")), "BA")
|
self.assertEqual(cache.get(("b", "a")), "BA")
|
||||||
self.assertEquals(len(cache), 3)
|
self.assertEqual(len(cache), 3)
|
||||||
|
|
||||||
def test_pop_twolevel(self):
|
def test_pop_twolevel(self):
|
||||||
cache = TreeCache()
|
cache = TreeCache()
|
||||||
cache[("a", "a")] = "AA"
|
cache[("a", "a")] = "AA"
|
||||||
cache[("a", "b")] = "AB"
|
cache[("a", "b")] = "AB"
|
||||||
cache[("b", "a")] = "BA"
|
cache[("b", "a")] = "BA"
|
||||||
self.assertEquals(cache.pop(("a", "a")), "AA")
|
self.assertEqual(cache.pop(("a", "a")), "AA")
|
||||||
self.assertEquals(cache.get(("a", "a")), None)
|
self.assertEqual(cache.get(("a", "a")), None)
|
||||||
self.assertEquals(cache.get(("a", "b")), "AB")
|
self.assertEqual(cache.get(("a", "b")), "AB")
|
||||||
self.assertEquals(cache.pop(("b", "a")), "BA")
|
self.assertEqual(cache.pop(("b", "a")), "BA")
|
||||||
self.assertEquals(cache.pop(("b", "a")), None)
|
self.assertEqual(cache.pop(("b", "a")), None)
|
||||||
self.assertEquals(len(cache), 1)
|
self.assertEqual(len(cache), 1)
|
||||||
|
|
||||||
def test_pop_mixedlevel(self):
|
def test_pop_mixedlevel(self):
|
||||||
cache = TreeCache()
|
cache = TreeCache()
|
||||||
cache[("a", "a")] = "AA"
|
cache[("a", "a")] = "AA"
|
||||||
cache[("a", "b")] = "AB"
|
cache[("a", "b")] = "AB"
|
||||||
cache[("b", "a")] = "BA"
|
cache[("b", "a")] = "BA"
|
||||||
self.assertEquals(cache.get(("a", "a")), "AA")
|
self.assertEqual(cache.get(("a", "a")), "AA")
|
||||||
popped = cache.pop(("a",))
|
popped = cache.pop(("a",))
|
||||||
self.assertEquals(cache.get(("a", "a")), None)
|
self.assertEqual(cache.get(("a", "a")), None)
|
||||||
self.assertEquals(cache.get(("a", "b")), None)
|
self.assertEqual(cache.get(("a", "b")), None)
|
||||||
self.assertEquals(cache.get(("b", "a")), "BA")
|
self.assertEqual(cache.get(("b", "a")), "BA")
|
||||||
self.assertEquals(len(cache), 1)
|
self.assertEqual(len(cache), 1)
|
||||||
|
|
||||||
self.assertEquals({"AA", "AB"}, set(iterate_tree_cache_entry(popped)))
|
self.assertEqual({"AA", "AB"}, set(iterate_tree_cache_entry(popped)))
|
||||||
|
|
||||||
def test_clear(self):
|
def test_clear(self):
|
||||||
cache = TreeCache()
|
cache = TreeCache()
|
||||||
cache[("a",)] = "A"
|
cache[("a",)] = "A"
|
||||||
cache[("b",)] = "B"
|
cache[("b",)] = "B"
|
||||||
cache.clear()
|
cache.clear()
|
||||||
self.assertEquals(len(cache), 0)
|
self.assertEqual(len(cache), 0)
|
||||||
|
|
||||||
def test_contains(self):
|
def test_contains(self):
|
||||||
cache = TreeCache()
|
cache = TreeCache()
|
||||||
|
Loading…
Reference in New Issue
Block a user