summaryrefslogtreecommitdiff
path: root/test/test_queue_monkeypatch.py
blob: 8ead9c05cc04adeaf9cf9c3dda56922071b2d04d (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
from __future__ import absolute_import

import mock

import pytest

from urllib3 import HTTPConnectionPool
from urllib3.exceptions import EmptyPoolError
from urllib3.packages.six.moves import queue


class BadError(Exception):
    """
    This should not be raised.
    """
    pass


class TestMonkeypatchResistance(object):
    """
    Test that connection pool works even with a monkey patched Queue module,
    see obspy/obspy#1599, kennethreitz/requests#3742, shazow/urllib3#1061.
    """
    def test_queue_monkeypatching(self):
        with mock.patch.object(queue, 'Empty', BadError):
            with HTTPConnectionPool(host="localhost", block=True) as http:
                http._get_conn(timeout=1)
                with pytest.raises(EmptyPoolError):
                    http._get_conn(timeout=1)