diff options
| author | David Robertson <davidr@element.io> | 2021-11-16 10:41:35 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-11-16 10:41:35 +0000 |
| commit | 0dda1a79687b8375dd5b23763ba1585e5dad030d (patch) | |
| tree | dd60cd7bd9585e775c1b65b23de8997db1298a39 /tests/handlers | |
| parent | change 'Home Server' to one word 'homeserver' (#11320) (diff) | |
| download | synapse-0dda1a79687b8375dd5b23763ba1585e5dad030d.tar.xz | |
Misc typing fixes for tests, part 2 of N (#11330)
Diffstat (limited to 'tests/handlers')
| -rw-r--r-- | tests/handlers/test_register.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/tests/handlers/test_register.py b/tests/handlers/test_register.py index db691c4c1c..cd6f2c77ae 100644 --- a/tests/handlers/test_register.py +++ b/tests/handlers/test_register.py @@ -193,7 +193,8 @@ class RegistrationTestCase(unittest.HomeserverTestCase): @override_config({"limit_usage_by_mau": True}) def test_get_or_create_user_mau_not_blocked(self): - self.store.count_monthly_users = Mock( + # Type ignore: mypy doesn't like us assigning to methods. + self.store.count_monthly_users = Mock( # type: ignore[assignment] return_value=make_awaitable(self.hs.config.server.max_mau_value - 1) ) # Ensure does not throw exception @@ -201,7 +202,8 @@ class RegistrationTestCase(unittest.HomeserverTestCase): @override_config({"limit_usage_by_mau": True}) def test_get_or_create_user_mau_blocked(self): - self.store.get_monthly_active_count = Mock( + # Type ignore: mypy doesn't like us assigning to methods. + self.store.get_monthly_active_count = Mock( # type: ignore[assignment] return_value=make_awaitable(self.lots_of_users) ) self.get_failure( @@ -209,7 +211,8 @@ class RegistrationTestCase(unittest.HomeserverTestCase): ResourceLimitError, ) - self.store.get_monthly_active_count = Mock( + # Type ignore: mypy doesn't like us assigning to methods. + self.store.get_monthly_active_count = Mock( # type: ignore[assignment] return_value=make_awaitable(self.hs.config.server.max_mau_value) ) self.get_failure( |