def test_parser_should_raise_for_frames_with_invalid_content_length():
encoded_frame = b"CONNECT\nabc:def\ncontent-length:3\n\n6chars\x00"
parser = Parser()
with pytest.raises(RuntimeError) as err:
parser.parse(encoded_frame)
assert "Frame doesn't end with NULL byte" in str(err.value)
def test_parsing_simple_frame(command):
parser = Parser()
parser.parse(Frame(command).encode())
parsed_frame = parser.pop_frame()
assert parsed_frame.command == command
assert parsed_frame.headers == {}
assert parsed_frame.body == b""
def test_parsing_frame_with_headers(headers):
parser = Parser()
frame = Frame(Command.CONNECT, headers)
parser.parse(frame.encode())
parsed_frame = parser.pop_frame()
assert parsed_frame.command == Command.CONNECT
assert parsed_frame.headers == headers
assert parsed_frame.body == b""
def test_parser_should_handle_frames_with_no_content_length():
encoded_frame = b"CONNECT\nabc:def\n\nzorro\x00"
parser = Parser()
parser.parse(encoded_frame)
parsed_frame = parser.pop_frame()
assert parsed_frame.command == Command.CONNECT
assert parsed_frame.headers == {"abc": "def"}
assert parsed_frame.body == b"zorro"
def test_parser_should_skip_heartbeat_frames():
parser = Parser()
heartbeats = b"\n\n\n\n\n"
encoded_frame = Frame(Command.CONNECT).encode()
parser.parse(heartbeats + encoded_frame)
assert parser.pending == 1
decoded_frame = parser.pop_frame()
assert decoded_frame is not None
assert decoded_frame.command == Command.CONNECT
def test_parsing_frame_with_headers_and_body(body):
parser = Parser()
frame = Frame(Command.CONNECT, {"abc": "def"}, body)
parser.parse(frame.encode())
parsed_frame = parser.pop_frame()
assert parsed_frame.command == Command.CONNECT
assert "abc" in parsed_frame.headers
assert parsed_frame.headers["abc"] == "def"
assert "content-length" in parsed_frame.headers
assert int(parsed_frame.headers["content-length"]) == len(body)
assert parsed_frame.body == body
def test_parser_should_accept_frames_with_crlf_eols():
parser = Parser()
frame = Frame(Command.CONNECT, {"abc": "def"}, b"zorro")
encoded_frame = frame.encode().replace(b"\n", b"\r\n")
parser.parse(encoded_frame)
parsed_frame = parser.pop_frame()
assert parsed_frame.command == Command.CONNECT
assert "abc" in parsed_frame.headers
assert parsed_frame.headers["abc"] == "def"
assert "content-length" in parsed_frame.headers
assert int(parsed_frame.headers["content-length"]) == 5
assert parsed_frame.body == b"zorro"
def test_parsing_multiple_frames_with_headers_and_body():
parser = Parser()
frame = Frame(Command.CONNECT, {"abc": "def"}, b"zorro")
parser.parse(frame.encode() * 2)
for _ in range(2):
parsed_frame = parser.pop_frame()
assert parsed_frame.command == Command.CONNECT
assert "abc" in parsed_frame.headers
assert parsed_frame.headers["abc"] == "def"
assert "content-length" in parsed_frame.headers
assert int(parsed_frame.headers["content-length"]) == len(b"zorro")
assert parsed_frame.body == b"zorro"
def test_parser_should_wait_until_frame_is_fully_transfered(encoded_frame):
parser = Parser()
# When iterating over bytes in py3 you get ints, not byte slices,
# so we need to use this quirky way of obtaining single-byte slices
single_bytes = [encoded_frame[i:i + 1] for i in range(len(encoded_frame))]
for byte in single_bytes[:-1]:
parser.parse(byte)
assert parser.pending == 0
assert parser.pop_frame() is None
parser.parse(single_bytes[-1])
assert parser.pending == 1
frame = parser.pop_frame()
assert frame is not None
assert frame.command == Command.CONNECT
assert frame.body == b"zorro"
def test_empty_parser():
parser = Parser()
assert parser.pending == 0
assert parser.pop_frame() is None