added more robust testing dummy servers

This commit is contained in:
Nickiel12 2024-05-29 12:40:19 -05:00
parent fc30a669a5
commit d23b3c711e
2 changed files with 62 additions and 29 deletions

View file

@ -1,30 +1,30 @@
import asyncio import asyncio
import websockets import websockets
socket_connections = {} socket_connections = {}
async def handle_client(websocket, path): async def handle_client(websocket, path):
global socket_connections global socket_connections
print("packet connected") print("packet connected")
socket_connections[websocket] = "" socket_connections[websocket] = ""
try: try:
while True: while True:
print("waiting for data") print("waiting for data")
data = await websocket.recv() data = await websocket.recv()
# await update_from_packet(data) # await update_from_packet(data)
except websockets.exceptions.ConnectionClosed: except websockets.exceptions.ConnectionClosed:
print("Connection closed with: ", websocket.remote_address) print("Connection closed with: ", websocket.remote_address)
finally: finally:
print("closing websocket") print("closing websocket")
del socket_connections[websocket] del socket_connections[websocket]
async def main(): async def main():
server = await websockets.serve(handle_client, "localhost", 8765) server = await websockets.serve(handle_client, "localhost", 8765)
print("server started") print("server started")
await server.wait_closed() await server.wait_closed()
asyncio.run(main()) asyncio.run(main())

View file

@ -0,0 +1,33 @@
import asyncio
import websockets
socket_connections = {}
async def handle_client(websocket, path):
global socket_connections
print("packet connected")
socket_connections[websocket] = ""
try:
while True:
print("waiting for data")
data = await websocket.recv()
if data == "Type?":
await websocket.send("Automated")
else:
await websocket.send("[1] 100:100 200:700\n[2] 250:250 750:750\n[3] 500:300 800:800")
except websockets.exceptions.ConnectionClosed:
print("Connection closed with: ", websocket.remote_address)
finally:
print("closing websocket")
del socket_connections[websocket]
async def main():
server = await websockets.serve(handle_client, "localhost", 6543)
print("server started")
await server.wait_closed()
asyncio.run(main())