在现代软件开发中，性能和用户体验是至关重要的，而负载测试和自动化测试可以帮助我们实现这一目标。在本文中，我们将讨论如何在 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

USERrootRUNapt-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 的配置和缓存目录ENVXDG_CONFIG_HOME=/tmp/.chromiumENVXDG_CACHE_HOME=/tmp/.chromiumCOPY. .RUNpip install -r requirements.txt

解释

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

requirements.txt 示例

locust=2.31.3selenium==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:1selector:

    matchLabels:

      app: locust

  template:

    metadata:

      labels:

        app: locust

    spec:

      containers:

        - name: locust

          image: your-docker-image:latest

          ports:

            - containerPort:8089env:

            - 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:8089targetPort:8089selector:

    app: locust

解释

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

结论

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

标签: none

添加新评论