Beispiel #1
0
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()
Beispiel #2
0
def client(mocker, auth_client):
    session = aiohttp.ClientSession()
    client = http.AIOConnection(auth_client=auth_client, session=session)
    yield client
    session.close()
Beispiel #3
0
 def _init_http_client(self):
     return http.AIOConnection(auth_client=self._init_auth())