113 lines
3.3 KiB
Python
113 lines
3.3 KiB
Python
"""
|
|
Test script for project settings endpoints
|
|
"""
|
|
|
|
import requests
|
|
import json
|
|
|
|
BASE_URL = "http://localhost:8000"
|
|
|
|
def test_project_settings():
|
|
"""Test project settings endpoints"""
|
|
|
|
# First, login to get a token
|
|
print("1. Logging in...")
|
|
login_response = requests.post(
|
|
f"{BASE_URL}/auth/login",
|
|
data={
|
|
"username": "admin@example.com",
|
|
"password": "admin123"
|
|
}
|
|
)
|
|
|
|
if login_response.status_code != 200:
|
|
print(f"❌ Login failed: {login_response.status_code}")
|
|
print(login_response.text)
|
|
return
|
|
|
|
token = login_response.json()["access_token"]
|
|
headers = {"Authorization": f"Bearer {token}"}
|
|
print("✓ Login successful")
|
|
|
|
# Get projects
|
|
print("\n2. Getting projects...")
|
|
projects_response = requests.get(f"{BASE_URL}/projects/", headers=headers)
|
|
|
|
if projects_response.status_code != 200:
|
|
print(f"❌ Failed to get projects: {projects_response.status_code}")
|
|
return
|
|
|
|
projects = projects_response.json()
|
|
if not projects:
|
|
print("❌ No projects found")
|
|
return
|
|
|
|
project_id = projects[0]["id"]
|
|
print(f"✓ Found project: {projects[0]['name']} (ID: {project_id})")
|
|
|
|
# Get project settings
|
|
print("\n3. Getting project settings...")
|
|
settings_response = requests.get(
|
|
f"{BASE_URL}/projects/{project_id}/settings",
|
|
headers=headers
|
|
)
|
|
|
|
if settings_response.status_code != 200:
|
|
print(f"❌ Failed to get settings: {settings_response.status_code}")
|
|
print(settings_response.text)
|
|
return
|
|
|
|
settings = settings_response.json()
|
|
print("✓ Current settings:")
|
|
print(json.dumps(settings, indent=2))
|
|
|
|
# Update project settings
|
|
print("\n4. Updating project settings...")
|
|
new_settings = {
|
|
"upload_data_location": "/mnt/projects/test-project/uploads",
|
|
"asset_task_templates": {
|
|
"characters": ["modeling", "surfacing", "rigging"],
|
|
"props": ["modeling", "surfacing"],
|
|
"sets": ["modeling", "surfacing"],
|
|
"vehicles": ["modeling", "surfacing", "rigging"]
|
|
},
|
|
"shot_task_templates": ["layout", "animation", "lighting", "compositing"]
|
|
}
|
|
|
|
update_response = requests.put(
|
|
f"{BASE_URL}/projects/{project_id}/settings",
|
|
headers=headers,
|
|
json=new_settings
|
|
)
|
|
|
|
if update_response.status_code != 200:
|
|
print(f"❌ Failed to update settings: {update_response.status_code}")
|
|
print(update_response.text)
|
|
return
|
|
|
|
updated_settings = update_response.json()
|
|
print("✓ Settings updated:")
|
|
print(json.dumps(updated_settings, indent=2))
|
|
|
|
# Verify the update
|
|
print("\n5. Verifying update...")
|
|
verify_response = requests.get(
|
|
f"{BASE_URL}/projects/{project_id}/settings",
|
|
headers=headers
|
|
)
|
|
|
|
if verify_response.status_code != 200:
|
|
print(f"❌ Failed to verify: {verify_response.status_code}")
|
|
return
|
|
|
|
verified_settings = verify_response.json()
|
|
print("✓ Verified settings:")
|
|
print(json.dumps(verified_settings, indent=2))
|
|
|
|
print("\n✅ All tests passed!")
|
|
|
|
if __name__ == "__main__":
|
|
print("Testing Project Settings API")
|
|
print("=" * 50)
|
|
test_project_settings()
|