python+selenium 自动化测试框架-学习记录

^{本人小白一枚，想着把学习时的东西以博客的方式记录下来，文章中有不正确的地方请大佬多多指点！！共同学习}

前期准备

　　安装python3、selenium、下载对应版本的webdriver；安装所需的第三方库，不多赘述，最基础的东西，不会的自行跳转^0^

项目介绍

功能简述

对webdriver常用方法进行二次封装，使用起来更方便，同时会有log记录
log日志会同时打印在控制台和写入log文件中
测试完成后，会自动发送邮件，邮件信息conf.ini可配置
采用PO模式编写，元素信息维护在对应页面中
支持chrome，Firefox，IE浏览器
支持web和wap模式切换，和-headless模式运行

项目结构

　　目录结构如图所示

　　该框架采用PO模式编写，PO提供了一种业务流程与页面元素操作分离的模式，这使得测试代码变得更加清晰。页面对象与用例分离，使得我们更好的复用对象，可复用的页面方法代码会变得更加优化，更加有效的命名方式使得我们更加清晰的知道方法所操作的UI元素。当某些页面的元素发生改变时，只需要对应的页面类即可，后面会有详细介绍^

1 log输出模块

　　使用python的logging模块分别输出日志到文件和控制台，其实现方法供参考

    def __printconsole(self, level, message):

        # 创建一个logger

        logger = logging.getLogger()

        logger.setLevel(logging.DEBUG)

        # 创建一个handler，用于写入日志文件

        fh = logging.FileHandler(self.logname, 'a', encoding='utf-8')

        fh.setLevel(logging.DEBUG)

        # 再创建一个handler，用于输出到控制台

        ch = logging.StreamHandler()

        ch.setLevel(logging.DEBUG)

        # 定义handler的输出格式

        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')

        fh.setFormatter(formatter)

        ch.setFormatter(formatter)

        # 给logger添加handler

        logger.addHandler(fh)

        logger.addHandler(ch)

        # 记录一条日志

        if level == 'info':

            logger.info(message)

        elif level == 'debug':

            logger.debug(message)

        elif level == 'warning':

            logger.warning(message)

        elif level == 'error':

            logger.error(message)

        logger.removeHandler(ch)

        logger.removeHandler(fh)

        # 关闭打开的文件

        fh.close()

    def debug(self, message):

        self.__printconsole('debug', message)

    def info(self, message):

        self.__printconsole('info', message)

    def warning(self, message):

        self.__printconsole('warning', message)

    def error(self, message):

        self.__printconsole('error', message)

2 读取配置文件

　　配置文件我选择的时.ini格式的，其实可以写成一个python文件或者txt文件，对应的读取方法都大同小异。

　　我的conf.ini文件如下

[browserType]

;browserName = Firefox

browserName = Chrome

#browserName = IE

[testServer]

URL = https://www.baidu.com

[pattern]

options = Web

is_visible = T

[Mail]

mail_subject = 邮件主题

mail_server = smtp.xxxx.com

mail_from = xxxx@xxx.com

mail_to = xxxxx@xx.com,xxxxx@xxx.com

mail_from_pwd = xxxxxx

读取的方法也很简单，网上也有很多教程，贴出来仅供参考

class ReadConfig:

    """

    专门读取配置文件的，.ini文件格式

    """

    def __init__(self, filename=conf_path):

        with open(filename, 'r', encoding='UTF-8') as f:

            data = f.read()

            if data[:3] == codecs.BOM_UTF8:

                data = data[3:]

                files = codecs.open(filename, "w")

                files.write(data)

                files.close()

        self.cf = configparser.ConfigParser()

        self.cf.read(filename, encoding='UTF-8')  

    def getValue(self, env, name):

        """读取配置文件中的值"""

        return self.cf.get(env, name)

3 选择浏览器

