mirror of
https://github.com/emmett-framework/granian.git
synced 2025-07-07 19:35:33 +00:00
c95fd0cba7
* Add lint and format tools and config * Format Python code * Format Rust code * Add lint CI workflow
68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
import json
|
|
import sys
|
|
|
|
import pytest
|
|
import websockets
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform == 'win32', reason='skip on windows')
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize('server', ['asgi', 'rsgi'], indirect=True)
|
|
@pytest.mark.parametrize('threading_mode', ['runtime', 'workers'])
|
|
async def test_messages(server, threading_mode):
|
|
async with server(threading_mode) as port:
|
|
async with websockets.connect(f'ws://localhost:{port}/ws_echo') as ws:
|
|
await ws.send('foo')
|
|
res_text = await ws.recv()
|
|
await ws.send(b'foo')
|
|
res_bytes = await ws.recv()
|
|
|
|
assert res_text == 'foo'
|
|
assert res_bytes == b'foo'
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize('server', ['asgi', 'rsgi'], indirect=True)
|
|
@pytest.mark.parametrize('threading_mode', ['runtime', 'workers'])
|
|
async def test_reject(server, threading_mode):
|
|
async with server(threading_mode) as port:
|
|
with pytest.raises(websockets.InvalidStatusCode) as exc:
|
|
async with websockets.connect(f'ws://localhost:{port}/ws_reject'):
|
|
pass
|
|
|
|
assert exc.value.status_code == 403
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize('threading_mode', ['runtime', 'workers'])
|
|
async def test_asgi_scope(asgi_server, threading_mode):
|
|
async with asgi_server(threading_mode) as port:
|
|
async with websockets.connect(f'ws://localhost:{port}/ws_info?test=true') as ws:
|
|
res = await ws.recv()
|
|
|
|
data = json.loads(res)
|
|
assert data['asgi'] == {'version': '3.0', 'spec_version': '2.3'}
|
|
assert data['type'] == 'websocket'
|
|
assert data['http_version'] == '1.1'
|
|
assert data['scheme'] == 'ws'
|
|
assert data['path'] == '/ws_info'
|
|
assert data['query_string'] == 'test=true'
|
|
assert data['headers']['host'] == f'localhost:{port}'
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize('threading_mode', ['runtime', 'workers'])
|
|
async def test_rsgi_scope(rsgi_server, threading_mode):
|
|
async with rsgi_server(threading_mode) as port:
|
|
async with websockets.connect(f'ws://localhost:{port}/ws_info?test=true') as ws:
|
|
res = await ws.recv()
|
|
|
|
data = json.loads(res)
|
|
assert data['proto'] == 'ws'
|
|
assert data['http_version'] == '1.1'
|
|
assert data['rsgi_version'] == '1.2'
|
|
assert data['scheme'] == 'http'
|
|
assert data['method'] == 'GET'
|
|
assert data['path'] == '/ws_info'
|
|
assert data['query_string'] == 'test=true'
|
|
assert data['headers']['host'] == f'localhost:{port}'
|