саранча игнорирует расширенные подклассы и создает экземпляры базовых классов

я хочу написать базовый пользовательский класс http и форму теста базовой нагрузки, а затем расширить их в подклассах, но саранча не понимает расширенные классы и не устанавливает базовые классы, это базовые классы 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)}

python oop locust
person S.Saghari    schedule 06.12.2020    source источник

Ответы (1)


arrow_upward
0
arrow_downward

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

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

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

BaseRps.user_spawn = {1: (10000, 100)}
person Cyberwiz    schedule 08.12.2020