FM / tests

/usr/share/doc/python3-pycurl/tests

Name	Size	Mode	Actions
certs/	-	0755	rm
ext/	-	0755	rm
fake-curl/	-	0755	rm
fixtures/	-	0755	rm
matrix/	-	0755	rm
tmp/	-	0755	rm
__pycache__/	-	0755	rm
app.py	4088	0644	edit dl rm
appmanager.py	1380	0644	edit dl rm
cadata_test.py	1452	0644	edit dl rm
certinfo_test.py	3468	0644	edit dl rm
close_socket_cb_test.py	2168	0644	edit dl rm
curl_object_test.py	4883	0644	edit dl rm
debug_test.py	2275	0644	edit dl rm
default_write_cb_test.py	3039	0644	edit dl rm
error_constants_test.py	662	0644	edit dl rm
error_test.py	2901	0644	edit dl rm
failonerror_test.py	3509	0644	edit dl rm
ftp_test.py	1537	0644	edit dl rm
global_init_test.py	1063	0644	edit dl rm
header_cb_test.py	1560	0644	edit dl rm
header_test.py	1848	0644	edit dl rm
high_level_curl_test.py	632	0644	edit dl rm
info_constants_test.py	387	0644	edit dl rm
internals_test.py	6674	0644	edit dl rm
matrix.py	6816	0644	edit dl rm
multi_option_constants_test.py	2503	0644	edit dl rm
multi_socket_test.py	3049	0644	edit dl rm
multi_test.py	12313	0644	edit dl rm
multi_timer_test.py	2695	0644	edit dl rm
open_socket_cb_test.py	4761	0644	edit dl rm
option_constants_test.py	16371	0644	edit dl rm
perform_test.py	2179	0644	edit dl rm
procmgr.py	2731	0644	edit dl rm
protocol_constants_test.py	1679	0644	edit dl rm
readdata_test.py	9623	0644	edit dl rm
read_cb_test.py	4340	0644	edit dl rm
relative_url_test.py	551	0644	edit dl rm
reload_test.py	430	0644	edit dl rm
reset_test.py	2699	0644	edit dl rm
resolve_test.py	780	0644	edit dl rm
run-quickstart.sh	768	0755	edit dl rm
run.sh	690	0755	edit dl rm
runwsgi.py	3733	0644	edit dl rm
seek_cb_constants_test.py	833	0644	edit dl rm
setopt_lifecycle_test.py	1797	0644	edit dl rm
setopt_string_test.py	816	0644	edit dl rm
setopt_test.py	3773	0644	edit dl rm
setopt_unicode_test.py	1092	0644	edit dl rm
setup_test.py	10035	0644	edit dl rm
share_test.py	1726	0644	edit dl rm
sockopt_cb_test.py	2453	0644	edit dl rm
unset_range_test.py	1482	0644	edit dl rm
user_agent_string_test.py	820	0644	edit dl rm
util.py	8291	0644	edit dl rm
version_comparison_test.py	514	0644	edit dl rm
version_constants_test.py	2056	0644	edit dl rm
version_test.py	313	0644	edit dl rm
vsftpd.conf	294	0644	edit dl rm
weakref_test.py	443	0644	edit dl rm
write_abort_test.py	1194	0644	edit dl rm
write_cb_bogus_test.py	1381	0644	edit dl rm
write_test.py	10048	0644	edit dl rm
write_to_stringio_test.py	1228	0644	edit dl rm
xferinfo_cb_test.py	2147	0644	edit dl rm
__init__.py	535	0644	edit dl rm

Edit: /usr/share/doc/python3-pycurl/tests/readdata_test.py (9623B)

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# vi:ts=4:et

from . import localhost
import pycurl
try:
    import unittest2 as unittest
except ImportError:
    import unittest
import sys
import os.path
try:
    import json
except ImportError:
    import simplejson as json
try:
    import urllib.parse as urllib_parse
except ImportError:
    import urllib as urllib_parse

from . import appmanager
from . import util

setup_module, teardown_module = appmanager.setup(('app', 8380))

POSTFIELDS = {
    'field1':'value1',
    'field2':'value2 with blanks',
    'field3':'value3',
}
POSTSTRING = urllib_parse.urlencode(POSTFIELDS)

class DataProvider(object):
    def __init__(self, data):
        self.data = data
        self.finished = False

def read(self, size):
        assert len(self.data) <= size
        if not self.finished:
            self.finished = True
            return self.data
        else:
            # Nothing more to read
            return ""

FORM_SUBMISSION_PATH = os.path.join(os.path.dirname(__file__), 'fixtures', 'form_submission.txt')

