The UN nuclear watchdog looked set to impose a deadline on Iran to prove it is not running a covert atomic weapons program as Tehran warned it could end cooperation with the organisation.