From d1772d410235592b482e3b08b1863f6624d9fe6b Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sun, 19 Feb 2023 15:52:21 +0100
Subject: Adding upstream version 2.0.3.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 deluge/tests/test_torrentmanager.py | 120 ++++++++++++++++++++++++++++++++++++
 1 file changed, 120 insertions(+)
 create mode 100644 deluge/tests/test_torrentmanager.py

(limited to 'deluge/tests/test_torrentmanager.py')

diff --git a/deluge/tests/test_torrentmanager.py b/deluge/tests/test_torrentmanager.py
new file mode 100644
index 0000000..bf84f45
--- /dev/null
+++ b/deluge/tests/test_torrentmanager.py
@@ -0,0 +1,120 @@
+# -*- coding: utf-8 -*-
+#
+# This file is part of Deluge and is licensed under GNU General Public License 3.0, or later, with
+# the additional special exception to link portions of this program with the OpenSSL library.
+# See LICENSE for more details.
+#
+
+from __future__ import unicode_literals
+
+import warnings
+from base64 import b64encode
+
+import mock
+import pytest
+from twisted.internet import defer, task
+
+from deluge import component
+from deluge.core.core import Core
+from deluge.core.rpcserver import RPCServer
+from deluge.error import InvalidTorrentError
+
+from . import common
+from .basetest import BaseTestCase
+
+warnings.filterwarnings('ignore', category=RuntimeWarning)
+warnings.resetwarnings()
+
+
+class TorrentmanagerTestCase(BaseTestCase):
+    def set_up(self):
+        common.set_tmp_config_dir()
+        self.rpcserver = RPCServer(listen=False)
+        self.core = Core()
+        self.core.config.config['lsd'] = False
+        self.clock = task.Clock()
+        self.tm = self.core.torrentmanager
+        self.tm.callLater = self.clock.callLater
+        return component.start()
+
+    def tear_down(self):
+        def on_shutdown(result):
+            del self.rpcserver
+            del self.core
+
+        return component.shutdown().addCallback(on_shutdown)
+
+    @defer.inlineCallbacks
+    def test_remove_torrent(self):
+        filename = common.get_test_data_file('test.torrent')
+        with open(filename, 'rb') as _file:
+            filedump = _file.read()
+        torrent_id = yield self.core.add_torrent_file_async(
+            filename, b64encode(filedump), {}
+        )
+        self.assertTrue(self.tm.remove(torrent_id, False))
+
+    def test_prefetch_metadata(self):
+        from deluge._libtorrent import lt
+
+        with open(common.get_test_data_file('test.torrent'), 'rb') as _file:
+            t_info = lt.torrent_info(lt.bdecode(_file.read()))
+        mock_alert = mock.MagicMock()
+        mock_alert.handle.info_hash = mock.MagicMock(
+            return_value='ab570cdd5a17ea1b61e970bb72047de141bce173'
+        )
+        mock_alert.handle.get_torrent_info = mock.MagicMock(return_value=t_info)
+
+        magnet = 'magnet:?xt=urn:btih:ab570cdd5a17ea1b61e970bb72047de141bce173'
+        d = self.tm.prefetch_metadata(magnet, 30)
+        self.tm.on_alert_metadata_received(mock_alert)
+
+        expected = (
+            'ab570cdd5a17ea1b61e970bb72047de141bce173',
+            {
+                b'piece length': 32768,
+                b'sha1': (
+                    b'2\xce\xb6\xa8"\xd7\xf0\xd4\xbf\xdc^K\xba\x1bh'
+                    b'\x9d\xc5\xb7\xac\xdd'
+                ),
+                b'name': b'azcvsupdater_2.6.2.jar',
+                b'private': 0,
+                b'pieces': (
+                    b'\xdb\x04B\x05\xc3\'\xdab\xb8su97\xa9u'
+                    b'\xca<w\\\x1ef\xd4\x9b\x16\xa9}\xc0\x9f:\xfd'
+                    b'\x97qv\x83\xa2"\xef\x9d7\x0by!\rl\xe5v\xb7'
+                    b'\x18{\xf7/"P\xe9\x8d\x01D\x9e8\xbd\x16\xe3'
+                    b'\xfb-\x9d\xaa\xbcM\x11\xba\x92\xfc\x13F\xf0'
+                    b'\x1c\x86x+\xc8\xd0S\xa9\x90`\xa1\xe4\x82\xe8'
+                    b'\xfc\x08\xf7\xe3\xe5\xf6\x85\x1c%\xe7%\n\xed'
+                    b'\xc0\x1f\xa1;\x9a\xea\xcf\x90\x0c/F>\xdf\xdagA'
+                    b'\xc42|\xda\x82\xf5\xa6b\xa1\xb8#\x80wI\xd8f'
+                    b'\xf8\xbd\xacW\xab\xc3s\xe0\xbbw\xf2K\xbe\xee'
+                    b'\xa8rG\xe1W\xe8\xb7\xc2i\xf3\xd8\xaf\x9d\xdc'
+                    b'\xd0#\xf4\xc1\x12u\xcd\x0bE?:\xe8\x9c\x1cu'
+                    b'\xabb(oj\r^\xd5\xd5A\x83\x88\x9a\xa1J\x1c?'
+                    b'\xa1\xd6\x8c\x83\x9e&'
+                ),
+                b'length': 307949,
+                b'name.utf-8': b'azcvsupdater_2.6.2.jar',
+                b'ed2k': b'>p\xefl\xfa]\x95K\x1b^\xc2\\;;e\xb7',
+            },
+        )
+        self.assertEqual(expected, self.successResultOf(d))
+
+    def test_prefetch_metadata_timeout(self):
+        magnet = 'magnet:?xt=urn:btih:ab570cdd5a17ea1b61e970bb72047de141bce173'
+        d = self.tm.prefetch_metadata(magnet, 30)
+        self.clock.advance(30)
+        expected = ('ab570cdd5a17ea1b61e970bb72047de141bce173', None)
+        return d.addCallback(self.assertEqual, expected)
+
+    @pytest.mark.todo
+    def test_remove_torrent_false(self):
+        """Test when remove_torrent returns False"""
+        common.todo_test(self)
+
+    def test_remove_invalid_torrent(self):
+        self.assertRaises(
+            InvalidTorrentError, self.tm.remove, 'torrentidthatdoesntexist'
+        )
-- 
cgit v1.2.3