在 Kubernetes 中运行 Locust 与 Selenium：安装 Chrome 和 ChromeDriver ...

在现代软件开发中，性能和用户体验是至关重要的，而负载测试和自动化测试可以资助我们实现这一目标。在本文中，我们将讨论怎样在 Kubernetes 环境中运行 Locust 和 Selenium，并具体先容怎样安装 Chrome 和 ChromeDriver。
1. Dockerfile 配置

首先，我们必要创建一个 Dockerfile，以构建一个包罗 Locust 和 Selenium 的 Docker 镜像。以下是 Dockerfile 的内容：

1. FROM locustio/locust:2.31.3
3. # 设置 Chrome 的版本
4. ENV CHROME_VERSION 130.0.6723.69
6. USER root
7. RUN apt-get update -qq -y && \
8.     apt-get install -y wget unzip && \
9.     apt-get install -y \
10.         libasound2 \
11.         libatk-bridge2.0-0 \
12.         libgtk-4-1 \
13.         libnss3 \
14.         xdg-utils && \
15.     wget -q -O chrome-linux64.zip https://storage.googleapis.com/chrome-for-testing-public/$CHROME_VERSION/linux64/chrome-linux64.zip && \
16.     unzip chrome-linux64.zip && \
17.     rm chrome-linux64.zip && \
18.     mv chrome-linux64 /opt/chrome/ && \
19.     ln -s /opt/chrome/chrome /usr/local/bin/ && \
20.     wget -q -O chromedriver-linux64.zip https://storage.googleapis.com/chrome-for-testing-public/$CHROME_VERSION/linux64/chromedriver-linux64.zip && \
21.     unzip -j chromedriver-linux64.zip chromedriver-linux64/chromedriver && \
22.     rm chromedriver-linux64.zip && \
23.     mv chromedriver /usr/local/bin/
25. # 设置 Chrome 的配置和缓存目录
26. ENV XDG_CONFIG_HOME=/tmp/.chromium
27. ENV XDG_CACHE_HOME=/tmp/.chromium
29. COPY . .
30. RUN pip install -r requirements.txt

复制代码

表明

• 基础镜像：利用 locustio/locust 作为基础镜像。
  
• 安装依赖：更新包管理器并安装必要的库，以确保 Chrome 和 ChromeDriver 正常运行。
  
• 下载和安装 Chrome 和 ChromeDriver：从 Google 的存储库下载 Chrome 和 ChromeDriver。
  
• 配置目录：通过环境变量设置 Chrome 的配置和缓存目录，这一步非常重要。若未设置正确，可能会在 Kubernetes 中出现权限问题，具体参考 puppeteer-sharp
  

requirements.txt 示例

1. locust=2.31.3
2. selenium==4.21.0

复制代码

2. Chrome 选项配置

在利用 Selenium 时，我们必要为 Chrome 配置一些选项，以确保它能够在无头模式下正常工作。以下是获取 Chrome 选项的代码示例：

1. import platform
2. from selenium import webdriver
4. def is_running_in_linux():
5.     return platform.system() == 'Linux'
7. def get_chrome_options():
8.     is_in_linux = is_running_in_linux()
9.     options_custom = webdriver.ChromeOptions()
11.     # Linux 下的 Chrome 选项
12.     if is_in_linux:
13.         options_custom.add_argument("--headless")  # 无头模式
14.         options_custom.add_argument('--disable-gpu')  # 禁用 GPU 加速
15.         options_custom.add_argument("--no-sandbox")  # 禁用沙箱模式
16.     else:
17.         options_custom.add_argument("--start-maximized")  # 启动时最大化窗口
19.     # 其他通用选项
20.     options_custom.add_argument("--disable-dev-shm-usage")  # 解决资源限制问题
21.     options_custom.add_argument("--ignore-ssl-errors=yes")  # 忽略 SSL 错误
22.     options_custom.add_argument("--disable-cache")  # 禁用缓存
24.     return options_custom

复制代码

表明

• 操作系统检测：根据当前操作系统选择得当的 Chrome 选项。
  
• 无头模式：在 Linux 环境中利用无头模式，以便在没有图形界面的环境下运行 Chrome。
  
• 禁用沙箱：在 Kubernetes 环境中，禁用沙箱模式可以避免潜在的权限问题。
  

3. Locust 用户定义

下面是一个简单的 Locust 用户示例，利用 Selenium 控制 Chrome 访问特定页面：

1. from locust import User, task
3. class GetUrl(User):   
4.     customdriver = None  
6.     def on_start(self):               
7.         self.customdriver = webdriver.Chrome(options=get_chrome_options())
9.     @task
10.     def load_page(self):
11.         self.customdriver.get("http://example.com")  # 根据需要替换为实际 URL

复制代码

表明

• 用户定义：创建一个继承自 User 的类，利用 Selenium 控制 Chrome。
  
• 启动时操作：在用户启动时初始化 customdriver。
  
• 任务定义：在 load_page 方法中执行实际的页面加载操作。
  

4. Kubernetes 部署

完成 Dockerfile 和代码后，可以将其构建为 Docker 镜像，并在 Kubernetes 中部署。以下是一个基本的 Kubernetes YAML 配置示例：

1. apiVersion: apps/v1
2. kind: Deployment
3. metadata:
4.   name: locust
5. spec:
6.   replicas: 1
7.   selector:
8.     matchLabels:
9.       app: locust
10.   template:
11.     metadata:
12.       labels:
13.         app: locust
14.     spec:
15.       containers:
16.         - name: locust
17.           image: your-docker-image:latest
18.           ports:
19.             - containerPort: 8089
20.           env:
21.             - name: XDG_CONFIG_HOME
22.               value: /tmp/.chromium  # 设置 Chrome 的配置目录
23.             - name: XDG_CACHE_HOME
24.               value: /tmp/.chromium  # 设置 Chrome 的缓存目录
25. ---
26. apiVersion: v1
27. kind: Service
28. metadata:
29.   name: locust-service
30. spec:
31.   type: NodePort
32.   ports:
33.     - port: 8089
34.       targetPort: 8089
35.   selector:
36.     app: locust

复制代码

表明

• Deployment：定义 Locust 的 Deployment，指定容器镜像和服务端口。
  
• Service：创建一个 Service，使外部能够访问 Locust Web 界面。
  

结论

通过以上步骤，我们乐成在 Kubernetes 中运行了 Locust 和 Selenium，并安装了 Chrome 和 ChromeDriver。确保配置正确的环境变量和 Chrome 选项，可以大大提高在 Kubernetes 环境中的稳定性。如果您有更多需求，可以根据项目标具体环境进行扩展和调整。

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。