48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
#!/usr/bin/env python
|
|
|
|
import websocket
|
|
import json
|
|
import traceback
|
|
import six
|
|
|
|
|
|
|
|
|
|
|
|
SERVER = 'ws://127.0.0.1:8642'
|
|
AGENT = 'py-websockets-cleint'
|
|
|
|
|
|
ws = websocket.create_connection(SERVER + "/getCaseCount")
|
|
count = json.loads(ws.recv())
|
|
ws.close()
|
|
|
|
success = 0
|
|
failed = 0
|
|
|
|
for case in range(1, count+1):
|
|
url = SERVER + '/runCase?case={}&agent={}'.format(case, AGENT)
|
|
status = websocket.STATUS_NORMAL
|
|
try:
|
|
ws = websocket.create_connection(url)
|
|
opcode, msg = ws.recv_data()
|
|
if opcode == websocket.ABNF.OPCODE_TEXT:
|
|
msg.decode("utf-8")
|
|
if opcode in (websocket.ABNF.OPCODE_TEXT, websocket.ABNF.OPCODE_BINARY):
|
|
ws.send(msg, opcode)
|
|
success += 1
|
|
except UnicodeDecodeError:
|
|
# this case is ok.
|
|
success += 1
|
|
status = websocket.STATUS_PROTOCOL_ERROR
|
|
except Exception as e:
|
|
failed += 1
|
|
status = websocket.STATUS_PROTOCOL_ERROR
|
|
print(traceback.format_exc())
|
|
finally:
|
|
ws.close(status)
|
|
|
|
print("Ran {} test cases. success: {}, faield: {}".format(case, success, failed))
|
|
url = SERVER + '/updateReports?agent={}'.format(AGENT)
|
|
ws = websocket.create_connection(url)
|