481f779349
* Continuation of updating code to Python 3 with Python 2 fallback. * Using io.open allows files to be encoded and decoded automatically on write and read. This maintains the python boundaries of unicode in code and bytes for output/files so less explicit encoding or decoding. * io.StringIO is the replacement for StringIO and will only accept unicode strings. * io.BytesIO is used where bytes output is required by the enclosing method. * Update bencode for full compatibility.
58 lines
2.1 KiB
Python
58 lines
2.1 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
|
|
|
|
import json as json_lib
|
|
from io import BytesIO
|
|
|
|
import twisted.web.client
|
|
from twisted.internet import defer, reactor
|
|
from twisted.trial.unittest import SkipTest
|
|
from twisted.web.client import Agent, FileBodyProducer
|
|
from twisted.web.http_headers import Headers
|
|
|
|
from deluge.common import utf8_encode_structure
|
|
|
|
from . import common
|
|
from .common import get_test_data_file
|
|
from .common_web import WebServerMockBase, WebServerTestBase
|
|
|
|
common.disable_new_release_check()
|
|
|
|
|
|
class WebServerTestCase(WebServerTestBase, WebServerMockBase):
|
|
|
|
@defer.inlineCallbacks
|
|
def test_get_torrent_info(self):
|
|
|
|
agent = Agent(reactor)
|
|
|
|
self.mock_authentication_ignore(self.deluge_web.auth)
|
|
self.mock_compress_body()
|
|
|
|
# This torrent file contains an uncommon field 'filehash' which must be hex
|
|
# encoded to allow dumping the torrent info to json. Otherwise it will fail with:
|
|
# UnicodeDecodeError: 'utf8' codec can't decode byte 0xe5 in position 0: invalid continuation byte
|
|
filename = get_test_data_file('filehash_field.torrent')
|
|
input_file = '{"params": ["%s"], "method": "web.get_torrent_info", "id": 22}' % filename
|
|
headers = {'User-Agent': ['Twisted Web Client Example'],
|
|
'Content-Type': ['application/json']}
|
|
url = 'http://127.0.0.1:%s/json' % self.webserver_listen_port
|
|
|
|
d = yield agent.request(b'POST', url.encode('utf-8'), Headers(utf8_encode_structure(headers)),
|
|
FileBodyProducer(BytesIO(input_file.encode('utf-8'))))
|
|
try:
|
|
body = yield twisted.web.client.readBody(d)
|
|
except AttributeError:
|
|
raise SkipTest('This test requires "t.w.c.readBody()" in Twisted version >= 13.2')
|
|
|
|
json = json_lib.loads(body)
|
|
self.assertEqual(None, json['error'])
|