Add additional specs and methods for mocking results