Clean up socket tests