async def test_http2_connection_post_request():
origin = Origin(b"https", b"example.com", 443)
stream = AsyncMockStream([
hyperframe.frame.SettingsFrame().serialize(),
hyperframe.frame.HeadersFrame(
stream_id=1,
data=hpack.Encoder().encode([
(b":status", b"200"),
(b"content-type", b"plain/text"),
]),
flags=["END_HEADERS"],
).serialize(),
hyperframe.frame.DataFrame(stream_id=1,
data=b"Hello, world!",
flags=["END_STREAM"]).serialize(),
])
async with AsyncHTTP2Connection(origin=origin, stream=stream) as conn:
response = await conn.request(
"POST",
"https://example.com/",
headers={b"content-length": b"17"},
content=b'{"data": "upload"}',
)
assert response.status == 200
assert response.content == b"Hello, world!"
async def test_http2_connection_attempt_close():
"""
A connection can only be closed when it is idle.
"""
origin = Origin(b"https", b"example.com", 443)
stream = AsyncMockStream([
hyperframe.frame.SettingsFrame().serialize(),
hyperframe.frame.HeadersFrame(
stream_id=1,
data=hpack.Encoder().encode([
(b":status", b"200"),
(b"content-type", b"plain/text"),
]),
flags=["END_HEADERS"],
).serialize(),
hyperframe.frame.DataFrame(stream_id=1,
data=b"Hello, world!",
flags=["END_STREAM"]).serialize(),
])
async with AsyncHTTP2Connection(origin=origin, stream=stream) as conn:
async with conn.stream("GET", "https://example.com/") as response:
await response.aread()
assert response.status == 200
assert response.content == b"Hello, world!"
await conn.aclose()
with pytest.raises(ConnectionNotAvailable):
await conn.request("GET", "https://example.com/")
async def test_http2_connection():
origin = Origin(b"https", b"example.com", 443)
stream = AsyncMockStream([
hyperframe.frame.SettingsFrame().serialize(),
hyperframe.frame.HeadersFrame(
stream_id=1,
data=hpack.Encoder().encode([
(b":status", b"200"),
(b"content-type", b"plain/text"),
]),
flags=["END_HEADERS"],
).serialize(),
hyperframe.frame.DataFrame(stream_id=1,
data=b"Hello, world!",
flags=["END_STREAM"]).serialize(),
])
async with AsyncHTTP2Connection(origin=origin,
stream=stream,
keepalive_expiry=5.0) as conn:
response = await conn.request("GET", "https://example.com/")
assert response.status == 200
assert response.content == b"Hello, world!"
assert conn.is_idle()
assert conn.is_available()
assert not conn.is_closed()
assert not conn.has_expired()
assert (conn.info() ==
"'https://example.com:443', HTTP/2, IDLE, Request Count: 1")
assert (
repr(conn) ==
"<AsyncHTTP2Connection ['https://example.com:443', IDLE, Request Count: 1]>"
)
async def test_http2_request_to_incorrect_origin():
"""
A connection can only send requests to whichever origin it is connected to.
"""
origin = Origin(b"https", b"example.com", 443)
stream = AsyncMockStream([])
async with AsyncHTTP2Connection(origin=origin, stream=stream) as conn:
with pytest.raises(RuntimeError):
await conn.request("GET", "https://other.com/")
async def test_http2_connection_with_remote_protocol_error():
"""
If a remote protocol error occurs, then no response will be returned,
and the connection will not be reusable.
"""
origin = Origin(b"https", b"example.com", 443)
stream = AsyncMockStream([b"Wait, this isn't valid HTTP!", b""])
async with AsyncHTTP2Connection(origin=origin, stream=stream) as conn:
with pytest.raises(RemoteProtocolError):
await conn.request("GET", "https://example.com/")
async def test_http2_connection_with_flow_control():
origin = Origin(b"https", b"example.com", 443)
stream = AsyncMockStream([
hyperframe.frame.SettingsFrame().serialize(),
# Available flow: 65,535
hyperframe.frame.WindowUpdateFrame(
stream_id=0, window_increment=10_000).serialize(),
hyperframe.frame.WindowUpdateFrame(
stream_id=1, window_increment=10_000).serialize(),
# Available flow: 75,535
hyperframe.frame.WindowUpdateFrame(
stream_id=0, window_increment=10_000).serialize(),
hyperframe.frame.WindowUpdateFrame(
stream_id=1, window_increment=10_000).serialize(),
# Available flow: 85,535
hyperframe.frame.WindowUpdateFrame(
stream_id=0, window_increment=10_000).serialize(),
hyperframe.frame.WindowUpdateFrame(
stream_id=1, window_increment=10_000).serialize(),
# Available flow: 95,535
hyperframe.frame.WindowUpdateFrame(
stream_id=0, window_increment=10_000).serialize(),
hyperframe.frame.WindowUpdateFrame(
stream_id=1, window_increment=10_000).serialize(),
# Available flow: 105,535
hyperframe.frame.HeadersFrame(
stream_id=1,
data=hpack.Encoder().encode([
(b":status", b"200"),
(b"content-type", b"plain/text"),
]),
flags=["END_HEADERS"],
).serialize(),
hyperframe.frame.DataFrame(stream_id=1,
data=b"100,000 bytes received",
flags=["END_STREAM"]).serialize(),
])
async with AsyncHTTP2Connection(origin=origin, stream=stream) as conn:
response = await conn.request(
"POST",
"https://example.com/",
content=b"x" * 100_000,
)
assert response.status == 200
assert response.content == b"100,000 bytes received"
async def test_http2_connection_with_rst_stream():
"""
If a stream reset occurs, then no response will be returned,
but the connection will remain reusable for other requests.
"""
origin = Origin(b"https", b"example.com", 443)
stream = AsyncMockStream([
hyperframe.frame.SettingsFrame().serialize(),
hyperframe.frame.HeadersFrame(
stream_id=1,
data=hpack.Encoder().encode([
(b":status", b"200"),
(b"content-type", b"plain/text"),
]),
flags=["END_HEADERS"],
).serialize(),
# Stream is closed midway through the first response...
hyperframe.frame.RstStreamFrame(stream_id=1, error_code=8).serialize(),
# ...Which doesn't prevent the second response.
hyperframe.frame.HeadersFrame(
stream_id=3,
data=hpack.Encoder().encode([
(b":status", b"200"),
(b"content-type", b"plain/text"),
]),
flags=["END_HEADERS"],
).serialize(),
hyperframe.frame.DataFrame(stream_id=3,
data=b"Hello, world!",
flags=["END_STREAM"]).serialize(),
b"",
])
async with AsyncHTTP2Connection(origin=origin, stream=stream) as conn:
with pytest.raises(RemoteProtocolError):
await conn.request("GET", "https://example.com/")
response = await conn.request("GET", "https://example.com/")
assert response.status == 200