Extending infrastructure test to WebSockets