US official says Iran knew team would have to leave the country shortly after World Cup match