Adding test for mocking third party function.