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

tests: add tests for streaming calls on api/v1/run (#7385)

Browse source 
* test: add concurrent streaming request tests for chat input type

Implemented a new test for concurrent streaming requests to the run endpoint with chat input type. Added a helper coroutine to validate the streaming response, ensuring proper event handling and result verification. This enhances the test coverage for the streaming functionality.

* refactor: replace session_getter with session_scope in API key CRUD operations

Updated the API key CRUD operations to utilize session_scope instead of session_getter for better session management. This change enhances the clarity and robustness of the database interactions.

* test: enhance assertions and error handling in streaming tests

Refactored assertions in the streaming tests to provide clearer error messages and improve robustness. Added error handling for JSON parsing in the stream response and ensured that all expected fields are validated with informative messages. Updated the test for concurrent streaming requests to use the correct project ID and modified input values for better clarity.

* test: refactor get_starter_project fixture for improved session management and data handling

Updated the `get_starter_project` fixture to use `session_scope` for better session management. Enhanced the flow data processing by replacing the OpenAI API key and ensuring the `load_from_db` flag is set to false, improving robustness and clarity in test setup.

* [autofix.ci] apply automated fixes

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Gabriel Luiz Freitas Almeida 2025-04-01 14:31:26 -03:00 • committed by GitHub
commit 7a87880e49
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 127 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

20
src/backend/tests/conftest.py
View file

@ -29,7 +29,7 @@ from langflow.services.database.models.transactions.model import TransactionTabl
from langflow.services.database.models.user.model import User, UserCreate, UserRead
from langflow.services.database.models.vertex_builds.crud import delete_vertex_builds_by_flow_id
from langflow.services.database.utils import session_getter
from langflow.services.deps import get_db_service
from langflow.services.deps import get_db_service, session_scope
from loguru import logger
from sqlalchemy.ext.asyncio import create_async_engine
from sqlalchemy.orm import selectinload
@ -655,13 +655,13 @@ async def get_simple_api_test(client, logged_in_headers, json_simple_api_test):
@pytest.fixture(name="starter_project")
async def get_starter_project(active_user):
async def get_starter_project(client, active_user): # noqa: ARG001
# once the client is created, we can get the starter project
async with session_getter(get_db_service()) as session:
async with session_scope() as session:
stmt = (
select(Flow)
.where(Flow.folder.has(Folder.name == STARTER_FOLDER_NAME))
.where(Flow.name == "Basic Prompting (Hello, World)")
.where(Flow.name == "Basic Prompting")
)
flow = (await session.exec(stmt)).first()
if not flow:
@ -669,7 +669,17 @@ async def get_starter_project(active_user):
raise ValueError(msg)
# ensure openai api key is set
get_openai_api_key()
openai_api_key = get_openai_api_key()
data_as_json = json.dumps(flow.data)
data_as_json = data_as_json.replace("OPENAI_API_KEY", openai_api_key)
# also replace `"load_from_db": true` with `"load_from_db": false`
if '"load_from_db": true' in data_as_json:
data_as_json = data_as_json.replace('"load_from_db": true', '"load_from_db": false')
if '"load_from_db": true' in data_as_json:
msg = "load_from_db should be false"
raise ValueError(msg)
flow.data = json.loads(data_as_json)
new_flow_create = FlowCreate(
name=flow.name,
description=flow.description,