|
1 | 1 | from __future__ import annotations |
2 | 2 |
|
| 3 | +from hyperframe.frame import Frame |
| 4 | +from typing_extensions import assert_type |
| 5 | + |
3 | 6 | from h2.connection import H2Connection |
| 7 | +from h2.events import Event |
4 | 8 | from h2.frame_buffer import FrameBuffer |
5 | 9 |
|
6 | 10 |
|
7 | | -def receive_data_accepts_buffer_types( |
8 | | - connection: H2Connection, |
9 | | - frame_buffer: FrameBuffer, |
10 | | -) -> None: |
| 11 | +def receive_data_accepts_buffer_types() -> None: |
| 12 | + connection = H2Connection() |
| 13 | + frame_buffer = FrameBuffer() |
11 | 14 | bytearray_data = bytearray(b"") |
12 | 15 | memoryview_data = memoryview(b"") |
13 | 16 |
|
14 | | - connection.receive_data(bytearray_data) |
15 | | - connection.receive_data(memoryview_data) |
| 17 | + bytearray_events = connection.receive_data(bytearray_data) |
| 18 | + memoryview_events = connection.receive_data(memoryview_data) |
16 | 19 | frame_buffer.add_data(bytearray_data) |
17 | 20 | frame_buffer.add_data(memoryview_data) |
| 21 | + buffered_frames = list(frame_buffer) |
| 22 | + |
| 23 | + assert_type(bytearray_events, list[Event]) |
| 24 | + assert_type(memoryview_events, list[Event]) |
| 25 | + assert_type(buffered_frames, list[Frame]) |
| 26 | + assert bytearray_events == [] |
| 27 | + assert memoryview_events == [] |
| 28 | + assert buffered_frames == [] |
0 commit comments