装饰器在鼻子测试案例，产生

Question

我有以下装饰应该包装在try/except块中的测试用例函数的实现和打印日志如果发生异常。

def print_log(test_case): 
    @wraps(test_case) 
    def run_test(self): 
     try: 
      test_case(self) 
     except: 
      Test_Loop.failure_teardown(self) 
      raise 
    return run_test 

然而，这似乎并没有对我的测试情形之一的工作，调用产量发电机

请多多包涵，因为这是一个基本的例子：

class Test_Loop: 
    # ton of implementation here (e.g. initialization, etc) 

    def runIt(self, name, ip, port): 
     # code here 

    @print_log 
    def test_log_looper(self): 
     for l in self.links: 
      # initialize variables seen below and other stuff 
      for n in names: 
       # do stuff 
       for i in ips: 
        # do stuff 
        for p in ports: 
         yield self.runIt, l, n, i, p 

从调试，当应用装饰器时，它似乎甚至不会进入第一个循环。我究竟做错了什么？

Answer 1

你需要迭代你的发生器。像这样修改你的修饰器：

def print_log(test_case): 
    @wraps(test_case) 
    def run_test(self): 
     try: 
      for _ in test_case(self): pass 
     except: 
      Test_Loop.failure_teardown(self) 
      raise 
    return run_test