From 46c0ab559b3713a99c9ac6453243c16ee66b1a3b Mon Sep 17 00:00:00 2001 From: Andrew Morgan Date: Thu, 9 Mar 2023 14:17:12 +0000 Subject: Move callback-related code from the PasswordAuthProvider to its own class --- tests/handlers/test_password_providers.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) (limited to 'tests') diff --git a/tests/handlers/test_password_providers.py b/tests/handlers/test_password_providers.py index aa91bc0a3d..dd55b0d285 100644 --- a/tests/handlers/test_password_providers.py +++ b/tests/handlers/test_password_providers.py @@ -727,7 +727,7 @@ class PasswordAuthProviderTests(unittest.HomeserverTestCase): self.called = True on_logged_out = Mock(side_effect=on_logged_out) - self.hs.get_password_auth_provider().on_logged_out_callbacks.append( + self.hs.get_module_api_callbacks().password_auth_provider.on_logged_out_callbacks.append( on_logged_out ) @@ -857,7 +857,9 @@ class PasswordAuthProviderTests(unittest.HomeserverTestCase): ) m = Mock(return_value=make_awaitable(False)) - self.hs.get_password_auth_provider().is_3pid_allowed_callbacks = [m] + self.hs.get_module_api_callbacks().password_auth_provider.is_3pid_allowed_callbacks = [ + m + ] self.register_user(username, "password") tok = self.login(username, "password") @@ -887,7 +889,9 @@ class PasswordAuthProviderTests(unittest.HomeserverTestCase): m.assert_called_once_with("email", "foo@test.com", registration) m = Mock(return_value=make_awaitable(True)) - self.hs.get_password_auth_provider().is_3pid_allowed_callbacks = [m] + self.hs.get_module_api_callbacks().password_auth_provider.is_3pid_allowed_callbacks = [ + m + ] channel = self.make_request( "POST", -- cgit 1.5.1