feat: code validation, endpoint and tests

tests/conftest.py

@@ -11,6 +11,11 @@ def pytest_configure():
         Path(__file__).parent.absolute() / "data" / "complex_example.json"
     )
 
+    pytest.CODE_WITH_SYNTAX_ERROR = """
+def get_text():
+    retun "Hello World"
+    """
+
 
 # Create client fixture for FastAPI
 @pytest.fixture(scope="module")

tests/test_custom_types.py

@@ -0,0 +1,19 @@
+# Test this:
+from typing import Optional, Callable
+from langflow.interface.custom_types import PythonFunction
+from langflow.utils import util, constants
+from pydantic import BaseModel, ValidationError, validator
+from langchain.agents import tool
+import pytest
+
+
+def test_python_function():
+    """Test Python function"""
+    func = PythonFunction(code=constants.DEFAULT_PYTHON_FUNCTION)
+    assert func.get_function()("text") == "text"
+    # the tool decorator should raise an error if
+    # the function is not str -> str
+
+    # This raises ValidationError
+    with pytest.raises(SyntaxError):
+        func = PythonFunction(code=pytest.CODE_WITH_SYNTAX_ERROR)

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)"]},
+    }

tests/test_validate_code.py

@@ -0,0 +1,39 @@
+from langflow.utils.validate import validate_code
+
+
+def test_validate_code():
+    # Test case with a valid import and function
+    code1 = """
+import math
+
+def square(x):
+    return x ** 2
+"""
+    errors1 = validate_code(code1)
+    assert errors1 == {"imports": {"errors": []}, "function": {"errors": []}}
+
+    # Test case with an invalid import and valid function
+    code2 = """
+import non_existent_module
+
+def square(x):
+    return x ** 2
+"""
+    errors2 = validate_code(code2)
+    assert errors2 == {
+        "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
+"""
+    errors3 = validate_code(code3)
+    assert errors3 == {
+        "imports": {"errors": []},
+        "function": {"errors": ["expected ':' (<unknown>, line 4)"]},
+    }