blob: 1ed4d711e925dea3cb515399f3ab5f76b270126f (
plain)
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
|
import json
import logging
import urllib.parse
import asyncio
import websockets
logging.basicConfig(level=logging.WARNING)
# Uncomment this line to make only websockets more verbose.
# logging.getLogger('websockets').setLevel(logging.DEBUG)
SERVER = "ws://127.0.0.1:8642"
AGENT = "websockets"
async def get_case_count(server):
uri = f"{server}/getCaseCount"
async with websockets.connect(uri) as ws:
msg = ws.recv()
return json.loads(msg)
async def run_case(server, case, agent):
uri = f"{server}/runCase?case={case}&agent={agent}"
async with websockets.connect(uri, max_size=2 ** 25, max_queue=1) as ws:
async for msg in ws:
await ws.send(msg)
async def update_reports(server, agent):
uri = f"{server}/updateReports?agent={agent}"
async with websockets.connect(uri):
pass
async def run_tests(server, agent):
cases = await get_case_count(server)
for case in range(1, cases + 1):
print(f"Running test case {case} out of {cases}", end="\r")
await run_case(server, case, agent)
print(f"Ran {cases} test cases ")
await update_reports(server, agent)
asyncio.run(run_tests(SERVER, urllib.parse.quote(AGENT)))
|
