Adding test mocks