def test_record_contents_err(): with pytest.raises(ValueError, match="Expected TLS record"): next(tls.handshake_record_contents(b"GET /error")) empty_record = bytes.fromhex("1603010000") with pytest.raises(ValueError, match="Record must not be empty"): next(tls.handshake_record_contents(empty_record))
def test_record_contents(): data = bytes.fromhex( "1603010002beef" "1603010001ff" ) assert list(tls.handshake_record_contents(data)) == [ b"\xbe\xef", b"\xff" ] for i in range(6): assert list(tls.handshake_record_contents(data[:i])) == []