Add ws integration test