Add tests for return_value mock param