Add test for WSClient