Skip to content
Snippets Groups Projects
Unverified Commit 7a7ca8f2 authored by V02460's avatar V02460 Committed by GitHub
Browse files

Use mock from standard library (#11588)

Instead of the backported version.
parent 8ad39438
No related branches found
No related tags found
No related merge requests found
Replace `mock` package by its standard library version.
...@@ -120,9 +120,7 @@ CONDITIONAL_REQUIREMENTS["mypy"] = [ ...@@ -120,9 +120,7 @@ CONDITIONAL_REQUIREMENTS["mypy"] = [
# Tests assume that all optional dependencies are installed. # Tests assume that all optional dependencies are installed.
# #
# parameterized_class decorator was introduced in parameterized 0.7.0 # parameterized_class decorator was introduced in parameterized 0.7.0
# CONDITIONAL_REQUIREMENTS["test"] = ["parameterized>=0.7.0"]
# We use `mock` library as that backports `AsyncMock` to Python 3.6
CONDITIONAL_REQUIREMENTS["test"] = ["parameterized>=0.7.0", "mock>=4.0.0"]
CONDITIONAL_REQUIREMENTS["dev"] = ( CONDITIONAL_REQUIREMENTS["dev"] = (
CONDITIONAL_REQUIREMENTS["lint"] CONDITIONAL_REQUIREMENTS["lint"]
......
...@@ -12,15 +12,14 @@ ...@@ -12,15 +12,14 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
# Use backported mock for AsyncMock support on Python 3.6. from unittest.mock import Mock
from mock import Mock
from twisted.internet.defer import Deferred, ensureDeferred from twisted.internet.defer import Deferred, ensureDeferred
from synapse.storage.background_updates import BackgroundUpdater from synapse.storage.background_updates import BackgroundUpdater
from tests import unittest from tests import unittest
from tests.test_utils import make_awaitable from tests.test_utils import make_awaitable, simple_async_mock
class BackgroundUpdateTestCase(unittest.HomeserverTestCase): class BackgroundUpdateTestCase(unittest.HomeserverTestCase):
...@@ -116,14 +115,14 @@ class BackgroundUpdateControllerTestCase(unittest.HomeserverTestCase): ...@@ -116,14 +115,14 @@ class BackgroundUpdateControllerTestCase(unittest.HomeserverTestCase):
) )
# Mock out the AsyncContextManager # Mock out the AsyncContextManager
self._update_ctx_manager = Mock(spec=["__aenter__", "__aexit__"]) class MockCM:
self._update_ctx_manager.__aenter__ = Mock( __aenter__ = simple_async_mock(return_value=None)
return_value=make_awaitable(None), __aexit__ = simple_async_mock(return_value=None)
)
self._update_ctx_manager.__aexit__ = Mock(return_value=make_awaitable(None)) self._update_ctx_manager = MockCM
# Mock out the `update_handler` callback # Mock out the `update_handler` callback
self._on_update = Mock(return_value=self._update_ctx_manager) self._on_update = Mock(return_value=self._update_ctx_manager())
# Define a default batch size value that's not the same as the internal default # Define a default batch size value that's not the same as the internal default
# value (100). # value (100).
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment