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

🐛 fix(test_database.py): remove unused imports to improve code readability

Browse source 
✨ feat(test_database.py): add support for session management using session_getter to improve code organization and maintainability
🐛 fix(test_endpoints.py): remove unused imports to improve code readability
✨ feat(test_endpoints.py): add support for session management using session_getter to improve code organization and maintainability
🐛 fix(test_login.py): remove unused imports to improve code readability
✨ feat(test_login.py): add support for session management using session_getter to improve code organization and maintainability
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-20 18:40:19 -03:00
commit 20e14d49b4
3 changed files with 26 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

22
tests/test_database.py
View file

@ -1,4 +1,6 @@
from langflow.services.database.models.base import orjson_dumps
from langflow.services.database.utils import session_getter
from langflow.services.getters import get_db_manager
import orjson
import pytest
@ -178,9 +180,7 @@ def test_upload_file(
assert response_data[1]["data"] == data
def test_download_file(
client: TestClient, session: Session, json_flow, active_user, logged_in_headers
):
def test_download_file(client: TestClient, json_flow, active_user, logged_in_headers):
flow = orjson.loads(json_flow)
data = flow["data"]
# Create test data
@ -190,18 +190,20 @@ def test_download_file(
FlowCreate(name="Flow 2", description="description", data=data),
]
)
for flow in flow_list.flows:
flow.user_id = active_user.id
db_flow = Flow.from_orm(flow)
session.add(db_flow)
session.commit()
db_manager = get_db_manager()
with session_getter(db_manager) as session:
for flow in flow_list.flows:
flow.user_id = active_user.id
db_flow = Flow.from_orm(flow)
session.add(db_flow)
session.commit()
# Make request to endpoint
response = client.get("api/v1/flows/download/", headers=logged_in_headers)
# Check response status code
assert response.status_code == 200
assert response.status_code == 200, response.json()
# Check response data
response_data = response.json()["flows"]
assert len(response_data) == 2
assert len(response_data) == 2, response_data
assert response_data[0]["name"] == "Flow 1"
assert response_data[0]["description"] == "description"
assert response_data[0]["data"] == data