You can buy it as long as the visa is valid at check-in. Best thing to do is that you get a unrestricted fare and have the depature date be the last day that your visa will expire.