(self, ClientRequestMock)
| 383 | |
| 384 | @unittest.mock.patch('aiohttp.connector.ClientRequest') |
| 385 | def test_https_auth(self, ClientRequestMock): |
| 386 | loop_mock = unittest.mock.Mock() |
| 387 | proxy_req = ClientRequest('GET', 'http://proxy.example.com', |
| 388 | auth=aiohttp.helpers.BasicAuth('user', |
| 389 | 'pass'), |
| 390 | loop=loop_mock) |
| 391 | ClientRequestMock.return_value = proxy_req |
| 392 | |
| 393 | proxy_resp = ClientResponse('get', 'http://proxy.example.com') |
| 394 | proxy_resp._loop = loop_mock |
| 395 | proxy_req.send = send_mock = unittest.mock.Mock() |
| 396 | send_mock.return_value = proxy_resp |
| 397 | proxy_resp.start = start_mock = unittest.mock.Mock() |
| 398 | self._fake_coroutine(start_mock, unittest.mock.Mock(status=200)) |
| 399 | |
| 400 | connector = aiohttp.ProxyConnector( |
| 401 | 'http://proxy.example.com', loop=loop_mock) |
| 402 | |
| 403 | tr, proto = unittest.mock.Mock(), unittest.mock.Mock() |
| 404 | self._fake_coroutine(loop_mock.create_connection, (tr, proto)) |
| 405 | |
| 406 | self.assertIn('AUTHORIZATION', proxy_req.headers) |
| 407 | self.assertNotIn('PROXY-AUTHORIZATION', proxy_req.headers) |
| 408 | |
| 409 | req = ClientRequest('GET', 'https://www.python.org', loop=self.loop) |
| 410 | self.assertNotIn('AUTHORIZATION', req.headers) |
| 411 | self.assertNotIn('PROXY-AUTHORIZATION', req.headers) |
| 412 | self.loop.run_until_complete(connector._create_connection(req)) |
| 413 | |
| 414 | self.assertEqual(req.path, '/') |
| 415 | self.assertNotIn('AUTHORIZATION', req.headers) |
| 416 | self.assertNotIn('PROXY-AUTHORIZATION', req.headers) |
| 417 | self.assertNotIn('AUTHORIZATION', proxy_req.headers) |
| 418 | self.assertIn('PROXY-AUTHORIZATION', proxy_req.headers) |
| 419 | |
| 420 | self.loop.run_until_complete(proxy_req.close()) |
| 421 | proxy_resp.close() |
| 422 | self.loop.run_until_complete(req.close()) |
nothing calls this directly
no test coverage detected