import websocket
import sys

print('Testing WS connection to Xray...')
try:
    ws = websocket.create_connection('ws://localhost:8443/vless', timeout=10)
    print('WS connection established successfully')
    ws.send(b'\x00')  # dummy data
    data = ws.recv()
    print('Received:', repr(data))
    ws.close()
    print('Test PASSED')
except Exception as e:
    print(f'WS connection FAILED: {e}')
    sys.exit(1)