def _verify_call_order(self, flex_objects): for exp in flex_objects[self.mock]: if (exp.method == self.method and not _match_args(self.args, exp.args) and not exp._times_called): raise CallOrderError( '%s called before %s' % (_format_args(self.method, self.args), _format_args(exp.method, exp.args))) if (exp.method == self.method and self.args and exp.args and # ignore default stub case _match_args(self.args, exp.args)): break
def __get_expectation(self, name=None, args=None): """Gets attached to the object under mock and is called in that context.""" if args == None: args = {'kargs': (), 'kwargs': {}} if not isinstance(args, dict): args = {'kargs': args, 'kwargs': {}} if not isinstance(args['kargs'], tuple): args['kargs'] = (args['kargs'],) if name and self in _flex_objects: for e in reversed(_flex_objects[self]): if e.method == name and _match_args(args, e.args): if e._ordered: e._verify_call_order(_flex_objects) return e