Added WebSocket related unit tests