I could mock this instead of writing all this logic if we had a mock framework. Just saying :)