日志(使用装饰器打印函数名及docstring到日志中)

装饰器如下:

'''
获取函数名及注释
'''
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)

打印日志结果如下:

 

posted @ 2020-05-07 18:13  小虫虫大虫虫  阅读(430)  评论(0编辑  收藏  举报