Donald Trump's threat to punish Iran is putting both China and the U.S. in a tough spot ahead of Xi's visit.