Wrote more test for the webSocket service #64