🐛 fix(flow.py): change flow field type from str to Dict to allow for JSON data
✨ feat(flow.py): add validator to ensure flow field is a valid JSON object with required fields
The flow field in the FlowBase model has been changed from a string to a dictionary to allow for JSON data. A validator has been added to ensure that the flow field is a valid JSON object with the required fields. The tests have been updated to reflect these changes.
This commit is contained in:
Gabriel Luiz Freitas Almeida
parent
5a46c1e1a0
commit
ec585b8acc
3 changed files with 126 additions and 38 deletions
|
|
@ -1,25 +1,29 @@
|
|||
from uuid import uuid4
|
||||
from langflow.api.schemas import FlowListCreate
|
||||
from langflow.database.models.flow import FlowCreate
|
||||
import json
|
||||
from sqlalchemy.orm import Session
|
||||
from langflow.database.models.flow import Flow
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
import threading
|
||||
|
||||
|
||||
def test_create_flow(client, json_flow):
|
||||
flow = FlowCreate(name="Test Flow", flow=json_flow)
|
||||
def test_create_flow(client: TestClient, json_flow: str):
|
||||
flow = FlowCreate(name="Test Flow", flow=json.loads(json_flow))
|
||||
response = client.post("/flows/", json=flow.dict())
|
||||
assert response.status_code == 200
|
||||
assert response.json()["name"] == flow.name
|
||||
assert response.json()["flow"] == flow.flow
|
||||
|
||||
|
||||
def test_read_flows(client, json_flow):
|
||||
flow = FlowCreate(name="Test Flow", flow=json_flow)
|
||||
def test_read_flows(client: TestClient, json_flow: str):
|
||||
flow = FlowCreate(name="Test Flow", flow=json.loads(json_flow))
|
||||
response = client.post("/flows/", json=flow.dict())
|
||||
assert response.status_code == 200
|
||||
assert response.json()["name"] == flow.name
|
||||
assert response.json()["flow"] == flow.flow
|
||||
flow = FlowCreate(name="Test Flow", flow=json_flow)
|
||||
flow = FlowCreate(name="Test Flow", flow=json.loads(json_flow))
|
||||
response = client.post("/flows/", json=flow.dict())
|
||||
assert response.status_code == 200
|
||||
assert response.json()["name"] == flow.name
|
||||
|
|
@ -30,8 +34,8 @@ def test_read_flows(client, json_flow):
|
|||
assert len(response.json()) > 0
|
||||
|
||||
|
||||
def test_read_flow(client, json_flow):
|
||||
flow = FlowCreate(name="Test Flow", flow=json_flow)
|
||||
def test_read_flow(client: TestClient, json_flow: str):
|
||||
flow = FlowCreate(name="Test Flow", flow=json.loads(json_flow))
|
||||
response = client.post("/flows/", json=flow.dict())
|
||||
flow_id = response.json()["id"]
|
||||
response = client.get(f"/flows/{flow_id}")
|
||||
|
|
@ -40,12 +44,13 @@ def test_read_flow(client, json_flow):
|
|||
assert response.json()["flow"] == flow.flow
|
||||
|
||||
|
||||
def test_update_flow(client, json_flow):
|
||||
flow = FlowCreate(name="Test Flow", flow=json_flow)
|
||||
def test_update_flow(client: TestClient, json_flow: str):
|
||||
flow = FlowCreate(name="Test Flow", flow=json.loads(json_flow))
|
||||
response = client.post("/flows/", json=flow.dict())
|
||||
flow_id = response.json()["id"]
|
||||
updated_flow = FlowCreate(
|
||||
name="Updated Flow", flow=json_flow.replace("BasicExample", "Updated Flow")
|
||||
name="Updated Flow",
|
||||
flow=json.loads(json_flow.replace("BasicExample", "Updated Flow")),
|
||||
)
|
||||
response = client.put(f"/flows/{flow_id}", json=updated_flow.dict())
|
||||
assert response.status_code == 200
|
||||
|
|
@ -53,8 +58,8 @@ def test_update_flow(client, json_flow):
|
|||
assert response.json()["flow"] == updated_flow.flow
|
||||
|
||||
|
||||
def test_delete_flow(client, json_flow):
|
||||
flow = FlowCreate(name="Test Flow", flow=json_flow)
|
||||
def test_delete_flow(client: TestClient, json_flow: str):
|
||||
flow = FlowCreate(name="Test Flow", flow=json.loads(json_flow))
|
||||
response = client.post("/flows/", json=flow.dict())
|
||||
flow_id = response.json()["id"]
|
||||
response = client.delete(f"/flows/{flow_id}")
|
||||
|
|
@ -62,12 +67,12 @@ def test_delete_flow(client, json_flow):
|
|||
assert response.json()["message"] == "Flow deleted successfully"
|
||||
|
||||
|
||||
def test_create_flows(client, session: Session):
|
||||
def test_create_flows(client: TestClient, session: Session, json_flow: str):
|
||||
# Create test data
|
||||
flow_list = FlowListCreate(
|
||||
flows=[
|
||||
FlowCreate(name="Flow 1", flow="Test flow 1"),
|
||||
FlowCreate(name="Flow 2", flow="Test flow 2"),
|
||||
FlowCreate(name="Flow 1", flow=json.loads(json_flow)),
|
||||
FlowCreate(name="Flow 2", flow=json.loads(json_flow)),
|
||||
]
|
||||
)
|
||||
# Make request to endpoint
|
||||
|
|
@ -78,26 +83,20 @@ def test_create_flows(client, session: Session):
|
|||
response_data = response.json()
|
||||
assert len(response_data) == 2
|
||||
assert response_data[0]["name"] == "Flow 1"
|
||||
assert response_data[0]["flow"] == "Test flow 1"
|
||||
assert response_data[0]["flow"] == json.loads(json_flow)
|
||||
assert response_data[1]["name"] == "Flow 2"
|
||||
assert response_data[1]["flow"] == "Test flow 2"
|
||||
assert response_data[1]["flow"] == json.loads(json_flow)
|
||||
|
||||
|
||||
def test_upload_file(client, session: Session):
|
||||
def test_upload_file(client: TestClient, session: Session, json_flow: str):
|
||||
# Create test data
|
||||
flow_list = FlowListCreate(
|
||||
flows=[
|
||||
FlowCreate(name="Flow 1", flow="Test flow 1"),
|
||||
FlowCreate(name="Flow 2", flow="Test flow 2"),
|
||||
FlowCreate(name="Flow 1", flow=json.loads(json_flow)),
|
||||
FlowCreate(name="Flow 2", flow=json.loads(json_flow)),
|
||||
]
|
||||
)
|
||||
file_contents = json.dumps(flow_list.dict())
|
||||
# Make request to endpoint
|
||||
# curl -X 'POST' \
|
||||
# 'http://127.0.0.1:7860/flows/upload/' \
|
||||
# -H 'accept: application/json' \
|
||||
# -H 'Content-Type: multipart/form-data' \
|
||||
# -F 'file=@examples.json;type=application/json'
|
||||
response = client.post(
|
||||
"/flows/upload/",
|
||||
files={"file": ("examples.json", file_contents, "application/json")},
|
||||
|
|
@ -108,17 +107,17 @@ def test_upload_file(client, session: Session):
|
|||
response_data = response.json()
|
||||
assert len(response_data) == 2
|
||||
assert response_data[0]["name"] == "Flow 1"
|
||||
assert response_data[0]["flow"] == "Test flow 1"
|
||||
assert response_data[0]["flow"] == json.loads(json_flow)
|
||||
assert response_data[1]["name"] == "Flow 2"
|
||||
assert response_data[1]["flow"] == "Test flow 2"
|
||||
assert response_data[1]["flow"] == json.loads(json_flow)
|
||||
|
||||
|
||||
def test_download_file(client, session: Session):
|
||||
def test_download_file(client: TestClient, session: Session, json_flow):
|
||||
# Create test data
|
||||
flow_list = FlowListCreate(
|
||||
flows=[
|
||||
FlowCreate(name="Flow 1", flow="Test flow 1"),
|
||||
FlowCreate(name="Flow 2", flow="Test flow 2"),
|
||||
FlowCreate(name="Flow 1", flow=json.loads(json_flow)),
|
||||
FlowCreate(name="Flow 2", flow=json.loads(json_flow)),
|
||||
]
|
||||
)
|
||||
for flow in flow_list.flows:
|
||||
|
|
@ -133,6 +132,68 @@ def test_download_file(client, session: Session):
|
|||
response_data = json.loads(response.json()["file"])
|
||||
assert len(response_data) == 2
|
||||
assert response_data[0]["name"] == "Flow 1"
|
||||
assert response_data[0]["flow"] == "Test flow 1"
|
||||
assert response_data[0]["flow"] == json.loads(json_flow)
|
||||
assert response_data[1]["name"] == "Flow 2"
|
||||
assert response_data[1]["flow"] == "Test flow 2"
|
||||
assert response_data[1]["flow"] == json.loads(json_flow)
|
||||
|
||||
|
||||
def test_create_flow_with_invalid_data(client: TestClient):
|
||||
flow = {"name": "a" * 256, "flow": "Invalid flow data"}
|
||||
response = client.post("/flows/", json=flow)
|
||||
assert response.status_code == 422
|
||||
|
||||
|
||||
def test_get_nonexistent_flow(client: TestClient):
|
||||
# uuid4 generates a random UUID
|
||||
uuid = uuid4()
|
||||
response = client.get(f"/flows/{uuid}")
|
||||
assert response.status_code == 404
|
||||
|
||||
|
||||
def test_update_flow_idempotency(client: TestClient, json_flow: str):
|
||||
flow = FlowCreate(name="Test Flow", flow=json.loads(json_flow))
|
||||
response = client.post("/flows/", json=flow.dict())
|
||||
flow_id = response.json()["id"]
|
||||
updated_flow = FlowCreate(name="Updated Flow", flow=json.loads(json_flow))
|
||||
response1 = client.put(f"/flows/{flow_id}", json=updated_flow.dict())
|
||||
response2 = client.put(f"/flows/{flow_id}", json=updated_flow.dict())
|
||||
assert response1.json() == response2.json()
|
||||
|
||||
|
||||
def test_update_nonexistent_flow(client: TestClient, json_flow: str):
|
||||
uuid = uuid4()
|
||||
updated_flow = FlowCreate(
|
||||
name="Updated Flow",
|
||||
flow=json.loads(json_flow.replace("BasicExample", "Updated Flow")),
|
||||
)
|
||||
response = client.put(f"/flows/{uuid}", json=updated_flow.dict())
|
||||
assert response.status_code == 404
|
||||
|
||||
|
||||
def test_delete_nonexistent_flow(client: TestClient):
|
||||
uuid = uuid4()
|
||||
response = client.delete(f"/flows/{uuid}")
|
||||
assert response.status_code == 404
|
||||
|
||||
|
||||
def test_read_empty_flows(client: TestClient):
|
||||
response = client.get("/flows/")
|
||||
assert response.status_code == 200
|
||||
assert len(response.json()) == 0
|
||||
|
||||
|
||||
def test_stress_create_flow(client: TestClient, json_flow: str):
|
||||
flow = FlowCreate(name="Test Flow", flow=json.loads(json_flow))
|
||||
|
||||
def create_flow():
|
||||
response = client.post("/flows/", json=flow.dict())
|
||||
assert response.status_code == 200
|
||||
|
||||
threads = []
|
||||
for i in range(100):
|
||||
t = threading.Thread(target=create_flow)
|
||||
threads.append(t)
|
||||
t.start()
|
||||
|
||||
for t in threads:
|
||||
t.join()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue