mirror of
https://git.anonymousland.org/anonymousland/synapse.git
synced 2025-05-05 10:34:57 -04:00
Prefer make_awaitable
over defer.succeed
in tests (#12505)
When configuring the return values of mocks, prefer awaitables from `make_awaitable` over `defer.succeed`. `Deferred`s are only awaitable once, so it is inappropriate for a mock to return the same `Deferred` multiple times. Also update `run_in_background` to support functions that return arbitrary awaitables. Signed-off-by: Sean Quah <seanq@element.io>
This commit is contained in:
parent
5ef673de4f
commit
78b99de7c2
14 changed files with 72 additions and 69 deletions
|
@ -15,7 +15,6 @@ from typing import Tuple
|
|||
from unittest.mock import Mock, patch
|
||||
from urllib.parse import quote
|
||||
|
||||
from twisted.internet import defer
|
||||
from twisted.test.proto_helpers import MemoryReactor
|
||||
|
||||
import synapse.rest.admin
|
||||
|
@ -30,6 +29,7 @@ from synapse.util import Clock
|
|||
|
||||
from tests import unittest
|
||||
from tests.storage.test_user_directory import GetUserDirectoryTables
|
||||
from tests.test_utils import make_awaitable
|
||||
from tests.test_utils.event_injection import inject_member_event
|
||||
from tests.unittest import override_config
|
||||
|
||||
|
@ -439,7 +439,7 @@ class UserDirectoryTestCase(unittest.HomeserverTestCase):
|
|||
)
|
||||
)
|
||||
|
||||
mock_remove_from_user_dir = Mock(return_value=defer.succeed(None))
|
||||
mock_remove_from_user_dir = Mock(return_value=make_awaitable(None))
|
||||
with patch.object(
|
||||
self.store, "remove_from_user_dir", mock_remove_from_user_dir
|
||||
):
|
||||
|
@ -454,7 +454,7 @@ class UserDirectoryTestCase(unittest.HomeserverTestCase):
|
|||
self.store.register_user(user_id=r_user_id, password_hash=None)
|
||||
)
|
||||
|
||||
mock_remove_from_user_dir = Mock(return_value=defer.succeed(None))
|
||||
mock_remove_from_user_dir = Mock(return_value=make_awaitable(None))
|
||||
with patch.object(
|
||||
self.store, "remove_from_user_dir", mock_remove_from_user_dir
|
||||
):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue