dragons_info_load_test.py

from locust import task, constant
from locust.contrib.fasthttp import FastHttpUser
from data.dragons_info import query_dragons_info
from common.load_test_custom_shape import MyCustomShape
from common.commons import processResponse, setRampConfig, setKPIConfig
from testConfig.dragon_info_test_config import DRAGON_INFO_KPIS, DRAGON_INFO_RAMP
import os
import gevent
import __init__

class DragonsInfoUser(FastHttpUser):
    wait_time = constant(10)
    basePath = "/"
    setRampConfig(DRAGON_INFO_RAMP)
    setKPIConfig(DRAGON_INFO_KPIS)
    def on_start(self):
        return super().on_start()        
    
    def get_dragons_info(self):
      with self.client.post(self.basePath, json={'query': query_dragons_info} ,catch_response=True, name="Get Dragons Info") as response:
        processResponse(response)
    
    @task
    def requests(self):
      gevent.spawn(self.get_dragons_info())