fix issue 28 - setup_method now works with pytest_generate_tests

This commit is contained in:
holger krekel
2011-03-02 18:03:43 +01:00
parent f1b5dae1fb
commit 6f3b84da9f
5 changed files with 23 additions and 7 deletions
+4 -5
View File
@@ -297,13 +297,8 @@ class Instance(PyCollectorMixin, pytest.Collector):
class FunctionMixin(PyobjMixin):
""" mixin for the code common to Function and Generator.
"""
def setup(self):
""" perform setup for this test function. """
if inspect.ismethod(self.obj):
name = 'setup_method'
else:
name = 'setup_function'
if hasattr(self, '_preservedparent'):
obj = self._preservedparent
elif isinstance(self.parent, Instance):
@@ -311,6 +306,10 @@ class FunctionMixin(PyobjMixin):
self.obj = self._getobj()
else:
obj = self.parent.obj
if inspect.ismethod(self.obj):
name = 'setup_method'
else:
name = 'setup_function'
setup_func_or_method = getattr(obj, name, None)
if setup_func_or_method is not None:
setup_func_or_method(self.obj)