async def test_regexp_example(self):
with aioresponses() as mocked:
credentials = mock.Mock(wraps=CredentialsStub())
mocked.get("http://example.com", status=500)
mocked.get("http://example.com", status=200)
session1 = aiohttp_requests.AuthorizedSession(credentials)
resp1 = await session1.request("GET", "http://example.com")
session2 = aiohttp_requests.AuthorizedSession(credentials)
resp2 = await session2.request("GET", "http://example.com")
assert resp1.status == 500
assert resp2.status == 200
await session1.close()
await session2.close()
def test_constructor_with_auth_request(self):
http = mock.create_autospec(aiohttp.ClientSession)
auth_request = aiohttp_requests.Request(http)
authed_session = aiohttp_requests.AuthorizedSession(
mock.sentinel.credentials, auth_request=auth_request)
assert authed_session._auth_request == auth_request
async def test_ctx(self):
with aioresponses() as mocked:
credentials = mock.Mock(wraps=CredentialsStub())
mocked.get("http://test.example.com", payload=dict(foo="bar"))
session = aiohttp_requests.AuthorizedSession(credentials)
resp = await session.request("GET", "http://test.example.com")
data = await resp.json()
assert dict(foo="bar") == data
await session.close()
async def test_request_refresh(self):
credentials = mock.Mock(wraps=CredentialsStub())
with aioresponses() as mocked:
mocked.get("http://example.com", status=401)
mocked.get("http://example.com", status=200)
authed_session = aiohttp_requests.AuthorizedSession(credentials)
response = await authed_session.request("GET",
"http://example.com")
assert credentials.refresh.called
assert response.status == 200
await authed_session.close()
async def test_http_headers(self):
with aioresponses() as mocked:
credentials = mock.Mock(wraps=CredentialsStub())
mocked.post(
"http://example.com",
payload=dict(),
headers=dict(connection="keep-alive"),
)
session = aiohttp_requests.AuthorizedSession(credentials)
resp = await session.request("POST", "http://example.com")
assert resp.headers["Connection"] == "keep-alive"
await session.close()
async def test_request(self):
with aioresponses() as mocked:
credentials = mock.Mock(wraps=CredentialsStub())
mocked.get(self.TEST_URL, status=200, body="test")
session = aiohttp_requests.AuthorizedSession(credentials)
resp = await session.request(
"GET",
"http://example.com/",
headers={
"Keep-Alive": "timeout=5, max=1000",
"fake": b"bytes"
},
)
assert resp.status == 200
assert "test" == await resp.text()
await session.close()
def _get_authorized_transport():
credentials, project_id = default_async(scopes=(utils.GCS_RW_SCOPE, ))
return tr_requests.AuthorizedSession(credentials)
def test_constructor(self):
authed_session = aiohttp_requests.AuthorizedSession(
mock.sentinel.credentials)
assert authed_session.credentials == mock.sentinel.credentials
