开发者

Access self from decorator

开发者 https://www.devze.com 2023-04-08 22:34 出处:网络
In setUp() method of unittest I\'ve setup some self variables, which are later referenced in actual tests. I\'ve also created a decorator to do some logging. Is there a way in which I can access those

In setUp() method of unittest I've setup some self variables, which are later referenced in actual tests. I've also created a decorator to do some logging. Is there a way in which I can access those self variables from decorator?

For the sake of s开发者_开发知识库implicity, I'm posting this code:

def decorator(func):
    def _decorator(*args, **kwargs):
        # access a from TestSample
        func(*args, **kwargs)
    return _decorator

class TestSample(unittest.TestCase):    
    def setUp(self):
        self.a = 10

    def tearDown(self):
        # tear down code

    @decorator
    def test_a(self):
        # testing code goes here

What would be the best way of accessing a (set in setUp()) from decorator?


Since you're decorating a method, and self is a method argument, your decorator has access to self at runtime. Obviously not at parsetime, because there are no objects yet, just a class.

So you change your decorator to:

def decorator(func):
    def _decorator(self, *args, **kwargs):
        # access a from TestSample
        print 'self is %s' % self
        return func(self, *args, **kwargs)
    return _decorator
0

精彩评论

暂无评论...
验证码 换一张
取 消

关注公众号