We generally avoid stubbing the system under test. IMO, let's roll with the alternative you provided in your comment.