33 lines
794 B
Python
33 lines
794 B
Python
|
from .._events import (
|
||
|
|
ConnectionClosed,
|
||
|
|
Data,
|
||
|
|
EndOfMessage,
|
||
|
|
Event,
|
||
|
|
InformationalResponse,
|
||
|
|
Request,
|
||
|
|
Response,
|
||
|
|
)
|
||
|
|
from .helpers import normalize_data_events
|
||
|
|
|
||
|
|
|
||
|
|
def test_normalize_data_events() -> None:
|
||
|
|
assert normalize_data_events(
|
||
|
|
[
|
||
|
|
Data(data=bytearray(b"1")),
|
||
|
|
Data(data=b"2"),
|
||
|
|
Response(status_code=200, headers=[]), # type: ignore[arg-type]
|
||
|
|
Data(data=b"3"),
|
||
|
|
Data(data=b"4"),
|
||
|
|
EndOfMessage(),
|
||
|
|
Data(data=b"5"),
|
||
|
|
Data(data=b"6"),
|
||
|
|
Data(data=b"7"),
|
||
|
|
]
|
||
|
|
) == [
|
||
|
|
Data(data=b"12"),
|
||
|
|
Response(status_code=200, headers=[]), # type: ignore[arg-type]
|
||
|
|
Data(data=b"34"),
|
||
|
|
EndOfMessage(),
|
||
|
|
Data(data=b"567"),
|
||
|
|
]
|