class ReaddataTest(unittest.TestCase):
    def setUp(self):
        self.curl = util.DefaultCurl()

def tearDown(self):
        self.curl.close()

def test_readdata_object(self):
        d = DataProvider(POSTSTRING)
        self.curl.setopt(self.curl.URL, 'http://%s:8380/postfields' % localhost)
        self.curl.setopt(self.curl.POST, 1)
        self.curl.setopt(self.curl.POSTFIELDSIZE, len(POSTSTRING))
        self.curl.setopt(self.curl.READDATA, d)
        sio = util.BytesIO()
        self.curl.setopt(pycurl.WRITEFUNCTION, sio.write)
        self.curl.perform()

actual = json.loads(sio.getvalue().decode())
        self.assertEqual(POSTFIELDS, actual)

def test_post_with_read_returning_bytes(self):
        self.check_bytes('world')

def test_post_with_read_returning_bytes_with_nulls(self):
        self.check_bytes("wor\0ld")

def test_post_with_read_returning_bytes_with_multibyte(self):
        self.check_bytes(util.u("Пушкин"))

def check_bytes(self, poststring):
        data = poststring.encode('utf8')
        assert type(data) == util.binary_type
        d = DataProvider(data)

self.curl.setopt(self.curl.URL, 'http://%s:8380/raw_utf8' % localhost)
        self.curl.setopt(self.curl.POST, 1)
        self.curl.setopt(self.curl.HTTPHEADER, ['Content-Type: application/octet-stream'])
        # length of bytes
        self.curl.setopt(self.curl.POSTFIELDSIZE, len(data))
        self.curl.setopt(self.curl.READDATA, d)
        sio = util.BytesIO()
        self.curl.setopt(pycurl.WRITEFUNCTION, sio.write)
        self.curl.perform()

# json should be ascii
        actual = json.loads(sio.getvalue().decode('ascii'))
        self.assertEqual(poststring, actual)

def test_post_with_read_callback_returning_unicode(self):
        self.check_unicode(util.u('world'))

def test_post_with_read_callback_returning_unicode_with_nulls(self):
        self.check_unicode(util.u("wor\0ld"))

def test_post_with_read_callback_returning_unicode_with_multibyte(self):
        try:
            self.check_unicode(util.u("Пушкин"))
            # prints:
            # UnicodeEncodeError: 'ascii' codec can't encode characters in position 6-11: ordinal not in range(128)
        except pycurl.error:
            err, msg = sys.exc_info()[1].args
            # we expect pycurl.E_WRITE_ERROR as the response
            self.assertEqual(pycurl.E_ABORTED_BY_CALLBACK, err)
            self.assertEqual('operation aborted by callback', msg)

def check_unicode(self, poststring):
        assert type(poststring) == util.text_type
        d = DataProvider(poststring)

self.curl.setopt(self.curl.URL, 'http://%s:8380/raw_utf8' % localhost)
        self.curl.setopt(self.curl.POST, 1)
        self.curl.setopt(self.curl.HTTPHEADER, ['Content-Type: application/octet-stream'])
        self.curl.setopt(self.curl.POSTFIELDSIZE, len(poststring))
        self.curl.setopt(self.curl.READDATA, d)
        sio = util.BytesIO()
        self.curl.setopt(pycurl.WRITEFUNCTION, sio.write)
        self.curl.perform()

# json should be ascii
        actual = json.loads(sio.getvalue().decode('ascii'))
        self.assertEqual(poststring, actual)

def test_readdata_file_binary(self):
        # file opened in binary mode
        f = open(FORM_SUBMISSION_PATH, 'rb')
        try:
            self.curl.setopt(self.curl.URL, 'http://%s:8380/postfields' % localhost)
            self.curl.setopt(self.curl.POST, 1)
            self.curl.setopt(self.curl.POSTFIELDSIZE, os.stat(FORM_SUBMISSION_PATH).st_size)
            self.curl.setopt(self.curl.READDATA, f)
            sio = util.BytesIO()
            self.curl.setopt(pycurl.WRITEDATA, sio)
            self.curl.perform()

actual = json.loads(sio.getvalue().decode())
            self.assertEqual({'foo': 'bar'}, actual)
        finally:
            f.close()

def test_readdata_file_text(self):
        # file opened in text mode
        f = open(FORM_SUBMISSION_PATH, 'rt')
        try:
            self.curl.setopt(self.curl.URL, 'http://%s:8380/postfields' % localhost)
            self.curl.setopt(self.curl.POST, 1)
            self.curl.setopt(self.curl.POSTFIELDSIZE, os.stat(FORM_SUBMISSION_PATH).st_size)
            self.curl.setopt(self.curl.READDATA, f)
            sio = util.BytesIO()
            self.curl.setopt(pycurl.WRITEDATA, sio)
            self.curl.perform()

