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_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_is_not_proxied_when_no_config(self):
self.config = {}
conn = ProxyAwareHTTPSConnection(self.config, 'example.com')
self.assertFalse(conn.proxied)
self.assertEqual(conn.host, 'example.com')
def test_is_proxied(self):
conn = ProxyAwareHTTPSConnection(self.config, 'example.com')
self.assertTrue(conn.proxied)
self.assertEqual(conn.host, 'host')