Pujara (133) and Ajinkya Rahane (132) added 217 runs to guide India to 622-9 declared in the first innings. The visitors then dismissed Sri Lanka for 183 and enforced follow-on after taking a 400-plus run lead.