mirror of
https://github.com/kiviktnm/decman.git
synced 2026-09-19 12:08:28 +00:00
27 lines
552 B
Python
27 lines
552 B
Python
from decman.core.module import Module
|
|
from decman.plugins import run_method_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_method_with_attribute(m, "__flag__") == 123
|
|
|
|
|
|
def test_returns_none_if_no_method_has_attribute():
|
|
class M(Module):
|
|
def foo(self):
|
|
return 1
|
|
|
|
m = M("m")
|
|
assert run_method_with_attribute(m, "__flag__") is None
|