Adding mocking tests