| SERVER = 'ws://127.0.0.1:8642' |
| AGENT = 'py-websockets-client' |
| ws = websocket.create_connection(SERVER + "/getCaseCount") |
| count = json.loads(ws.recv()) |
| for case in range(1, count+1): |
| url = SERVER + '/runCase?case={0}&agent={1}'.format(case, AGENT) |
| status = websocket.STATUS_NORMAL |
| ws = websocket.create_connection(url) |
| opcode, msg = ws.recv_data() |
| if opcode == websocket.ABNF.OPCODE_TEXT: |
| if opcode in (websocket.ABNF.OPCODE_TEXT, websocket.ABNF.OPCODE_BINARY): |
| except UnicodeDecodeError: |
| status = websocket.STATUS_PROTOCOL_ERROR |
| except websocket.WebSocketProtocolException: |
| status = websocket.STATUS_PROTOCOL_ERROR |
| except websocket.WebSocketPayloadException: |
| status = websocket.STATUS_INVALID_PAYLOAD |
| # status = websocket.STATUS_PROTOCOL_ERROR |
| print(traceback.format_exc()) |
| print("Ran {} test cases.".format(case)) |
| url = SERVER + '/updateReports?agent={0}'.format(AGENT) |
| ws = websocket.create_connection(url) |