Novak Djokovic survived the first serious test of his French Open credentials by fighting back from two sets to one down to beat Diego Schwartzman.