(self, ClientRequestMock)
| 94 | |
| 95 | @unittest.mock.patch('aiohttp.connector.ClientRequest') |
| 96 | def test_auth(self, ClientRequestMock): |
| 97 | proxy_req = ClientRequest('GET', 'http://proxy.example.com', |
| 98 | auth=aiohttp.helpers.BasicAuth('user', |
| 99 | 'pass'), |
| 100 | loop=self.loop) |
| 101 | ClientRequestMock.return_value = proxy_req |
| 102 | self.assertIn('AUTHORIZATION', proxy_req.headers) |
| 103 | self.assertNotIn('PROXY-AUTHORIZATION', proxy_req.headers) |
| 104 | |
| 105 | loop_mock = unittest.mock.Mock() |
| 106 | connector = aiohttp.ProxyConnector( |
| 107 | 'http://proxy.example.com', loop=loop_mock, |
| 108 | proxy_auth=aiohttp.helpers.BasicAuth('user', 'pass')) |
| 109 | connector._resolve_host = resolve_mock = unittest.mock.Mock() |
| 110 | self._fake_coroutine(resolve_mock, [unittest.mock.MagicMock()]) |
| 111 | |
| 112 | tr, proto = unittest.mock.Mock(), unittest.mock.Mock() |
| 113 | self._fake_coroutine(loop_mock.create_connection, (tr, proto)) |
| 114 | |
| 115 | req = ClientRequest('GET', 'http://www.python.org', loop=self.loop) |
| 116 | self.assertNotIn('AUTHORIZATION', req.headers) |
| 117 | self.assertNotIn('PROXY-AUTHORIZATION', req.headers) |
| 118 | conn = self.loop.run_until_complete(connector.connect(req)) |
| 119 | |
| 120 | self.assertEqual(req.path, 'http://www.python.org/') |
| 121 | self.assertNotIn('AUTHORIZATION', req.headers) |
| 122 | self.assertIn('PROXY-AUTHORIZATION', req.headers) |
| 123 | self.assertNotIn('AUTHORIZATION', proxy_req.headers) |
| 124 | self.assertNotIn('PROXY-AUTHORIZATION', proxy_req.headers) |
| 125 | |
| 126 | ClientRequestMock.assert_called_with( |
| 127 | 'GET', 'http://proxy.example.com', |
| 128 | auth=aiohttp.helpers.BasicAuth('user', 'pass'), |
| 129 | loop=unittest.mock.ANY, headers=unittest.mock.ANY) |
| 130 | conn.close() |
| 131 | |
| 132 | def test_auth_utf8(self): |
| 133 | proxy_req = ClientRequest( |
nothing calls this directly
no test coverage detected