From 71473fcec6613d820624ff47ef504bf068ac38ed Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 28 Nov 2023 14:35:50 -0300 Subject: [PATCH] Refactor test_endpoints.py file --- tests/test_endpoints.py | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) diff --git a/tests/test_endpoints.py b/tests/test_endpoints.py index aacc009d0..d958f3aec 100644 --- a/tests/test_endpoints.py +++ b/tests/test_endpoints.py @@ -1,18 +1,17 @@ -from collections import namedtuple +import time import uuid -from langflow.processing.process import Result -from langflow.services.auth.utils import get_password_hash -from langflow.services.database.models.api_key.model import ApiKey -from langflow.services.deps import get_settings_service -from langflow.services.database.utils import session_getter -from langflow.services.deps import get_db_service +from collections import namedtuple + import pytest from fastapi.testclient import TestClient from langflow.interface.tools.constants import CUSTOM_TOOLS +from langflow.processing.process import Result +from langflow.services.auth.utils import get_password_hash +from langflow.services.database.models.api_key.model import ApiKey +from langflow.services.database.utils import session_getter +from langflow.services.deps import get_db_service, get_settings_service from langflow.template.frontend_node.chains import TimeTravelGuideChainNode -import time - def run_post(client, flow_id, headers, post_data): response = client.post( @@ -25,7 +24,7 @@ def run_post(client, flow_id, headers, post_data): # Helper function to poll task status -def poll_task_status(client, headers, href, max_attempts=20, sleep_time=1): +def poll_task_status(client, headers, href, max_attempts=20, sleep_time=3): for _ in range(max_attempts): task_status_response = client.get( href,