Add Webocket tests