mirror of
https://github.com/google/bumble.git
synced 2026-05-09 04:08:02 +00:00
support netsim instance numbers
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user