summaryrefslogtreecommitdiff
path: root/tests/failonerror_test.py
blob: 519aed821cdc1fdd4454cdb1dfec0dd89c7f5c21 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#! /usr/bin/env python
# -*- coding: utf-8 -*-
# vi:ts=4:et

from . import localhost
import pycurl
import unittest

from . import appmanager
from . import util

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

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

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

    # not sure what the actual min is but 7.26 is too old
    # and does not include status text, only the status code
    @util.min_libcurl(7, 38, 0)
    # no longer supported by libcurl: https://github.com/curl/curl/issues/6615
    @util.removed_in_libcurl(7, 75, 0)
    def test_failonerror(self):
        self.curl.setopt(pycurl.URL, 'http://%s:8380/status/403' % localhost)
        sio = util.BytesIO()
        self.curl.setopt(pycurl.WRITEDATA, sio)
        self.curl.setopt(pycurl.FAILONERROR, True)
        #self.curl.setopt(pycurl.VERBOSE, True)
        try:
            self.curl.perform()
        except pycurl.error as e:
            self.assertEqual(pycurl.E_HTTP_RETURNED_ERROR, e.args[0])
            self.assertEqual('The requested URL returned error: 403 Forbidden', e.args[1])
            self.assertEqual(util.u('The requested URL returned error: 403 Forbidden'), self.curl.errstr())
            self.assertEqual(util.b('The requested URL returned error: 403 Forbidden'), self.curl.errstr_raw())
        else:
            self.fail('Should have raised pycurl.error')
    
    @util.only_python2
    # not sure what the actual min is but 7.26 is too old
    # and does not include status text, only the status code
    @util.min_libcurl(7, 38, 0)
    # no longer supported by libcurl: https://github.com/curl/curl/issues/6615
    @util.removed_in_libcurl(7, 75, 0)
    def test_failonerror_status_line_invalid_utf8_python2(self):
        self.curl.setopt(pycurl.URL, 'http://%s:8380/status_invalid_utf8' % localhost)
        sio = util.BytesIO()
        self.curl.setopt(pycurl.WRITEDATA, sio)
        self.curl.setopt(pycurl.FAILONERROR, True)
        #self.curl.setopt(pycurl.VERBOSE, True)
        try:
            self.curl.perform()
        except pycurl.error as e:
            self.assertEqual(pycurl.E_HTTP_RETURNED_ERROR, e.args[0])
            self.assertEqual('The requested URL returned error: 555 \xb3\xd2\xda\xcd\xd7', e.args[1])
            self.assertEqual('The requested URL returned error: 555 \xb3\xd2\xda\xcd\xd7', self.curl.errstr())
            self.assertEqual('The requested URL returned error: 555 \xb3\xd2\xda\xcd\xd7', self.curl.errstr_raw())
        else:
            self.fail('Should have raised pycurl.error')

    @util.only_python3
    # not sure what the actual min is but 7.26 is too old
    # and does not include status text, only the status code
    @util.min_libcurl(7, 38, 0)
    # no longer supported by libcurl: https://github.com/curl/curl/issues/6615
    @util.removed_in_libcurl(7, 75, 0)
    def test_failonerror_status_line_invalid_utf8_python3(self):
        self.curl.setopt(pycurl.URL, 'http://%s:8380/status_invalid_utf8' % localhost)
        sio = util.BytesIO()
        self.curl.setopt(pycurl.WRITEDATA, sio)
        self.curl.setopt(pycurl.FAILONERROR, True)
        #self.curl.setopt(pycurl.VERBOSE, True)
        try:
            self.curl.perform()
        except pycurl.error as e:
            self.assertEqual(pycurl.E_HTTP_RETURNED_ERROR, e.args[0])
            assert e.args[1].startswith('The requested URL returned error: 555 ')
            try:
                self.curl.errstr()
            except UnicodeDecodeError:
                pass
            else:
                self.fail('Should have raised')
            self.assertEqual(util.b('The requested URL returned error: 555 \xb3\xd2\xda\xcd\xd7'), self.curl.errstr_raw())
        else:
            self.fail('Should have raised pycurl.error')