def test_request_uses_original_url_when_not_proxied(self, mock_request): self.config = {} conn = ProxyAwareHTTPConnection(self.config, 'example.com') conn._create_connection = Mock() conn.request('GET', '/foobar') mock_request.assert_called_once_with( 'GET', '/foobar', None, headers={} )
def test_request_uses_absolute_url(self, mock_request): conn = ProxyAwareHTTPConnection(self.config, 'example.com') conn._create_connection = Mock() conn.request('GET', '/foobar') mock_request.assert_called_once_with( 'GET', 'http://example.com/foobar', None, headers={'Proxy-Authorization': 'Basic dXNlcm5hbWU6cGFzc3dvcmQ='} )
def test_raises_exception_when_invalid_socks_scheme(self, mock_socks): conf = namedtuple('ProxyConf', 'scheme username password hostport') self.config = { 'http': conf(*_parse_proxy('socks6://socks_user:socks_pass@socks_host:3128')), 'no_proxy': 'localhost,127.0.0.1,dev_server:8080' } conn = ProxyAwareHTTPConnection(self.config, 'example.com') with self.assertRaises(TypeError): conn.connect()
def test_connect_uses_remote_dns(self, mock_socks): conf = namedtuple('ProxyConf', 'scheme username password hostport') self.config = { 'http': conf(*_parse_proxy( 'socks5h://socks_user:socks_pass@socks_host:3128')), 'no_proxy': 'localhost,127.0.0.1,dev_server:8080' } mock_socks.PROXY_TYPE_SOCKS5 = socks.PROXY_TYPE_SOCKS5 conn = ProxyAwareHTTPConnection(self.config, 'example.com') conn.connect() mock_socks.create_connection.assert_called_once_with( ('example.com', 80), socket._GLOBAL_DEFAULT_TIMEOUT, None, socks.PROXY_TYPE_SOCKS5, 'socks_host', 3128, True, 'socks_user', 'socks_pass', ((socket.IPPROTO_TCP, socket.TCP_NODELAY, 1), ))
def test_connect_does_not_use_socks_proxy(self, mock_socks): conn = ProxyAwareHTTPConnection(self.config, 'example.com') conn._create_connection = Mock() conn.connect() assert mock_socks._create_connection.call_count == 0
def test_is_not_proxied_when_no_config(self): self.config = {} conn = ProxyAwareHTTPConnection(self.config, 'example.com') self.assertFalse(conn.use_proxy) self.assertEqual(conn.host, 'example.com')
def test_use_proxy(self): conn = ProxyAwareHTTPConnection(self.config, 'example.com') self.assertTrue(conn.use_proxy) self.assertEqual(conn.host, 'host')
def test_is_not_proxied_when_no_proxy(self): self.config['no_proxy'] += ',example.com' conn = ProxyAwareHTTPConnection(self.config, 'example.com') self.assertFalse(conn.proxied) self.assertEqual(conn.host, 'example.com')