new awesome build
This commit is contained in:
33
APP_PROFILER/test_server_process.py
Normal file
33
APP_PROFILER/test_server_process.py
Normal file
@@ -0,0 +1,33 @@
|
||||
from fastapi.testclient import TestClient
|
||||
from main import app
|
||||
import sys
|
||||
import os
|
||||
|
||||
# Add project root to sys.path
|
||||
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
||||
|
||||
client = TestClient(app)
|
||||
|
||||
def test_get_stats():
|
||||
response = client.get("/server/process/stats")
|
||||
print(f"Stats response status: {response.status_code}")
|
||||
print(f"Stats response body: {response.json()}")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert "pid" in data
|
||||
assert "cpu_percent" in data
|
||||
assert "memory_mb" in data
|
||||
|
||||
def test_control_invalid():
|
||||
response = client.post("/server/process/invalid_action")
|
||||
print(f"Invalid action response: {response.status_code}")
|
||||
assert response.status_code == 400
|
||||
|
||||
if __name__ == "__main__":
|
||||
print("Running API tests...")
|
||||
try:
|
||||
test_get_stats()
|
||||
test_control_invalid()
|
||||
print("Tests passed!")
|
||||
except Exception as e:
|
||||
print(f"Tests failed: {e}")
|
||||
Reference in New Issue
Block a user