async def test_wait_closed_before_start() -> None:
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
await ws.close()
def function2625(function271):
var4377 = function271('GET', '/')
var1358 = WebSocketResponse()
yield from var1358.prepare(var4377)
with pytest.raises(TypeError):
var1358.send_str(b'bytes')
def function1324(function271):
var3572 = function271('GET', '/')
var2977 = WebSocketResponse()
yield from var2977.prepare(var3572)
with pytest.raises(TypeError):
var2977.send_json(set())
def test_nonstarted_send_bytes():
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
ws.send_bytes(b'bytes')
def test_nonstarted_close():
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
yield from ws.close()
def test_write_eof_not_started():
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
yield from ws.write_eof()
def test_nonstarted_pong():
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
ws.pong()
def test_send_bytes_nonbytes(make_request):
req = make_request('GET', '/')
ws = WebSocketResponse()
yield from ws.prepare(req)
with pytest.raises(TypeError):
ws.send_bytes('string')
def test_send_json_nonjson(make_request):
req = make_request('GET', '/')
ws = WebSocketResponse()
yield from ws.prepare(req)
with pytest.raises(TypeError):
ws.send_json(set())
async def test_nonstarted_send_json() -> None:
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
await ws.send_json({"type": "json"})
async def test_nonstarted_receive_bytes() -> None:
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
await ws.receive_bytes()
async def test_nonstarted_send_bytes() -> None:
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
await ws.send_bytes(b"bytes")
async def test_no_transfer_encoding_header(make_request, mocker) -> None:
req = make_request("GET", "/")
ws = WebSocketResponse()
await ws._start(req)
assert "Transfer-Encoding" not in ws.headers
async def test_write_eof_not_started() -> None:
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
await ws.write_eof()
def test_start_without_upgrade(make_request):
req = make_request('GET', '/', headers=CIMultiDict({}))
ws = WebSocketResponse()
with pytest.raises(HTTPBadRequest):
yield from ws.prepare(req)
def test_write_non_prepared():
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
ws.write(b'data')
def test_wait_closed_before_start():
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
yield from ws.close()
def test_can_prepare_ok(make_request):
req = make_request('GET', '/', protocols=True)
ws = WebSocketResponse(protocols=('chat', ))
assert (True, 'chat') == ws.can_prepare(req)
def test_can_start_ok(make_request):
req = make_request('GET', '/', protocols=True)
ws = WebSocketResponse(protocols=('chat', ))
with pytest.warns(DeprecationWarning):
assert (True, 'chat') == ws.can_start(req)
def test_can_prepare_unknown_protocol(make_request):
req = make_request('GET', '/')
ws = WebSocketResponse()
assert (True, None) == ws.can_prepare(req)
def test_nonstarted_send_str():
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
ws.send_str('string')
def test_can_prepare_invalid_method(make_request):
req = make_request('POST', '/')
ws = WebSocketResponse()
assert (False, None) == ws.can_prepare(req)
def test_nonstarted_send_json():
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
ws.send_json({'type': 'json'})
def test_can_prepare_without_upgrade(make_request):
req = make_request('GET', '/', headers=CIMultiDict({}))
ws = WebSocketResponse()
assert (False, None) == ws.can_prepare(req)
def test_nonstarted_receive_str():
ws = WebSocketResponse()
with pytest.raises(RuntimeError):
yield from ws.receive_str()
def test_closed_after_ctor():
ws = WebSocketResponse()
assert not ws.closed
assert ws.close_code is None
def function612(function271):
var540 = function271('GET', '/')
var2482 = WebSocketResponse()
yield from var2482.prepare(var540)
with pytest.raises(TypeError):
var2482.send_bytes('string')
def test_start_invalid_method(make_request):
req = make_request('POST', '/')
ws = WebSocketResponse()
with pytest.raises(HTTPMethodNotAllowed):
yield from ws.prepare(req)
def function2812():
var550 = WebSocketResponse()
with pytest.raises(RuntimeError):
var550.write(b'data')
async def test_prepare_without_upgrade(make_request) -> None:
req = make_request("GET", "/", headers=CIMultiDict({}))
ws = WebSocketResponse()
with pytest.raises(HTTPBadRequest):
await ws.prepare(req)
