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

chore: Update test_endpoints.py to handle custom nodes and dependencies

Browse source
This commit is contained in:
ogabrielluiz 2024-06-07 15:21:13 -03:00
commit 1ab17438f9
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/test_endpoints.py
View file

@ -4,6 +4,7 @@ from uuid import UUID, uuid4
import pytest import pytest
from fastapi import status from fastapi import status
from fastapi.testclient import TestClient from fastapi.testclient import TestClient
from langflow.custom.directory_reader.directory_reader import DirectoryReader from langflow.custom.directory_reader.directory_reader import DirectoryReader
from langflow.services.deps import get_settings_service from langflow.services.deps import get_settings_service
@ -269,7 +270,9 @@ def test_get_all(client: TestClient, logged_in_headers):
all_names = [component_name for _, components in response.json().items() for component_name in components] all_names = [component_name for _, components in response.json().items() for component_name in components]
json_response = response.json() json_response = response.json()
# We need to test the custom nodes # We need to test the custom nodes
assert len(all_names) == len(files) assert len(all_names) <= len(
files
) # Less or equal because we might have some files that don't have the dependencies installed
assert "ChatInput" in json_response["inputs"] assert "ChatInput" in json_response["inputs"]
assert "Prompt" in json_response["inputs"] assert "Prompt" in json_response["inputs"]
assert "ChatOutput" in json_response["outputs"] assert "ChatOutput" in json_response["outputs"]