Key assumptions about Australia's COVID-19 response underpin Treasury's optimistic outlook for the territory budget.