def test_connect_does_not_use_socks_proxy(self, mock_set_tunnel, mock_connect, mock_socks): conn = ProxyAwareHTTPSConnection(self.config, 'example.com') conn.connect() assert mock_socks._create_connection.call_count == 0 mock_connect.assert_called_once_with()
def test_raises_exception_when_invalid_socks_scheme(self, mock_socks): conf = namedtuple('ProxyConf', 'scheme username password hostport') self.config = { 'https': conf(*_parse_proxy('socks6://socks_user:socks_pass@socks_host:3128')), 'no_proxy': 'localhost,127.0.0.1,dev_server:8080' } conn = ProxyAwareHTTPSConnection(self.config, 'example.com', context=Mock()) with self.assertRaises(TypeError): conn.connect()
def test_set_tunnel_is_not_called_when_socks(self, mock_set_tunnel): conf = namedtuple('ProxyConf', 'scheme username password hostport') self.config = { 'https': conf(*_parse_proxy('socks5://username:password@host:3128')) } ProxyAwareHTTPSConnection(self.config, 'example.com') self.assertEqual(mock_set_tunnel.call_count, 0)
def test_connect_uses_socks4_proxy(self, mock_socks): conf = namedtuple('ProxyConf', 'scheme username password hostport') self.config = { 'https': conf(*_parse_proxy( 'socks4://socks_user:socks_pass@socks_host:3128')), 'no_proxy': 'localhost,127.0.0.1,dev_server:8080' } mock_socks.PROXY_TYPE_SOCKS4 = socks.PROXY_TYPE_SOCKS4 conn = ProxyAwareHTTPSConnection(self.config, 'example.com', context=Mock()) conn.connect() mock_socks.create_connection.assert_called_once_with( ('example.com', 443), socket._GLOBAL_DEFAULT_TIMEOUT, None, socks.PROXY_TYPE_SOCKS4, 'socks_host', 3128, False, 'socks_user', 'socks_pass', ((socket.IPPROTO_TCP, socket.TCP_NODELAY, 1), ))
def test_set_tunnel_is_not_called(self, mock_set_tunnel): self.config = {} ProxyAwareHTTPSConnection(self.config, 'example.com') self.assertEqual(mock_set_tunnel.call_count, 0)
def test_is_not_proxied_when_no_config(self): self.config = {} conn = ProxyAwareHTTPSConnection(self.config, 'example.com') self.assertFalse(conn.use_proxy) self.assertEqual(conn.host, 'example.com')
def test_set_tunnel_is_called(self, mock_set_tunnel): ProxyAwareHTTPSConnection(self.config, 'example.com') mock_set_tunnel.assert_called_once_with( 'example.com', headers={'Proxy-Authorization': 'Basic dXNlcm5hbWU6cGFzc3dvcmQ='})
def test_use_proxy(self): conn = ProxyAwareHTTPSConnection(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 = ProxyAwareHTTPSConnection(self.config, 'example.com') self.assertFalse(conn.proxied) self.assertEqual(conn.host, 'example.com')