actual = json.loads(sio.getvalue().decode())
            self.assertEqual({'foo': 'bar'}, actual)
        finally:
            f.close()

def test_readdata_file_like(self):
        data = 'hello=world'
        data_provider = DataProvider(data)
        self.curl.setopt(self.curl.URL, 'http://%s:8380/postfields' % localhost)
        self.curl.setopt(self.curl.POST, 1)
        self.curl.setopt(self.curl.POSTFIELDSIZE, len(data))
        self.curl.setopt(self.curl.READDATA, data_provider)
        sio = util.BytesIO()
        self.curl.setopt(pycurl.WRITEDATA, sio)
        self.curl.perform()

actual = json.loads(sio.getvalue().decode())
        self.assertEqual({'hello': 'world'}, actual)

def test_readdata_and_readfunction_file_like(self):
        data = 'hello=world'
        data_provider = DataProvider(data)
        # data must be the same length
        function_provider = DataProvider('aaaaa=bbbbb')
        self.curl.setopt(self.curl.URL, 'http://%s:8380/postfields' % localhost)
        self.curl.setopt(self.curl.POST, 1)
        self.curl.setopt(self.curl.POSTFIELDSIZE, len(data))
        self.curl.setopt(self.curl.READDATA, data_provider)
        self.curl.setopt(self.curl.READFUNCTION, function_provider.read)
        sio = util.BytesIO()
        self.curl.setopt(pycurl.WRITEDATA, sio)
        self.curl.perform()

actual = json.loads(sio.getvalue().decode())
        self.assertEqual({'aaaaa': 'bbbbb'}, actual)

def test_readfunction_and_readdata_file_like(self):
        data = 'hello=world'
        data_provider = DataProvider(data)
        # data must be the same length
        function_provider = DataProvider('aaaaa=bbbbb')
        self.curl.setopt(self.curl.URL, 'http://%s:8380/postfields' % localhost)
        self.curl.setopt(self.curl.POST, 1)
        self.curl.setopt(self.curl.POSTFIELDSIZE, len(data))
        self.curl.setopt(self.curl.READFUNCTION, function_provider.read)
        self.curl.setopt(self.curl.READDATA, data_provider)
        sio = util.BytesIO()
        self.curl.setopt(pycurl.WRITEDATA, sio)
        self.curl.perform()

actual = json.loads(sio.getvalue().decode())
        self.assertEqual({'hello': 'world'}, actual)

def test_readdata_and_readfunction_real_file(self):
        # data must be the same length
        with open(FORM_SUBMISSION_PATH) as f:
            function_provider = DataProvider('aaa=bbb')
            self.curl.setopt(self.curl.URL, 'http://%s:8380/postfields' % localhost)
            self.curl.setopt(self.curl.POST, 1)
            self.curl.setopt(self.curl.POSTFIELDSIZE, os.stat(FORM_SUBMISSION_PATH).st_size)
            self.curl.setopt(self.curl.READDATA, f)
            self.curl.setopt(self.curl.READFUNCTION, function_provider.read)
            sio = util.BytesIO()
            self.curl.setopt(pycurl.WRITEDATA, sio)
            self.curl.perform()

actual = json.loads(sio.getvalue().decode())
            self.assertEqual({'aaa': 'bbb'}, actual)

def test_readfunction_and_readdata_real_file(self):
        # data must be the same length
        with open(FORM_SUBMISSION_PATH) as f:
            function_provider = DataProvider('aaa=bbb')
            self.curl.setopt(self.curl.URL, 'http://%s:8380/postfields' % localhost)
            self.curl.setopt(self.curl.POST, 1)
            self.curl.setopt(self.curl.POSTFIELDSIZE, os.stat(FORM_SUBMISSION_PATH).st_size)
            self.curl.setopt(self.curl.READFUNCTION, function_provider.read)
            self.curl.setopt(self.curl.READDATA, f)
            sio = util.BytesIO()
            self.curl.setopt(pycurl.WRITEDATA, sio)
            self.curl.perform()

actual = json.loads(sio.getvalue().decode())
            self.assertEqual({'foo': 'bar'}, actual)

def test_readdata_not_file_like(self):
        not_file_like = object()
        try:
            self.curl.setopt(self.curl.READDATA, not_file_like)
        except TypeError as exc:
            self.assertIn('object given without a read method', str(exc))
        else:
            self.fail('TypeError not raised')

cmd: