From af5933524a62f4d40cdc0b6d93366b783cdd2978 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Fri, 8 Mar 2024 10:23:43 -0300
Subject: [PATCH] Remove unused test files

---
 tests/test_embeddings_template.py | 59 -------------------------------
 tests/test_prompts_template.py    | 12 -------
 tests/test_websocket.py           | 50 --------------------------
 3 files changed, 121 deletions(-)
 delete mode 100644 tests/test_embeddings_template.py
 delete mode 100644 tests/test_prompts_template.py
 delete mode 100644 tests/test_websocket.py

diff --git a/tests/test_embeddings_template.py b/tests/test_embeddings_template.py
deleted file mode 100644
index 7334c2abd..000000000
--- a/tests/test_embeddings_template.py
+++ /dev/null
@@ -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
diff --git a/tests/test_prompts_template.py b/tests/test_prompts_template.py
deleted file mode 100644
index 06989cad0..000000000
--- a/tests/test_prompts_template.py
+++ /dev/null
@@ -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)
diff --git a/tests/test_websocket.py b/tests/test_websocket.py
deleted file mode 100644
index c4c9ee322..000000000
--- a/tests/test_websocket.py
+++ /dev/null
@@ -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()