Not being able to afford an employee is obviously a valid reason to dismiss them. So is them refusing to work. But the scenario you describe is contrived anyway

Not being able to afford an employee is obviously a valid reason to dismiss them. So is them refusing to work. But the scenario you describe is contrived anyway: Someone motivated enough to work hard won't randomly stop being so. The "legal headaches" are only a serious problem if your legal system is broken.