mirror of
https://github.com/logos-storage/deluge.git
synced 2026-08-02 16:33:15 +00:00
b1cdc32f73
The move to using auto-formatter makes it easier to read, submit and speeds up development time. https://github.com/ambv/black/ Although I would prefer 79 chars, the default line length of 88 chars used by black suffices. The flake8 line length remains at 120 chars since black does not touch comments or docstrings and this will require another round of fixes. The only black setting that is not standard is the use of double-quotes for strings so disabled any formatting of these. Note however that flake8 will still flag usage of double-quotes. I may change my mind on double vs single quotes but for now leave them. A new pyproject.toml file has been created for black configuration.
200 lines
7.4 KiB
Python
200 lines
7.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright (C) 2016 bendikro <bro.devel+deluge@gmail.com>
|
|
#
|
|
# 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
|
|
|
|
from io import BytesIO
|
|
|
|
from twisted.internet import defer, reactor
|
|
from twisted.python.failure import Failure
|
|
from twisted.web.client import Agent, FileBodyProducer
|
|
from twisted.web.http_headers import Headers
|
|
from twisted.web.static import File
|
|
|
|
import deluge.component as component
|
|
from deluge.ui.client import client
|
|
|
|
from . import common
|
|
from .common_web import WebServerTestBase
|
|
|
|
common.disable_new_release_check()
|
|
|
|
|
|
class WebAPITestCase(WebServerTestBase):
|
|
def test_connect_invalid_host(self):
|
|
d = self.deluge_web.web_api.connect('id')
|
|
d.addCallback(self.fail)
|
|
d.addErrback(self.assertIsInstance, Failure)
|
|
return d
|
|
|
|
def test_connect(self):
|
|
d = self.deluge_web.web_api.connect(self.host_id)
|
|
|
|
def on_connect(result):
|
|
self.assertEqual(type(result), tuple)
|
|
self.assertTrue(len(result) > 0)
|
|
self.addCleanup(client.disconnect)
|
|
return result
|
|
|
|
d.addCallback(on_connect)
|
|
d.addErrback(self.fail)
|
|
return d
|
|
|
|
def test_disconnect(self):
|
|
d = self.deluge_web.web_api.connect(self.host_id)
|
|
|
|
@defer.inlineCallbacks
|
|
def on_connect(result):
|
|
self.assertTrue(self.deluge_web.web_api.connected())
|
|
yield self.deluge_web.web_api.disconnect()
|
|
self.assertFalse(self.deluge_web.web_api.connected())
|
|
|
|
d.addCallback(on_connect)
|
|
d.addErrback(self.fail)
|
|
return d
|
|
|
|
def test_get_config(self):
|
|
config = self.deluge_web.web_api.get_config()
|
|
self.assertEqual(self.webserver_listen_port, config['port'])
|
|
|
|
def test_set_config(self):
|
|
config = self.deluge_web.web_api.get_config()
|
|
config['pwd_salt'] = 'new_salt'
|
|
config['pwd_sha1'] = 'new_sha'
|
|
config['sessions'] = {
|
|
'233f23632af0a74748bc5dd1d8717564748877baa16420e6898e17e8aa365e6e': {
|
|
'login': 'skrot',
|
|
'expires': 1460030877.0,
|
|
'level': 10,
|
|
}
|
|
}
|
|
self.deluge_web.web_api.set_config(config)
|
|
web_config = component.get('DelugeWeb').config.config
|
|
self.assertNotEquals(config['pwd_salt'], web_config['pwd_salt'])
|
|
self.assertNotEquals(config['pwd_sha1'], web_config['pwd_sha1'])
|
|
self.assertNotEquals(config['sessions'], web_config['sessions'])
|
|
|
|
@defer.inlineCallbacks
|
|
def get_host_status(self):
|
|
host = list(self.deluge_web.web_api._get_host(self.host_id))
|
|
host[3] = 'Online'
|
|
host[4] = '2.0.0.dev562'
|
|
status = yield self.deluge_web.web_api.get_host_status(self.host_id)
|
|
self.assertEqual(status, tuple(status))
|
|
|
|
def test_get_host(self):
|
|
self.assertFalse(self.deluge_web.web_api._get_host('invalid_id'))
|
|
conn = list(self.deluge_web.web_api.hostlist.get_hosts_info()[0])
|
|
self.assertEqual(self.deluge_web.web_api._get_host(conn[0]), conn[0:4])
|
|
|
|
def test_add_host(self):
|
|
conn = ['abcdef', '10.0.0.1', 0, 'user123', 'pass123']
|
|
self.assertFalse(self.deluge_web.web_api._get_host(conn[0]))
|
|
# Add valid host
|
|
result, host_id = self.deluge_web.web_api.add_host(
|
|
conn[1], conn[2], conn[3], conn[4]
|
|
)
|
|
self.assertEqual(result, True)
|
|
conn[0] = host_id
|
|
self.assertEqual(self.deluge_web.web_api._get_host(conn[0]), conn[0:4])
|
|
|
|
# Add already existing host
|
|
ret = self.deluge_web.web_api.add_host(conn[1], conn[2], conn[3], conn[4])
|
|
self.assertEqual(ret, (False, 'Host details already in hostlist'))
|
|
|
|
# Add invalid port
|
|
conn[2] = 'bad port'
|
|
ret = self.deluge_web.web_api.add_host(conn[1], conn[2], conn[3], conn[4])
|
|
self.assertEqual(ret, (False, 'Invalid port. Must be an integer'))
|
|
|
|
def test_remove_host(self):
|
|
conn = ['connection_id', '', 0, '', '']
|
|
self.deluge_web.web_api.hostlist.config['hosts'].append(conn)
|
|
self.assertEqual(self.deluge_web.web_api._get_host(conn[0]), conn[0:4])
|
|
# Remove valid host
|
|
self.assertTrue(self.deluge_web.web_api.remove_host(conn[0]))
|
|
self.assertFalse(self.deluge_web.web_api._get_host(conn[0]))
|
|
# Remove non-existing host
|
|
self.assertFalse(self.deluge_web.web_api.remove_host(conn[0]))
|
|
|
|
def test_get_torrent_info(self):
|
|
filename = common.get_test_data_file('test.torrent')
|
|
ret = self.deluge_web.web_api.get_torrent_info(filename)
|
|
self.assertEqual(ret['name'], 'azcvsupdater_2.6.2.jar')
|
|
self.assertEqual(ret['info_hash'], 'ab570cdd5a17ea1b61e970bb72047de141bce173')
|
|
self.assertTrue('files_tree' in ret)
|
|
|
|
def test_get_magnet_info(self):
|
|
ret = self.deluge_web.web_api.get_magnet_info(
|
|
'magnet:?xt=urn:btih:SU5225URMTUEQLDXQWRB2EQWN6KLTYKN'
|
|
)
|
|
self.assertEqual(ret['name'], '953bad769164e8482c7785a21d12166f94b9e14d')
|
|
self.assertEqual(ret['info_hash'], '953bad769164e8482c7785a21d12166f94b9e14d')
|
|
self.assertTrue('files_tree' in ret)
|
|
|
|
@defer.inlineCallbacks
|
|
def test_get_torrent_files(self):
|
|
yield self.deluge_web.web_api.connect(self.host_id)
|
|
filename = common.get_test_data_file('test.torrent')
|
|
torrents = [
|
|
{'path': filename, 'options': {'download_location': '/home/deluge/'}}
|
|
]
|
|
yield self.deluge_web.web_api.add_torrents(torrents)
|
|
ret = yield self.deluge_web.web_api.get_torrent_files(
|
|
'ab570cdd5a17ea1b61e970bb72047de141bce173'
|
|
)
|
|
self.assertEqual(ret['type'], 'dir')
|
|
self.assertEqual(
|
|
ret['contents'],
|
|
{
|
|
'azcvsupdater_2.6.2.jar': {
|
|
'priority': 4,
|
|
'index': 0,
|
|
'offset': 0,
|
|
'progress': 0.0,
|
|
'path': 'azcvsupdater_2.6.2.jar',
|
|
'type': 'file',
|
|
'size': 307949,
|
|
}
|
|
},
|
|
)
|
|
|
|
@defer.inlineCallbacks
|
|
def test_download_torrent_from_url(self):
|
|
filename = 'ubuntu-9.04-desktop-i386.iso.torrent'
|
|
self.deluge_web.top_level.putChild(
|
|
filename.encode(), File(common.get_test_data_file(filename))
|
|
)
|
|
url = 'http://localhost:%d/%s' % (self.webserver_listen_port, filename)
|
|
res = yield self.deluge_web.web_api.download_torrent_from_url(url)
|
|
self.assertTrue(res.endswith(filename))
|
|
|
|
@defer.inlineCallbacks
|
|
def test_invalid_json(self):
|
|
"""
|
|
If json_api._send_response does not return server.NOT_DONE_YET
|
|
this error is thrown when json is invalid:
|
|
exceptions.RuntimeError: Request.write called on a request after Request.finish was called.
|
|
|
|
"""
|
|
agent = Agent(reactor)
|
|
bad_body = b'{ method": "auth.login" }'
|
|
d = yield agent.request(
|
|
b'POST',
|
|
b'http://127.0.0.1:%i/json' % self.webserver_listen_port,
|
|
Headers(
|
|
{
|
|
b'User-Agent': [b'Twisted Web Client Example'],
|
|
b'Content-Type': [b'application/json'],
|
|
}
|
|
),
|
|
FileBodyProducer(BytesIO(bad_body)),
|
|
)
|
|
yield d
|