Added two test, a mock and an integration