def test_make_response_websocket_reject(self):
environ = {
'asgi.send': AsyncMock(),
'HTTP_SEC_WEBSOCKET_VERSION': 'foo',
}
_run(
async_asgi.make_response('401 UNAUTHORIZED', [('foo', 'bar')],
b'payload', environ))
environ['asgi.send'].mock.assert_called_with(
{'type': 'websocket.close'})
def test_make_response(self):
environ = {
'asgi.send': AsyncMock()
}
_run(async_asgi.make_response('202 ACCEPTED', [('foo', 'bar')],
b'payload', environ))
environ['asgi.send'].mock.assert_any_call(
{'type': 'http.response.start', 'status': 202,
'headers': [(b'foo', b'bar')]})
environ['asgi.send'].mock.assert_any_call(
{'type': 'http.response.body', 'body': b'payload'})
def test_make_response_websocket_accept(self):
environ = {
'asgi.send': AsyncMock(),
'HTTP_SEC_WEBSOCKET_VERSION': 'foo',
}
_run(
async_asgi.make_response('200 OK', [('foo', 'bar')], b'payload',
environ))
environ['asgi.send'].mock.assert_called_with({
'type':
'websocket.accept',
'headers': [(b'foo', b'bar')]
})