def test_http_client_default(provide_session, mocker): """AIOConnection is created with expected attributes.""" session = None if provide_session: session = aiohttp.ClientSession() auth_client = mocker.Mock(auth.GAuthClient) auth_client._session = aiohttp.ClientSession() creds = mocker.Mock() auth_client.creds = creds client = http.AIOConnection(auth_client=auth_client, session=session) if provide_session: assert session is client._session assert auth_client._session is not client._session else: assert auth_client._session is client._session assert session is not client._session client._session.close()
def client(mocker, auth_client): session = aiohttp.ClientSession() client = http.AIOConnection(auth_client=auth_client, session=session) yield client session.close()
def _init_http_client(self): return http.AIOConnection(auth_client=self._init_auth())