Python+Selenium框架内封装日志类和浏览器引擎类

最新推荐文章于 2024-05-31 17:29:17 发布

转载最新推荐文章于 2024-05-31 17:29:17 发布 · 579 阅读

python+selenium 专栏收录该内容

18 篇文章

订阅专栏

本文详细介绍了使用Selenium进行自动化测试的过程，包括配置文件的设计、浏览器引擎类的实现、日志记录方式以及具体的测试脚本编写。通过实例展示了如何设置浏览器类型、测试服务器URL，如何启动浏览器并最大化窗口，以及如何执行百度搜索的自动化测试。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.创建如下格式的项目层级结构，文件名称保持一致

层级命名

2.2. 相关文件代码如下

配置文件config.ini相关设计如下：

 #this is config file, only store browser type and server URL

[browserType]
browserName = Chrome

[testServer]
URL = https://www.baidu.com

浏览器引擎类browser_engine.py相关脚本代码如下

# -*- coding:utf-8 -*-
import configparser
import os.path
import time
from selenium import webdriver
from automation_framework_demo.framework.logger import Logger

logger = Logger(logger="BrowserEngine").getlog()


class BrowserEngine(object):

    def __init__(self):

        pass


    def open_browser(self):
        config = configparser.ConfigParser()
        file_path = os.path.dirname(os.path.abspath('.')) + '/config/config.ini'
        config.read(file_path)

        browser = config.get("browserType", "browserName")
        logger.info("You had select %s browser." % browser)
        url = config.get("testServer", "URL")
        logger.info("The test server url is: %s" % url)
        self.driver = webdriver.Chrome("C:\\Users\suxiahua\AppData\Local\Programs\Python\Python36\Lib\site-packages\chromedriver_win32\chromedriver.exe")
        logger.info("Starting Chrome browser.")


        self.driver.get(url)
        logger.info("Open url: %s" % url)
        self.driver.maximize_window()
        logger.info("Maximize the current window.")
        time.sleep(2)
        logger.info("Set implicitly wait 2 seconds.")

        return self.driver

日志类logger.py相关脚本代码如下：

# _*_ coding: utf-8 _*_
import logging
import os.path
import time


class Logger(object):

    def __init__(self, logger):
        '''
            指定保存日志的文件路径，日志级别，以及调用文件
            将日志存入到指定的文件中
        '''

        # 创建一个logger
        self.logger = logging.getLogger(logger)
        self.logger.setLevel(logging.DEBUG)

        # 创建一个handler，用于写入日志文件
        rq = time.strftime('%Y%m%d%H%M', time.localtime(time.time()))
        # log_path = os.path.dirname(os.getcwd()) + '/Logs/'  # 项目根目录下/Logs 保存日志
        log_path = os.path.dirname(os.path.abspath('.')) + '/logs/'
        # 如果case组织结构式 /testsuit/featuremodel/xxx.py ， 那么得到的相对路径的父路径就是项目根目录
        log_name = log_path + rq + '.log'
        fh = logging.FileHandler(log_name)
        fh.setLevel(logging.INFO)

        # 再创建一个handler，用于输出到控制台
        ch = logging.StreamHandler()
        ch.setLevel(logging.INFO)

        # 定义handler的输出格式
        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
        fh.setFormatter(formatter)
        ch.setFormatter(formatter)

        # 给logger添加handler
        self.logger.addHandler(fh)
        self.logger.addHandler(ch)

    def getlog(self):
        return self.logger

测试脚本baidu_search.py相关脚本如下：

# coding=utf-8
import time
import unittest
from automation_framework_demo.framework.browser_engine import BrowserEngine


class BaiduSearch(unittest.TestCase):

    def setUp(self):

        browse = BrowserEngine()

        self.driver = browse.open_browser()



    def test_baidu_search(self):


        self.driver.find_element_by_id('kw').send_keys('selenium')
        time.sleep(1)
        try:
            assert 'selenium' in self.driver.title
            print('Test Pass.')
        except Exception as e:
            print('Test Fail.', format(e))

    def tearDown(self):
        self.driver.quit()

if __name__ == '__main__':
        unittest.main()

最后运行测试脚本baidu_search.py