　　这部分要做到的是根据配置文件选择要使用的浏览器，以及相应的运行模式，先上代码*

    def open_browser(self, driver):

        read = ReadConfig()

        browser = read.getValue("browserType", "browserName")

        options = read.getValue("pattern", "options")

        url = read.getValue("testServer", "URL")

        is_visible = read.getValue("pattern", "is_visible")

        if options.lower() == 'web':

            if browser == "Firefox":

                if is_visible == 'F':

                    options = webdriver.FirefoxOptions()

                    options.add_argument('-headless')

                    # options.add_argument('--disable-gpu')

                    driver = webdriver.Firefox(executable_path=firefox_driver, options=options)

                else:

                    driver = webdriver.Firefox(executable_path=firefox_driver)

                    log.info("启动{}浏览器".format(browser))

            elif browser == "Chrome":

                if is_visible == 'F':

                    option = webdriver.ChromeOptions()

                    option.add_argument('headless')

                    driver = webdriver.Chrome(chrome_driver, chrome_options=option)

                else:

                    driver = webdriver.Chrome(chrome_driver)

                    log.info("启动{}浏览器".format(browser))

        elif options.lower() == 'wap':

            mobileEmulation = {"deviceName": "iPhone 6"}

            options = webdriver.ChromeOptions()

            options.add_experimental_option('mobileEmulation', mobileEmulation)

            options.add_argument('--disable-search-geolocation-disclosure')

            driver = webdriver.Chrome(chrome_options=options)

　　通过读取配置文件获取相应的信息，选择相应的浏览器，支持web和wap两种模式，如果你想运行时，隐藏浏览器窗口可以使用浏览器的headless模式，代码中 if is_visible == 'F' 时，执行的就行浏览器的headless模式，浏览器在后台运行，不影响测试效果。

4 二次封装selenium方法

　　二次封装selenium中的常用方法可以让我们更方便的使用，也可以定制化我们的需求，比如加上日志输出，执行耗时等等

class BasePage:

    """测试基类"""

    def __init__(self, driver):

        self.driver = driver

    def get_img(self, rq=time.strftime('%Y%m%d%H%M', time.localtime(time.time()))):

        """截图"""

        path = os.path.join(os.path.abspath('..'), 'report', 'img')

        # path = os.path.join(getcwd.get_cwd(), 'screenshots/')  # 拼接截图保存路径

        # rq = time.strftime('%Y%m%d%H%M', time.localtime(time.time()))  # 按格式获取当前时间

        screen_name = path + rq + '.png'  # 拼接截图文件名

        # noinspection PyBroadException

        try:

            self.driver.get_screenshot_as_file(screen_name)

            log.info("截图保存成功{}".format(screen_name))

        except BaseException as e:

            log.error("截图失败{}".format(e))

    def find_element(self, selector):

        """定位元素"""

        by = selector[0]

        value = selector[1]

        element = None

        if by in ['id', 'name', 'class', 'tag', 'link', 'plink', 'css', 'xpath']:

            # noinspection PyBroadException

            try:

                if by == 'id':

                    element = self.driver.find_element_by_id(value)

                elif by == 'name':

                    element = self.driver.find_element_by_name(value)

                elif by == 'class':

                    element = self.driver.find_element_by_class_name(value)

                elif by == 'tag':

                    element = self.driver.find_element_by_tag_name(value)

                elif by == 'link':

                    element = self.driver.find_element_by_link_text(value)

                elif by == 'plink':

                    element = self.driver.find_element_by_partial_link_text(value)

                elif by == 'css':

                    element = self.driver.find_element_by_css_selector(value)

                elif by == 'xpath':

                    element = self.driver.find_element_by_xpath(value)

                else:

                    log.error('没有找到元素')

                log.info('元素定位成功。定位方式：%s,使用的值%s：' % (by, value))

                return element

            except NoSuchElementException as e:

                log.error("报错信息：{}".format(e))

                self.get_img()  # 调用截图

        else:

            log.error('输入的元素定位方式错误,参考[id, name, class, tag, link, plink, css,xpath]')

    def type(self, selector, value):

        """输入内容"""

        element = self.find_element(selector)

        element.clear()

        log.info('清空输入内容')

        # noinspection PyBroadException

        try:

            element.send_keys(value)

            log.info('输入的内容：%s' % value)

        except BaseException as e:

            log.error('内容输入报错{}'.format(e))

            self.get_img()

    def click(self, selector):

        """点击元素"""

        element = self.find_element(selector)

        # noinspection PyBroadException

        try:

            element.click()

            log.info('点击元素成功')

        except BaseException as e:

            display = self.isdisplayed(element)

            if display is True:

                self.my_sleep(3)

                element.click()

                log.info('点击元素成功')

            else:

                self.get_img()

                log.error('点击元素报错{}'.format(e))

