feat: new tests for endpoints
This commit is contained in:
parent
d93ecee876
commit
287ef0d4e4
3 changed files with 36 additions and 2 deletions
22
tests/test_endpoints.py
Normal file
22
tests/test_endpoints.py
Normal file
|
|
@ -0,0 +1,22 @@
|
|||
# # build router
|
||||
# router = APIRouter()
|
||||
|
||||
|
||||
# @router.get("/all")
|
||||
# def get_all():
|
||||
# return build_langchain_types_dict()
|
||||
|
||||
|
||||
# Buil test for /all endpoint
|
||||
from langflow.utils.constants import CUSTOM_TOOLS
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
|
||||
def test_get_all(client: TestClient):
|
||||
response = client.get("/all")
|
||||
assert response.status_code == 200
|
||||
json_response = response.json()
|
||||
# We need to test the custom nodes
|
||||
assert "ZeroShotPrompt" in json_response["prompts"]
|
||||
# All CUSTOM_TOOLS(dict) should be in the response
|
||||
assert all(tool in json_response["tools"] for tool in CUSTOM_TOOLS.keys())
|
||||
Loading…
Add table
Add a link
Reference in a new issue