80 lines
1.9 KiB
Python
80 lines
1.9 KiB
Python
"""
|
|
py.test.mark / keyword plugin
|
|
|
|
"""
|
|
import py
|
|
|
|
def pytest_namespace(config):
|
|
mark = KeywordDecorator({})
|
|
return {'mark': mark}
|
|
|
|
class KeywordDecorator:
|
|
""" decorator for setting function attributes. """
|
|
def __init__(self, keywords, lastname=None):
|
|
self._keywords = keywords
|
|
self._lastname = lastname
|
|
|
|
def __call__(self, func=None, **kwargs):
|
|
if func is None:
|
|
kw = self._keywords.copy()
|
|
kw.update(kwargs)
|
|
return KeywordDecorator(kw)
|
|
elif not hasattr(func, 'func_dict'):
|
|
kw = self._keywords.copy()
|
|
name = self._lastname
|
|
if name is None:
|
|
name = "mark"
|
|
kw[name] = func
|
|
return KeywordDecorator(kw)
|
|
func.func_dict.update(self._keywords)
|
|
return func
|
|
|
|
def __getattr__(self, name):
|
|
if name[0] == "_":
|
|
raise AttributeError(name)
|
|
kw = self._keywords.copy()
|
|
kw[name] = True
|
|
return self.__class__(kw, lastname=name)
|
|
|
|
def test_pytest_mark_getattr():
|
|
mark = KeywordDecorator({})
|
|
def f(): pass
|
|
|
|
mark.hello(f)
|
|
assert f.hello == True
|
|
|
|
mark.hello("test")(f)
|
|
assert f.hello == "test"
|
|
|
|
py.test.raises(AttributeError, "mark._hello")
|
|
py.test.raises(AttributeError, "mark.__str__")
|
|
|
|
def test_pytest_mark_call():
|
|
mark = KeywordDecorator({})
|
|
def f(): pass
|
|
mark(x=3)(f)
|
|
assert f.x == 3
|
|
def g(): pass
|
|
mark(g)
|
|
assert not g.func_dict
|
|
|
|
mark.hello(f)
|
|
assert f.hello == True
|
|
|
|
mark.hello("test")(f)
|
|
assert f.hello == "test"
|
|
|
|
mark("x1")(f)
|
|
assert f.mark == "x1"
|
|
|
|
def test_mark_plugin(testdir):
|
|
p = testdir.makepyfile("""
|
|
import py
|
|
pytest_plugins = "keyword"
|
|
@py.test.mark.hello
|
|
def test_hello():
|
|
assert hasattr(test_hello, 'hello')
|
|
""")
|
|
result = testdir.runpytest(p)
|
|
assert result.stdout.fnmatch_lines(["*passed*"])
|