Python Selenium搭建UI自动化测试框架_selenium python ui自动化测试框架

├── report
│ ├── init.py
│ └── test_report.html
│ └── screenshots
│ ├── test_login_failure_2022-05-01_08-30-00.png
│ └── test_login_success_2022-05-01_08-30-00.png
├── testcase
│ ├── init.py
│ └── test_login.py
├── utils
│ ├── init.py
│ ├── driver.py
│ ├── logger.py
│ └── read_config.py
│ └── take_screenshot.py
├── .gitignore
├── README.md
├── requirements.txt
└── run.py

config：存放配置文件，包括config.ini和logger.ini。


data：存放测试数据，包括Excel和JSON格式的数据。


logs：存放日志文件，包括test.log。


page：存放页面对象，每个页面对应一个.py文件。


report：存放测试报告，包括test\_report.html。


screenshots：用于存放测试过程中的截图。


testcase：存放测试用例，每个用例对应一个.py文件。


utils：存放工具类，包括driver.py、logger.py和read\_config.py。


.gitignore：Git忽略文件列表。


README.md：项目说明文件。


requirements.txt：Python依赖包列表。


run.py：测试执行入口。


三、框架搭建  
 1.创建项目目录  
 在本地创建一个项目目录，进入该目录，执行以下命令：

mkdir config data logs page report report/screenshots testcase utils
touch .gitignore README.md requirements.txt run.py
cd config && touch init.py config.ini logger.ini && cd …
cd data && touch init.py test_data.xlsx test_data.json && cd …
cd logs && touch init.py test.log && cd …
cd page && touch init.py login_page.py && cd …
cd report && touch init.py test_report.html && cd …
cd testcase && touch init.py test_login.py && cd …
cd utils && touch init.py driver.py logger.py read_config.py && cd …

### 2.安装依赖包


#### 在项目目录下执行以下命令，安装依赖包：

pip install selenium openpyxl configparser loguru

### 3.编写配置文件


#### 在config目录下的config.ini文件中，配置浏览器类型和网站地址：

[Browser]
browser_name = chrome

[URL]
base_url = https://www.example.com

#### 在config目录下的logger.ini文件中，配置日志文件路径和日志级别：

[loggers]
keys=root

[handlers]
keys=consoleHandler,fileHandler

[formatters]
keys=formatter

[logger_root]
level=DEBUG
handlers=consoleHandler,fileHandler

[handler_consoleHandler]
class=StreamHandler
level=DEBUG
formatter=formatter
args=(sys.stdout,)

[handler_fileHandler]
class=handlers.TimedRotatingFileHandler
level=DEBUG
formatter=formatter
args=(‘logs/test.log’, ‘D’, 1, 30)

[formatter_formatter]
format=%(asctime)s - %(name)s - %(levelname)s - %(message)s
datefmt=%Y-%m-%d %H:%M:%S

### 4.编写工具类


#### 在utils目录下的driver.py文件中，封装获取浏览器驱动的方法：

import os
from selenium import webdriver
from configparser import ConfigParser

def get_driver():
config = ConfigParser()
config.read(os.path.join(os.path.dirname(file), ‘…/config/config.ini’))

browser_name = config.get('Browser', 'browser_name')
if browser_name.lower() == 'chrome':
    return webdriver.Chrome()
elif browser_name.lower() == 'firefox':
    return webdriver.Firefox()
elif browser_name.lower() == 'edge':
    return webdriver.Edge()
else:
    raise ValueError('Unsupported browser: {}'.format(browser_name))

#### 在utils目录下的logger.py文件中，封装日志记录的方法：

import os
from loguru import logger
from configparser import ConfigParser

config = ConfigParser()
config.read(os.path.join(os.path.dirname(file), ‘…/config/logger.ini’))

logger.add(config.get(‘handler_fileHandler’, ‘class’), level=config.get(‘logger_root’, ‘level’), rotation=config.getint(‘handler_fileHandler’, ‘args’)[2], retention=config.getint(‘handler_fileHandler’, ‘args’)[3])

def get_logger(name):
return logger.bind(name=name)

#### 在utils目录下的read\_config.py文件中，封装读取配置文件的方法：

