Add withSocketsDo to tests