　　这里只贴出一部分的常用的方法的封装示例，可以根据自己的需求DIY

5 用例筛选

　　当你的用例写到一定量时，如果你验证一部分用例，每次全部执行一遍太耗时了，所以加个用例筛选功能，自己选择要执行的用例文件，不想执行的直接注释掉

user/test01case

#user/test02case

#user/test03case

#user/test04case

#user/test05case

#shop/test_shop_list

#shop/test_my_shop

#shop/test_new_shop

　　然后再写一个方法去逐行读取这个文件，如果行首没有#号，就把这个用例添加到suite中执行即可，具体实现：

class AllTest:  # 定义一个类AllTest

    def __init__(self):  # 初始化一些参数和数据

        global report_name

        report_path = os.path.join(root_path, 'report', 'testreport')

        now = time.strftime('%Y-%m-%d_%H_%M_%S')

        report_name = os.path.join(report_path, 'TestResult{}.html'.format(now))

        self.caseListFile = os.path.join('.\\', "caselist.txt")  # 配置执行哪些测试文件的配置文件路径

        self.caseFile = os.path.join('.\\', "testcase")  # 真正的测试断言文件路径

        self.caseList = []

        log.info('report_name' + report_name)  # 将resultPath的值输入到日志，方便定位查看问题

        log.info('caseListFile' + self.caseListFile)  # 同理

        log.info('caseList' + str(self.caseList))  # 同理

    def set_case_list(self):

        """

        读取caselist.txt文件中的用例名称，并添加到caselist元素组

        :return:

        """

        fb = open(self.caseListFile)

        for value in fb.readlines():

            data = str(value)

            if data != '' and not data.startswith("#"):  # 如果data非空且不以#开头

                self.caseList.append(data.replace("\n", ""))  # 读取每行数据会将换行转换为\n，去掉每行数据中的\n

        fb.close()

    def set_case_suite(self):

        """

        :return:

        """

        self.set_case_list()  # 通过set_case_list()拿到caselist元素组

        test_suite = unittest.TestSuite()

        suite_module = []

        for case in self.caseList:  # 从caselist元素组中循环取出case

            case_name = case.split("/")[-1]  # 通过split函数来将aaa/bbb分割字符串，-1取后面，0取前面

            # 批量加载用例，第一个参数为用例存放路径，第一个参数为路径文件名

            discover = unittest.defaultTestLoader.discover(self.caseFile, pattern=case_name + '.py', top_level_dir=None)

            suite_module.append(discover)  # 将discover存入suite_module元素组

        if len(suite_module) > 0:  # 判断suite_module元素组是否存在元素

            for suite in suite_module:  # 如果存在，循环取出元素组内容，命名为suite

                for test_name in suite:  # 从discover中取出test_name，使用addTest添加到测试集

                    test_suite.addTest(test_name)

        else:

            log.error('suite_module中没有测试集')

            return None

        return test_suite  # 返回测试集

    def run(self):

        """

        run test

        :return:

        """

        try:

            suit = self.set_case_suite()  # 调用set_case_suite获取test_suite

            if suit is not None:  # 判断test_suite是否为空

                # bf(suite).report(description='用例名称xx', filename=now, log_path=report_path)

                with open(report_name, 'wb') as f:  # encoding='UTF-8'

                    # 调用HTMLTestRunner

                    runner = HTMLTestRunner.HTMLTestRunner(stream=f, title='Test Report',

                                                           description='Test Description')

                    runner.run(suit)

            else:

                log.error('没有case')

        except Exception as e:

            log.error('{}'.format(e))

　　当然，如果你不想使用这中方式也可以全部执行……　　

　　源码地址：https://gitee.com/fh1105/selenium_PO 需要的自行clone

　　第一次写博客记录学习，不好的地方请指出来，谢谢！