Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def mock_method(runtime_self, *kargs, **kwargs):
arguments = {'kargs': kargs, 'kwargs': kwargs}
expectation = FlexmockContainer.get_flexmock_expectation(
self, name, arguments)
if expectation:
if not expectation.runnable():
raise StateError('%s expected to be called when %s is True' %
(name, expectation._get_runnable()))
expectation.times_called += 1
expectation.verify(final=False)
_pass_thru = _getattr(expectation, '_pass_thru')
_replace_with = _getattr(expectation, '_replace_with')
if _pass_thru:
return pass_thru(expectation, runtime_self, *kargs, **kwargs)
elif _replace_with:
return _replace_with(*kargs, **kwargs)
return_values = _getattr(expectation, 'return_values')
if return_values:
return_value = return_values[0]
def mock_method(runtime_self, *kargs, **kwargs):
arguments = {'kargs': kargs, 'kwargs': kwargs}
expectation = FlexmockContainer.get_flexmock_expectation(
self, method, arguments)
if expectation:
expectation.times_called += 1
if expectation._pass_thru:
return pass_thru(expectation, *kargs, **kwargs)
elif expectation._replace_with:
return expectation._replace_with(*kargs, **kwargs)
if expectation.yield_values:
return generator_method(expectation.yield_values)
elif expectation.return_values:
return_value = expectation.return_values[0]
expectation.return_values = expectation.return_values[1:]
expectation.return_values.append(return_value)
else:
return_value = ReturnValue()
if return_value.raises: