Lines Matching full:self
65 def __init__(self, name):
66 self.name = name
68 def __repr__(self):
69 return '<SentinelObject "%s">' % self.name
73 def __init__(self):
74 self._sentinels = {}
76 def __getattr__(self, name):
77 return self._sentinels.setdefault(name, SentinelObject(name))
99 def __init__(self, spec=None, side_effect=None, return_value=DEFAULT,
101 self._parent = parent
102 self._name = name
106 self._methods = spec
107 self._children = {}
108 self._return_value = return_value
109 self.side_effect = side_effect
110 self._wraps = wraps
112 self.reset_mock()
115 def reset_mock(self):
116 self.called = False
117 self.call_args = None
118 self.call_count = 0
119 self.call_args_list = []
120 self.method_calls = []
121 for child in self._children.itervalues():
123 if isinstance(self._return_value, Mock):
124 self._return_value.reset_mock()
127 def __get_return_value(self):
128 if self._return_value is DEFAULT:
129 self._return_value = Mock()
130 return self._return_value
132 def __set_return_value(self, value):
133 self._return_value = value
138 def __call__(self, *args, **kwargs):
139 self.called = True
140 self.call_count += 1
141 self.call_args = (args, kwargs)
142 self.call_args_list.append((args, kwargs))
144 parent = self._parent
145 name = self._name
154 if self.side_effect is not None:
155 if (isinstance(self.side_effect, Exception) or
156 isinstance(self.side_effect, (type, ClassType)) and
157 issubclass(self.side_effect, Exception)):
158 raise self.side_effect
160 ret_val = self.side_effect(*args, **kwargs)
162 ret_val = self.return_value
164 if self._wraps is not None and self._return_value is DEFAULT:
165 return self._wraps(*args, **kwargs)
167 ret_val = self.return_value
171 def __getattr__(self, name):
172 if self._methods is not None:
173 if name not in self._methods:
178 if name not in self._children:
180 if self._wraps is not None:
181 wraps = getattr(self._wraps, name)
182 self._children[name] = Mock(parent=self, name=name, wraps=wraps)
184 return self._children[name]
187 def assert_called_with(self, *args, **kwargs):
188 assert self.call_args == (args, kwargs), 'Expected: %s\nCalled with: %s' % ((args, kwargs), self.call_args)
211 def __init__(self, target, attribute, new, spec, create):
212 self.target = target
213 self.attribute = attribute
214 self.new = new
215 self.spec = spec
216 self.create = create
217 self.has_local = False
220 def __call__(self, func):
222 func.patchings.append(self)
239 patched.patchings = [self]
246 def get_original(self):
247 target = self.target
248 name = self.attribute
249 create = self.create
263 def __enter__(self):
264 new, spec, = self.new, self.spec
265 original = self.get_original()
277 self.temp_original = original
278 setattr(self.target, self.attribute, new)
282 def __exit__(self, *_):
283 if self.temp_original is not DEFAULT:
284 setattr(self.target, self.attribute, self.temp_original)
286 delattr(self.target, self.attribute)
287 del self.temp_original
