码迷,mamicode.com
首页 > 其他好文 > 详细

通过装饰器实现获取当前执行的用例名称

时间:2020-03-25 16:10:28      阅读:75      评论:0      收藏:0      [点我收藏+]

标签:module   span   测试   默认值   rgs   sel   case   获取   实现   

import inspect
"""
动态获取用例名称,并赋值给用例调用类中的某个固定参数pvid
通过装饰器的调用,可以在用例执行时将用例名称动态传给pvid,同时,如果用例名称不满足要求,pvid保持默认值pvid_test
"""


def get_current_function_name():
    """
    inspect.stack():动态获取运行的函数名
    """
    return inspect.stack()


def wrap_class(cls):
    """
    函数装饰类装饰器
    :param cls:
    :return:
    """
    def inner(**kwargs):
        res = get_current_function_name()
        func_names = [res[2][3], res[1][3], res[0][3]]
        print(func_names)
        
        case_name = ‘‘
        for func_name in func_names:
            if func_name.startswith(test_):
                case_name = func_name
                break
        if case_name.startswith(test_):
            return cls(pvid=case_name)
        else:
            return cls(**kwargs)
    return inner


@wrap_class
class ModuleProto:
    """
    测试用例中调用的类
    """
    def __init__(self, **kwargs):
        self.proto = {
            "pvid": "pvid_test"
        }
        if "pvid" in kwargs.keys():
            self.proto["pvid"] = kwargs["pvid"]
            
            
def test_case_1():
    """
    测试用例
    :return:
    """
    module_proto = ModuleProto()
    print("pvid: ", module_proto.proto[pvid])
  
  
if __name__ == __main__:
    test_case_1()

 

通过装饰器实现获取当前执行的用例名称

标签:module   span   测试   默认值   rgs   sel   case   获取   实现   

原文地址:https://www.cnblogs.com/jiyanjiao-702521/p/12566741.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!