Skip to content

Commit

Permalink
Fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
MatMaul committed Mar 1, 2024
1 parent 3103c8b commit ec6ee57
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions tests/rest/client/test_register.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@
import datetime
import os
from typing import Any, Dict, List, Tuple
from unittest.mock import AsyncMock

import pkg_resources

Expand All @@ -35,6 +36,7 @@
)
from synapse.api.errors import Codes
from synapse.appservice import ApplicationService
from synapse.http.server import HttpServer
from synapse.rest.client import account, account_validity, login, logout, register, sync
from synapse.server import HomeServer
from synapse.storage._base import db_to_json
Expand All @@ -45,11 +47,20 @@
from tests.unittest import override_config


# Let's override the email register servlet to mock send_email, since the smtp server doesn't really exists
def override_email_register_servlet(hs: "HomeServer", http_server: HttpServer) -> None:
if hs.config.worker.worker_app is None and hs.config.email.can_verify_email:
email_register_servlet = register.EmailRegisterRequestTokenRestServlet(hs)
email_register_servlet.already_in_use_mailer.send_email = AsyncMock() # type: ignore[method-assign]
email_register_servlet.register(http_server)


class RegisterRestServletTestCase(unittest.HomeserverTestCase):
servlets = [
login.register_servlets,
register.register_servlets,
synapse.rest.admin.register_servlets,
override_email_register_servlet,
]
url = b"/_matrix/client/r0/register"

Expand Down

0 comments on commit ec6ee57

Please sign in to comment.