Australia were sitting pretty against Afghanistan – with a rampant Travis Head and Steve Smith in complete control – before heavy rain hit Lahore.