support netsim instance numbers

This commit is contained in:
Gilles Boccon-Gibod
2023-09-22 14:22:04 -07:00
parent 4c29a16271
commit 64937c3f77
4 changed files with 55 additions and 38 deletions

View File

@@ -38,10 +38,14 @@ async def open_ws_client_transport(spec: str) -> Transport:
websocket = await websockets.client.connect(spec)
transport = PumpedTransport(
class WsTransport(PumpedTransport):
async def close(self):
await super().close()
await websocket.close()
transport = WsTransport(
PumpedPacketSource(websocket.recv),
PumpedPacketSink(websocket.send),
websocket.close,
)
transport.start()
return transport