jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🐛 fix(locustfile.py): update host URL to include correct port number

Browse source 
✨ feat(locustfile.py): add x-api-key header to requests for authentication
🐛 fix(locustfile.py): provide default value for FLOW_ID environment variable to prevent errors
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-01 17:21:26 -03:00
commit b6fa053291
1 changed files with 13 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/locust/locustfile.py
View file

@ -6,18 +6,24 @@ import os
class NameTest(FastHttpUser): class NameTest(FastHttpUser):
host = "http://localhost/api/v1" # make sure the port number is correct host = "http://localhost:7860/api/v1" # make sure the port number is correct
wait_time = between(1, 5) wait_time = between(1, 5)
with open("names.txt", "r") as file: with open("names.txt", "r") as file:
names = [line.strip() for line in file.readlines()] names = [line.strip() for line in file.readlines()]
headers = {
# api-key
"x-api-key": "lf-a1EsRC75ybWiKRCRLyhX1R9rPlqQKYlnRQoZWysg6NM",
}
def poll_task(self, task_id, sleep_time=1): def poll_task(self, task_id, sleep_time=1):
while True: while True:
with self.rest( with self.rest(
"GET", "GET",
f"/task/{task_id}/status", f"/task/{task_id}/status",
name="task_status", name="task_status",
headers=self.headers,
) as response: ) as response:
status = response.js.get("status") status = response.js.get("status")
if status == "SUCCESS": if status == "SUCCESS":
@ -29,14 +35,18 @@ class NameTest(FastHttpUser):
@task @task
def send_name_and_check(self): def send_name_and_check(self):
name = random.choice(self.names) name = random.choice(self.names)
flow_id = os.getenv("FLOW_ID") flow_id = os.getenv("FLOW_ID", "88989f6d-8da4-4d6f-8205-480693c36200")
session_id = f"{name}-{time.time()}" session_id = f"{name}-{time.time()}"
def process(flow_id, payload): def process(flow_id, payload):
task_id = None task_id = None
print(f"Processing {payload}") print(f"Processing {payload}")
with self.rest( with self.rest(
"POST", f"/process/{flow_id}", json=payload, name="process" "POST",
f"/process/{flow_id}",
json=payload,
name="process",
headers=self.headers,
) as response: ) as response:
if response.status_code != 200: if response.status_code != 200:
response.failure("Process call failed") response.failure("Process call failed")