Sri Lanka want to follow India's Test lead