Adding more mocking tests