feat: code validation, endpoint and tests

tests/test_endpoints.py

@@ -1,5 +1,7 @@
 from langflow.interface.listing import CUSTOM_TOOLS
+from langflow.utils.constants import DEFAULT_PYTHON_FUNCTION
 from fastapi.testclient import TestClient
+import pytest
 
 
 def test_get_all(client: TestClient):
@@ -10,3 +12,67 @@ def test_get_all(client: TestClient):
     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())
+
+
+def test_post_validate_code(client: TestClient):
+    # Test case with a valid import and function
+    code1 = """
+import math
+
+def square(x):
+    return x ** 2
+"""
+    response1 = client.post("/validate", json={"code": code1})
+    assert response1.status_code == 200
+    assert response1.json() == {"imports": {"errors": []}, "function": {"errors": []}}
+
+    # Test case with an invalid import and valid function
+    code2 = """
+import non_existent_module
+
+def square(x):
+    return x ** 2
+"""
+    response2 = client.post("/validate", json={"code": code2})
+    assert response2.status_code == 200
+    assert response2.json() == {
+        "imports": {"errors": ["No module named 'non_existent_module'"]},
+        "function": {"errors": []},
+    }
+
+    # Test case with a valid import and invalid function syntax
+    code3 = """
+import math
+
+def square(x)
+    return x ** 2
+"""
+    response3 = client.post("/validate", json={"code": code3})
+    assert response3.status_code == 200
+    assert response3.json() == {
+        "imports": {"errors": []},
+        "function": {"errors": ["expected ':' (<unknown>, line 4)"]},
+    }
+
+    # Test case with invalid JSON payload
+    response4 = client.post("/validate", json={"invalid_key": code1})
+    assert response4.status_code == 422
+
+    # Test case with an empty code string
+    response5 = client.post("/validate", json={"code": ""})
+    assert response5.status_code == 200
+    assert response5.json() == {"imports": {"errors": []}, "function": {"errors": []}}
+
+    # Test case with a syntax error in the code
+    code6 = """
+import math
+
+def square(x)
+    return x ** 2
+"""
+    response6 = client.post("/validate", json={"code": code6})
+    assert response6.status_code == 200
+    assert response6.json() == {
+        "imports": {"errors": []},
+        "function": {"errors": ["expected ':' (<unknown>, line 4)"]},
+    }