Блог

я хочу написать базовый класс пользователя http и форму базового нагрузочного теста, а затем расширить их в подклассах, но locust не понимает расширенные классы и устанавливает базовые классы. это базовые классы helpers.py:

 from locust.contrib.fasthttp import FastHttpUser
import  string
from locust import LoadTestShape, constant_pacing
from dotenv import load_dotenv
import os
load_dotenv()
# init parameters
host_address = "127.0.0.1"
class BaseHttpUser(FastHttpUser):
    host = host_address
    wait_time = constant_pacing(5)
    chars = string.ascii_uppercase   string.ascii_lowercase   string.digits
    start_time = 0

class BaseRps(LoadTestShape):
    time_limit = 600
    user_spawn = {1: (1500, 10)}

    def tick(self):
        step = len(self.user_spawn.keys())
        run_time = self.get_run_time()
        print(step, )
        for idx in range(1, step 1):
            print(run_time , idx , self.time_limit)
            if run_time < idx * self.time_limit / step:
                print("here", self.user_spawn.get(idx))
                return self.user_spawn.get(idx)
        return None
 

и это файл, который я запускаю
minio.py

 from locust import task
from helpers import BaseHttpUser, BaseRps
import os

host_address = "127.0.0.1"
test_name = "minio"
log_file_path = 'log.log'
base_url = os.getenv("MINIO_URL")

class HttpUser(BaseHttpUser):
    host = host_address
    base_url = base_url

    @task
    def download(self):
        self.client.get(f'{self.base_url}/magnix-server-media/ads-images/ff.png', name='download')


    
class Rps(BaseRps):
    user_spawn = {1: (10000, 100)} 
 

Базовым пользовательским классам требуется, чтобы атрибут abstract = True не создавался. https://docs.locust.io/en/stable/api.html#locust.User.abstract

Я не думаю, что вы можете сделать то же самое с классами load shape, но вы можете использовать атрибуты класса (которыми вы можете манипулировать в своем locustfile после его импорта)

Например, удаление класса Rps и вместо этого просто выполнение

 BaseRps.user_spawn = {1: (10000, 100)}