1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
import aiohttp
from aiohttp import helpers, web
from aiohttp._ws_impl import WSMsgType
async def test_server_ws_async_for(loop, test_server):
closed = helpers.create_future(loop)
async def handler(request):
ws = web.WebSocketResponse()
await ws.prepare(request)
async for msg in ws:
assert msg.type == aiohttp.MsgType.TEXT
s = msg.data
ws.send_str(s + '/answer')
await ws.close()
closed.set_result(1)
return ws
app = web.Application(loop=loop)
app.router.add_route('GET', '/', handler)
server = await test_server(app)
resp = await aiohttp.ws_connect(server.make_url('/'), loop=loop)
items = ['q1', 'q2', 'q3']
for item in items:
resp.send_str(item)
msg = await resp.receive()
assert msg.type == aiohttp.MsgType.TEXT
assert item + '/answer' == msg.data
await resp.close()
await closed
async def test_closed_async_for(loop, test_client):
closed = helpers.create_future(loop)
async def handler(request):
ws = web.WebSocketResponse()
await ws.prepare(request)
messages = []
async for msg in ws:
messages.append(msg)
if 'stop' == msg.data:
ws.send_str('stopping')
await ws.close()
assert 1 == len(messages)
assert messages[0].type == WSMsgType.TEXT
assert messages[0].data == 'stop'
closed.set_result(None)
return ws
app = web.Application(loop=loop)
app.router.add_get('/', handler)
client = await test_client(app)
ws = await client.ws_connect('/')
ws.send_str('stop')
msg = await ws.receive()
assert msg.type == WSMsgType.TEXT
assert msg.data == 'stopping'
await ws.close()
await closed
|
