diff --git a/deluge/tests/test_torrent.py b/deluge/tests/test_torrent.py index 96abb32ac..007bc67b9 100644 --- a/deluge/tests/test_torrent.py +++ b/deluge/tests/test_torrent.py @@ -25,6 +25,11 @@ from deluge.core.rpcserver import RPCServer from deluge.core.torrent import Torrent from deluge.core.torrentmanager import TorrentManager, TorrentState +try: + from unittest.mock import AsyncMock +except ImportError: + from mock import AsyncMock + class TestTorrent(BaseTestCase): def setup_config(self): @@ -340,7 +345,7 @@ class TestTorrent(BaseTestCase): handle = self.session.add_torrent(atp) self.torrent = Torrent(handle, {}) # Ignore TorrentManager method call - TorrentManager.save_resume_data = mock.MagicMock + TorrentManager.save_resume_data = AsyncMock() result = self.torrent.rename_folder('unicode_filenames', 'Горбачёв') assert isinstance(result, defer.DeferredList) diff --git a/requirements-tests.txt b/requirements-tests.txt index 5e8fb20ad..2fd885219 100644 --- a/requirements-tests.txt +++ b/requirements-tests.txt @@ -9,3 +9,4 @@ flake8-isort pep8-naming mccabe pylint +asyncmock; python_version <= '3.7'