Update tests/websockets.py