Remove unused test files
This commit is contained in:
parent
d777df3dbf
commit
af5933524a
3 changed files with 0 additions and 121 deletions
|
|
@ -1,59 +0,0 @@
|
|||
from langflow.template.field.base import TemplateField
|
||||
from langflow.template.frontend_node.embeddings import EmbeddingFrontendNode
|
||||
|
||||
|
||||
def test_format_jina_fields():
|
||||
field = TemplateField(name="jina")
|
||||
EmbeddingFrontendNode.format_jina_fields(field)
|
||||
assert field.show is True
|
||||
assert field.advanced is False
|
||||
|
||||
field = TemplateField(name="auth")
|
||||
EmbeddingFrontendNode.format_jina_fields(field)
|
||||
assert field.password is True
|
||||
assert field.show is True
|
||||
assert field.advanced is False
|
||||
|
||||
field = TemplateField(name="jina_api_url")
|
||||
EmbeddingFrontendNode.format_jina_fields(field)
|
||||
assert field.show is True
|
||||
assert field.advanced is True
|
||||
assert field.display_name == "Jina API URL"
|
||||
assert field.password is False
|
||||
|
||||
|
||||
def test_format_openai_fields():
|
||||
field = TemplateField(name="openai")
|
||||
EmbeddingFrontendNode.format_openai_fields(field)
|
||||
assert field.show is True
|
||||
assert field.advanced is True
|
||||
assert field.display_name == "OpenAI"
|
||||
|
||||
field = TemplateField(name="openai_api_key")
|
||||
EmbeddingFrontendNode.format_openai_fields(field)
|
||||
assert field.password is True
|
||||
assert field.show is True
|
||||
assert field.advanced is False
|
||||
|
||||
|
||||
def test_format_field():
|
||||
field = TemplateField(name="headers")
|
||||
EmbeddingFrontendNode.format_field(field)
|
||||
assert field.show is False
|
||||
|
||||
field = TemplateField(name="jina")
|
||||
EmbeddingFrontendNode.format_field(field)
|
||||
assert field.advanced is False
|
||||
assert field.show is True
|
||||
|
||||
field = TemplateField(name="openai")
|
||||
EmbeddingFrontendNode.format_field(field)
|
||||
assert field.advanced is True
|
||||
assert field.show is True
|
||||
assert field.display_name == "OpenAI"
|
||||
|
||||
field = TemplateField(name="test_field", required=True)
|
||||
EmbeddingFrontendNode.format_field(field)
|
||||
assert field.advanced is False
|
||||
assert field.show is True
|
||||
assert field.required is True
|
||||
|
|
@ -1,12 +0,0 @@
|
|||
from fastapi.testclient import TestClient
|
||||
|
||||
from langflow.services.deps import get_settings_service
|
||||
|
||||
|
||||
def test_prompts_settings(client: TestClient, logged_in_headers):
|
||||
settings_service = get_settings_service()
|
||||
response = client.get("api/v1/all", headers=logged_in_headers)
|
||||
assert response.status_code == 200
|
||||
json_response = response.json()
|
||||
prompts = json_response["prompts"]
|
||||
assert set(prompts.keys()) == set(settings_service.settings.PROMPTS)
|
||||
|
|
@ -1,50 +0,0 @@
|
|||
from fastapi import WebSocketDisconnect
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
# from langflow.services.chat.manager import ChatService
|
||||
|
||||
import pytest
|
||||
|
||||
|
||||
def test_init_build(client, active_user, logged_in_headers):
|
||||
response = client.post(
|
||||
"api/v1/build/init/test",
|
||||
json={"id": "test", "data": {"key": "value"}},
|
||||
headers=logged_in_headers,
|
||||
)
|
||||
assert response.status_code == 201
|
||||
assert response.json() == {"flowId": "test"}
|
||||
|
||||
|
||||
# def test_stream_build(client):
|
||||
# client.post(
|
||||
# "api/v1/build/init", json={"id": "stream_test", "data": {"key": "value"}}
|
||||
# )
|
||||
|
||||
# # Test the stream
|
||||
# response = client.get("api/v1/build/stream/stream_test")
|
||||
# assert response.status_code == 200
|
||||
# assert response.headers["content-type"] == "text/event-stream; charset=utf-8"
|
||||
|
||||
|
||||
def test_websocket_endpoint(client: TestClient, active_user, logged_in_headers):
|
||||
# Assuming your websocket_endpoint uses chat_service which caches data from stream_build
|
||||
access_token = logged_in_headers["Authorization"].split(" ")[1]
|
||||
with pytest.raises(WebSocketDisconnect):
|
||||
with client.websocket_connect(f"api/v1/chat/non_existing_client_id?token={access_token}") as websocket:
|
||||
websocket.send_json({"type": "test"})
|
||||
data = websocket.receive_json()
|
||||
assert "Please, build the flow before sending messages" in data["message"]
|
||||
|
||||
|
||||
def test_websocket_endpoint_after_build(client, basic_graph_data):
|
||||
# Assuming your websocket_endpoint uses chat_service which caches data from stream_build
|
||||
client.post("api/v1/build/init", json=basic_graph_data)
|
||||
client.get("api/v1/build/stream/websocket_test")
|
||||
|
||||
# There should be more to test here, but it depends on the inner workings of your websocket handler
|
||||
# and how your chat_service and other classes behave. The following is just an example structure.
|
||||
with pytest.raises(WebSocketDisconnect):
|
||||
with client.websocket_connect("api/v1/chat/websocket_test") as websocket:
|
||||
websocket.send_json({"input": "test"})
|
||||
websocket.receive_json()
|
||||
Loading…
Add table
Add a link
Reference in a new issue