装饰器如下:
''' 获取函数名及注释 ''' def get_fun_name_doc(func): @wraps(func) def inner(*args): logging.info('正在执行:{},测试场景为:{}'.format(func.__name__, func.__doc__)) if args: ret = func(*args) else: ret = func() return ret return inner
使用如下:
@Screen(driver) # 添加装饰器修饰,失败时自动截图 @get_fun_name_doc # 添加装饰器,打印函数名和docstring到日志中 def test_search_2(self): '''搜索关键词selenium''' self.driver.get(self.base_url) self.page.search_input.send_keys('selenium') self.page.search_button.click() time.sleep(2) self.assertEqual(self.driver.title, 'selenium_百度搜索') time.sleep(2)
打印日志结果如下: