mirror of
https://github.com/kiviktnm/decman.git
synced 2026-09-19 12:08:28 +00:00
41 lines
832 B
Python
41 lines
832 B
Python
from decman.core.module import Module
|
|
from decman.plugins import run_methods_with_attribute
|
|
|
|
|
|
def mark(attr):
|
|
attr.__flag__ = True
|
|
return attr
|
|
|
|
|
|
def test_runs_marked_method_and_returns_value():
|
|
class M(Module):
|
|
@mark
|
|
def foo(self):
|
|
return 123
|
|
|
|
m = M("m")
|
|
assert run_methods_with_attribute(m, "__flag__") == [123]
|
|
|
|
|
|
def test_runs_marked_methods_and_returns_value():
|
|
class M(Module):
|
|
@mark
|
|
def foo(self):
|
|
return 123
|
|
|
|
@mark
|
|
def bar(self):
|
|
return 321
|
|
|
|
m = M("m")
|
|
assert run_methods_with_attribute(m, "__flag__") == [321, 123]
|
|
|
|
|
|
def test_returns_none_if_no_method_has_attribute():
|
|
class M(Module):
|
|
def foo(self):
|
|
return 1
|
|
|
|
m = M("m")
|
|
assert run_methods_with_attribute(m, "__flag__") == []
|