点此获取更多相关资料

本文为霍格沃兹测试开发学社学员学习笔记分享
原文链接：https://ceshiren.com/t/topic/27692

黑名单处理

• 运行过程中不定时弹框（广告弹窗，升级提示框，新消息提示框等等）

• 弹框不是 BUG（UI 界面提示，警告的作用）

方法一：常规处理

def find(self, by, locator):
    try:
        return self.driver.find_element(by, locator)
    except Exception as e:
        for black in black_list:
            eles = self.driver.find_elements(*black)
            if len(eles) > 0:
                eles[0].click()
                return find(by, locator)
        raise e

方法二：装饰器处理

装饰器的优势

• 对原有函数的功能增强
• 不改变原有函数的逻辑
• 使代码更简洁、易维护

__BLACK_LIST = []

def black_handle(func):
    """黑名单处理机制"""
    def _inner(*args, **kwargs):
        inner_arg = args[0]
        try:
            func(*args, **kwargs)
            # logger.info(f"查找元素：{locator}--->成功")
        except Exception as e:
            logger.info("未找到元素")
            logger.info(f"异常信息：{e}")
            logger.info("处理异常")
            for i in __BLACK_LIST:
                elements = inner_arg.driver.find_elements(*i)
                if len(elements) > 0:
                    elements[0].click()
                    return func(*args, **kwargs)
            raise e

    return _inner

装饰函数或方法

@black_handle
def do_find(self, locator: tuple):
        """获取单个元素"""
        element = self.driver.find_element(*locator)
        logger.info(f"查找元素：{locator}--->成功")
        return element