mirror of
https://github.com/msoedov/agentic_security.git
synced 2026-07-06 03:37:49 +02:00
feat(restruct tests):
This commit is contained in:
@@ -0,0 +1,25 @@
|
||||
from fastapi import FastAPI
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
import agentic_security.test_spec_assets as test_spec_assets
|
||||
from agentic_security.routes.scan import router
|
||||
|
||||
app = FastAPI()
|
||||
app.include_router(router)
|
||||
client = TestClient(app)
|
||||
|
||||
|
||||
def test_upload_csv_and_run():
|
||||
# Create a sample CSV content
|
||||
csv_content = "id,prompt\nspec1,value1\nspec2,value3"
|
||||
# Send a POST request to the /upload-csv endpoint
|
||||
response = client.post(
|
||||
"/scan-csv?optimize=false&enableMultiStepAttack=false&maxBudget=1000",
|
||||
files={
|
||||
"file": ("test.csv", csv_content, "text/csv"),
|
||||
"llmSpec": ("spec.txt", test_spec_assets.SAMPLE_SPEC, "text/plain"),
|
||||
},
|
||||
)
|
||||
|
||||
assert response.status_code == 200
|
||||
assert "Scan completed." in response.text
|
||||
Reference in New Issue
Block a user