Как создать задачу Locust для динамического списка URL?

Question

Я пытаюсь получить информацию о времени для всех моих конечных точек API на основе OpenAPI JSON, поэтому я хотел бы сделать что-то подобное в методе setup:

for api_path in self.client.get('/api/doc/?format=openapi').json()['paths'].keys():
    self.get_[api_path] = [@task decorated method which retrieves the API path]

Или есть другой способ генерировать задачи во время выполнения?

Answer 1

Нашел в API документах :

def setup(self) -> None:
    for api_path in self.client.get('/api/doc/?format=openapi').json()['paths'].keys():
        self.schedule_task(self.api_task, args=[api_path])

def api_task(self, path: str) -> None:
    response = self.client.get("/api{}?limit=1000".format(path))
    assert response.status_code == HTTPStatus.OK, response.content.decode()