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

fix: auth current user on code validation (#6911)

Browse source 
* Add user auth for /code endpoint

* import

* revert

* [autofix.ci] apply automated fixes

* [autofix.ci] apply automated fixes (attempt 2/3)

* ruff

* fix tests

* [autofix.ci] apply automated fixes

* ruff

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Nadir J <31660040+NadirJ@users.noreply.github.com>
This commit is contained in:
Jordan Frazier 2025-03-05 09:04:03 -08:00 • committed by GitHub
commit faac4db133
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 36 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

30
src/backend/tests/unit/api/v1/test_validate.py
View file

@ -1,14 +1,16 @@
import pytest
from fastapi import status
from httpx import AsyncClient
async def test_post_validate_code(client: AsyncClient):
@pytest.mark.usefixtures("active_user")
async def test_post_validate_code(client: AsyncClient, logged_in_headers):
good_code = """
from pprint import pprint
var = {"a": 1, "b": 2}
pprint(var)
"""
response = await client.post("api/v1/validate/code", json={"code": good_code})
response = await client.post("api/v1/validate/code", json={"code": good_code}, headers=logged_in_headers)
result = response.json()
assert response.status_code == status.HTTP_200_OK
@ -17,7 +19,8 @@ pprint(var)
assert "function" in result, "The result must have a 'function' key"
async def test_post_validate_prompt(client: AsyncClient):
@pytest.mark.usefixtures("active_user")
async def test_post_validate_prompt(client: AsyncClient, logged_in_headers):
basic_case = {
"name": "string",
"template": "string",
@ -48,10 +51,29 @@ async def test_post_validate_prompt(client: AsyncClient):
"metadata": {},
},
}
response = await client.post("api/v1/validate/prompt", json=basic_case)
response = await client.post("api/v1/validate/prompt", json=basic_case, headers=logged_in_headers)
result = response.json()
assert response.status_code == status.HTTP_200_OK
assert isinstance(result, dict), "The result must be a dictionary"
assert "frontend_node" in result, "The result must have a 'frontend_node' key"
assert "input_variables" in result, "The result must have an 'input_variables' key"
@pytest.mark.usefixtures("active_user")
async def test_post_validate_prompt_with_invalid_data(client: AsyncClient, logged_in_headers):
invalid_case = {
"name": "string",
# Missing required fields
"frontend_node": {"template": {}, "is_input": True},
}
response = await client.post("api/v1/validate/prompt", json=invalid_case, headers=logged_in_headers)
assert response.status_code == status.HTTP_422_UNPROCESSABLE_ENTITY
async def test_post_validate_code_with_unauthenticated_user(client: AsyncClient):
code = """
print("Hello World")
"""
response = await client.post("api/v1/validate/code", json={"code": code}, headers={"Authorization": "Bearer fake"})
assert response.status_code == status.HTTP_401_UNAUTHORIZED