我有一大堆 Django 测试用例,它们使用 setUp 方法来初始化在众多测试中使用的一些属性,它们的构造方式和相互依赖的方式是我想要从测试用例中移出的逻辑,重用
def setUp(self):
self.property_1 = ##some logic
...
我想将它们重写为一些方便的包装器,可以通过简单的继承或装饰器注入(inject)到类中,例如
@with_property_1(x=1, y=2)
def setUp(self):
...
def with_property_1(**model_kwargs):
def wrapper(f):
def wrapped(*args, **kwargs):
self = args[0]
self.property_1 = ## logic
f(*args, **kwargs)
return wrapped
return wrapper
但问题是 PyCharm 无法识别这些实例属性的存在,因为 TestCase 类内部没有任何内容设置它们。是否有另一种方法可以很好地实现这一目标,或者有一种方法可以诱使 PyCharm 认识到这些属性在装饰器存在的情况下是合法的?
最佳答案
免责声明:未经测试。
这里的问题是 Python 不能神奇地将 self
放入装饰器的上下文中(这在任何 IDE 中都不起作用)。您可能忘记的是,self
是调用每个类方法时传递给它的参数之一。因此,它存在于您的 *args
中,您可以对其进行操作。
这是我的试用代码:
def with_property_1(**model_kwargs):
def wrapper(f):
def wrapped(*args, **kwargs):
for key, value in model_kwargs.items():
setattr(args[0], key, value)
f(*args, **kwargs)
return wrapped
return wrapper
说明:
迭代
**model_kwargs
中的每个键/值。使用
setattr
使用提供的 kwargs 修改args[0]
,它应该是self
变量。使用更新后的
self
变量正常调用该函数。
关于python - 通过装饰器注入(inject)类属性,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55209924/