Wanindu Hasaranga is set to return to action in ODIs and T20Is since tearing his hamstring early in the T20 World Cup in February