在现代软件开发中,性能和用户体验是至关重要的,而负载测试和自动化测试可以帮助我们实现这一目标。在本文中,我们将讨论如何在 Kubernetes 环境中运行 Locust 和 Selenium,并详细介绍如何安装 Chrome 和 ChromeDriver。
1. Dockerfile 配置
首先,我们需要创建一个 Dockerfile,以构建一个包含 Locust 和 Selenium 的 Docker 镜像。以下是 Dockerfile 的内容:
FROM locustio/locust:2.31.3
# 设置 Chrome 的版本
ENV CHROME_VERSION 130.0.6723.69
USER root
RUN apt-get update -qq -y && \
apt-get install -y wget unzip && \
apt-get install -y \
libasound2 \
libatk-bridge2.0-0 \
libgtk-4-1 \
libnss3 \
xdg-utils && \
wget -q -O chrome-linux64.zip https://storage.googleapis.com/chrome-for-testing-public/$CHROME_VERSION/linux64/chrome-linux64.zip && \
unzip chrome-linux64.zip && \
rm chrome-linux64.zip && \
mv chrome-linux64 /opt/chrome/ && \
ln -s /opt/chrome/chrome /usr/local/bin/ && \
wget -q -O chromedriver-linux64.zip https://storage.googleapis.com/chrome-for-testing-public/$CHROME_VERSION/linux64/chromedriver-linux64.zip && \
unzip -j chromedriver-linux64.zip chromedriver-linux64/chromedriver && \
rm chromedriver-linux64.zip && \
mv chromedriver /usr/local/bin/
# 设置 Chrome 的配置和缓存目录
ENV XDG_CONFIG_HOME=/tmp/.chromium
ENV XDG_CACHE_HOME=/tmp/.chromium
COPY . .
RUN pip install -r requirements.txt
解释
- 基础镜像 :使用
locustio/locust
作为基础镜像。 - 安装依赖:更新包管理器并安装必要的库,以确保 Chrome 和 ChromeDriver 正常运行。
- 下载和安装 Chrome 和 ChromeDriver :从 Google 的存储库下载 Chrome 和 ChromeDriver。
- 配置目录 :通过环境变量设置 Chrome 的配置和缓存目录,这一步非常重要。若未设置正确,可能会在 Kubernetes 中出现权限问题,具体参考 puppeteer-sharp
requirements.txt 示例
locust=2.31.3
selenium==4.21.0
2. Chrome 选项配置
在使用 Selenium 时,我们需要为 Chrome 配置一些选项,以确保它能够在无头模式下正常工作。以下是获取 Chrome 选项的代码示例:
import platform
from selenium import webdriver
def is_running_in_linux():
return platform.system() == 'Linux'
def get_chrome_options():
is_in_linux = is_running_in_linux()
options_custom = webdriver.ChromeOptions()
# Linux 下的 Chrome 选项
if is_in_linux:
options_custom.add_argument("--headless") # 无头模式
options_custom.add_argument('--disable-gpu') # 禁用 GPU 加速
options_custom.add_argument("--no-sandbox") # 禁用沙箱模式
else:
options_custom.add_argument("--start-maximized") # 启动时最大化窗口
# 其他通用选项
options_custom.add_argument("--disable-dev-shm-usage") # 解决资源限制问题
options_custom.add_argument("--ignore-ssl-errors=yes") # 忽略 SSL 错误
options_custom.add_argument("--disable-cache") # 禁用缓存
return options_custom
解释
- 操作系统检测:根据当前操作系统选择适当的 Chrome 选项。
- 无头模式:在 Linux 环境中使用无头模式,以便在没有图形界面的情况下运行 Chrome。
- 禁用沙箱:在 Kubernetes 环境中,禁用沙箱模式可以避免潜在的权限问题。
3. Locust 用户定义
下面是一个简单的 Locust 用户示例,使用 Selenium 控制 Chrome 访问特定页面:
from locust import User, task
class GetUrl(User):
customdriver = None
def on_start(self):
self.customdriver = webdriver.Chrome(options=get_chrome_options())
@task
def load_page(self):
self.customdriver.get("http://example.com") # 根据需要替换为实际 URL
解释
- 用户定义 :创建一个继承自
User
的类,使用 Selenium 控制 Chrome。 - 启动时操作 :在用户启动时初始化
customdriver
。 - 任务定义 :在
load_page
方法中执行实际的页面加载操作。
4. Kubernetes 部署
完成 Dockerfile 和代码后,可以将其构建为 Docker 镜像,并在 Kubernetes 中部署。以下是一个基本的 Kubernetes YAML 配置示例:
apiVersion: apps/v1
kind: Deployment
metadata:
name: locust
spec:
replicas: 1
selector:
matchLabels:
app: locust
template:
metadata:
labels:
app: locust
spec:
containers:
- name: locust
image: your-docker-image:latest
ports:
- containerPort: 8089
env:
- name: XDG_CONFIG_HOME
value: /tmp/.chromium # 设置 Chrome 的配置目录
- name: XDG_CACHE_HOME
value: /tmp/.chromium # 设置 Chrome 的缓存目录
---
apiVersion: v1
kind: Service
metadata:
name: locust-service
spec:
type: NodePort
ports:
- port: 8089
targetPort: 8089
selector:
app: locust
解释
- Deployment:定义 Locust 的 Deployment,指定容器镜像和服务端口。
- Service:创建一个 Service,使外部能够访问 Locust Web 界面。
结论
通过以上步骤,我们成功在 Kubernetes 中运行了 Locust 和 Selenium,并安装了 Chrome 和 ChromeDriver。确保配置正确的环境变量和 Chrome 选项,可以大大提高在 Kubernetes 环境中的稳定性。如果您有更多需求,可以根据项目的具体情况进行扩展和调整。