Steve Smith is set to captain Australia in a three-match Test series against Sri Lanka