12 lines
472 B
Python
12 lines
472 B
Python
from langflow.interface.listing 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())
|