MPs will have the whip withdrawn and be deselected if they rebel against the government, a source says.