import os
from configparser import ConfigParser

config = ConfigParser()
config.read(os.path.join(os.path.dirname(file), ‘…/config/config.ini’))

def get_base_url():
return config.get(‘URL’, ‘base_url’)

#### 在utils目录下新增了一个take\_screenshot.py文件，用于封装截图的方法：

import os
from datetime import datetime
from utils.driver import get_driver

def take_screenshot(name):
driver = get_driver()
screenshot_dir = os.path.join(os.path.dirname(file), ‘…/report/screenshots’)
os.makedirs(screenshot_dir, exist_ok=True)
screenshot_file = os.path.join(screenshot_dir, ‘{}{}.png’.format(name, datetime.now().strftime('%Y-%m-%d%H-%M-%S’)))
driver.save_screenshot(screenshot_file)

### 5.编写页面对象


#### 在page目录下的login\_page.py文件中，封装登录页面的元素定位和操作方法：

from selenium.webdriver.common.by import By
from utils.driver import get_driver

class LoginPage:
url = ‘/login.html’
username_input = (By.ID, ‘username’)
password_input = (By.ID, ‘password’)
login_button = (By.ID, ‘login-button’)

def __init__(self):
    self.driver = get_driver()

def open(self):
    self.driver.get(get_base_url() + self.url)

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

def login(self, username, password):
    self.driver.find_element(*self.username_input).send_keys(username)
    self.driver.find_element(*self.password_input).send_keys(password)
    self.driver.find_element(*self.login_button).click()

### 6.编写测试用例


#### 在testcase目录下的test\_login.py文件中，编写登录测试用例：

import unittest
from page.login_page import LoginPage
from utils.take_screenshot import take_screenshot

class TestLogin(unittest.TestCase):
def setUp(self):
self.page = LoginPage()

def tearDown(self):
    self.page.close()

def test_login_success(self):
    self.page.open()
    self.page.login('admin', 'admin123')
    self.assertIn('Welcome', self.page.driver.title)

def test_login_failure(self):
    self.page.open()
    self.page.login('admin', 'wrong_password')
    self.assertIn('Login failed', self.page.driver.page_source)
    take_screenshot('test_login_failure')

### 7.编写测试执行入口


#### 在run.py文件中，编写测试执行入口：

import unittest
from datetime import datetime
from utils.logger import get_logger
from report import HTMLTestRunner

logger = get_logger(name)

if name == ‘main’:
logger.info(‘Start testing…’)
suite = unittest.defaultTestLoader.discover(‘testcase’)
report_file = ‘report/test_report_{}.html’.format(datetime.now().strftime(‘%Y-%m-%d_%H-%M-%S’))
with open(report_file, ‘wb’) as f:
runner = HTMLTestRunner.HTMLTestRunner(stream=f, title=‘Test Report’, description=‘Test Result’)
runner.run(suite)
logger.info(‘Testing finished. Report file: {}’.format(report_file))

# 将截图嵌入测试报告
with open(report_file, 'r+', encoding='utf-8') as f:
    content = f.read()
    for root, dirs, files in os.walk('report/screenshots'):
        for file in files:
            screenshot_file = os.path.join(root, file)
            if 'test_report' not in screenshot_file:
                screenshot_name = os.path.splitext(os.path.basename(screenshot_file))[0]
                screenshot_time = datetime.strptime(screenshot_name.split('_')[-2], '%Y-%m-%d')
                screenshot_url = os.path.join(get_base_url(), screenshot_file)
                content = content.replace('{}"'.format(screenshot_name), '{}" width="50%"'.format(screenshot_url))
    f.seek(0)
    f.write(content)

## 四、测试执行



![img](https://img-blog.csdnimg.cn/img_convert/5c38cd511f6a0ee856729bcd48a816ce.png)
![img](https://img-blog.csdnimg.cn/img_convert/995e4a55eade1f755489fd4b85e7a195.png)

**网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。**

**一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！**

)
        f.seek(0)
        f.write(content)

四、测试执行

[外链图片转存中…(img-bWHtwgmw-1719233910627)]
[外链图片转存中…(img-9sNkn2Ig-1719233910627)]

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！