Fix socket timeout test -- mock the side_effect