147 lines
5.4 KiB
Python
147 lines
5.4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import requests
|
|
import json
|
|
|
|
# Configuration
|
|
BASE_URL = "http://localhost:8000"
|
|
LOGIN_URL = f"{BASE_URL}/auth/login"
|
|
ASSETS_URL = f"{BASE_URL}/assets/"
|
|
|
|
def test_complete_task_status_functionality():
|
|
print("Testing Complete Task Status Functionality")
|
|
print("=" * 60)
|
|
|
|
# Login
|
|
print("1. Logging in...")
|
|
login_data = {"email": "admin@vfx.com", "password": "admin123"}
|
|
response = requests.post(LOGIN_URL, json=login_data)
|
|
|
|
if response.status_code != 200:
|
|
print(f"❌ Login failed: {response.status_code}")
|
|
return
|
|
|
|
token = response.json()["access_token"]
|
|
headers = {"Authorization": f"Bearer {token}"}
|
|
print("✅ Login successful")
|
|
|
|
# Get project
|
|
projects_response = requests.get(f"{BASE_URL}/projects/", headers=headers)
|
|
project_id = projects_response.json()[0]["id"]
|
|
print(f"✅ Using project ID: {project_id}")
|
|
|
|
# Test 1: Basic asset listing with task status
|
|
print("\n2. Testing basic asset listing with task status...")
|
|
params = {"project_id": project_id}
|
|
response = requests.get(ASSETS_URL, params=params, headers=headers)
|
|
|
|
if response.status_code == 200:
|
|
assets = response.json()
|
|
print(f"✅ Found {len(assets)} assets")
|
|
|
|
if assets:
|
|
asset = assets[0]
|
|
print(f" Asset: {asset['name']}")
|
|
print(f" Task status: {asset.get('task_status', {})}")
|
|
print(f" Task details count: {len(asset.get('task_details', []))}")
|
|
else:
|
|
print(f"❌ Failed: {response.status_code}")
|
|
return
|
|
|
|
# Test 2: Task status filtering
|
|
print("\n3. Testing task status filtering...")
|
|
test_filters = [
|
|
"modeling:not_started",
|
|
"modeling:in_progress",
|
|
"surfacing:not_started",
|
|
"rigging:not_started"
|
|
]
|
|
|
|
for filter_value in test_filters:
|
|
params = {"project_id": project_id, "task_status_filter": filter_value}
|
|
response = requests.get(ASSETS_URL, params=params, headers=headers)
|
|
|
|
if response.status_code == 200:
|
|
filtered_assets = response.json()
|
|
print(f" Filter '{filter_value}': {len(filtered_assets)} assets")
|
|
else:
|
|
print(f" ❌ Filter '{filter_value}' failed: {response.status_code}")
|
|
|
|
# Test 3: Task status sorting
|
|
print("\n4. Testing task status sorting...")
|
|
sort_tests = [
|
|
("name", "asc"),
|
|
("category", "desc"),
|
|
("modeling_status", "asc"),
|
|
("surfacing_status", "desc"),
|
|
("rigging_status", "asc")
|
|
]
|
|
|
|
for sort_by, sort_direction in sort_tests:
|
|
params = {
|
|
"project_id": project_id,
|
|
"sort_by": sort_by,
|
|
"sort_direction": sort_direction
|
|
}
|
|
response = requests.get(ASSETS_URL, params=params, headers=headers)
|
|
|
|
if response.status_code == 200:
|
|
sorted_assets = response.json()
|
|
print(f" Sort by '{sort_by}' ({sort_direction}): {len(sorted_assets)} assets")
|
|
|
|
# Show first few asset names to verify sorting
|
|
if sorted_assets and sort_by == "name":
|
|
names = [a['name'] for a in sorted_assets[:3]]
|
|
print(f" First 3: {names}")
|
|
else:
|
|
print(f" ❌ Sort by '{sort_by}' failed: {response.status_code}")
|
|
|
|
# Test 4: Category-aware task columns
|
|
print("\n5. Testing category-aware task columns...")
|
|
params = {"project_id": project_id}
|
|
response = requests.get(ASSETS_URL, params=params, headers=headers)
|
|
|
|
if response.status_code == 200:
|
|
assets = response.json()
|
|
|
|
for asset in assets:
|
|
category = asset['category']
|
|
task_status = asset.get('task_status', {})
|
|
|
|
print(f" {asset['name']} ({category}):")
|
|
print(f" Modeling: {task_status.get('modeling', 'N/A')}")
|
|
print(f" Surfacing: {task_status.get('surfacing', 'N/A')}")
|
|
|
|
# Rigging should only be available for characters and vehicles
|
|
if category in ['characters', 'vehicles']:
|
|
print(f" Rigging: {task_status.get('rigging', 'N/A')}")
|
|
else:
|
|
print(f" Rigging: N/A (not applicable to {category})")
|
|
|
|
# Test 5: Combined filtering and sorting
|
|
print("\n6. Testing combined filtering and sorting...")
|
|
params = {
|
|
"project_id": project_id,
|
|
"task_status_filter": "modeling:not_started",
|
|
"sort_by": "name",
|
|
"sort_direction": "asc"
|
|
}
|
|
response = requests.get(ASSETS_URL, params=params, headers=headers)
|
|
|
|
if response.status_code == 200:
|
|
combined_assets = response.json()
|
|
print(f"✅ Combined filter + sort: {len(combined_assets)} assets")
|
|
else:
|
|
print(f"❌ Combined test failed: {response.status_code}")
|
|
|
|
print("\n✅ All task status functionality tests completed successfully!")
|
|
print("\nSummary of implemented features:")
|
|
print("- ✅ Asset list API includes task status information")
|
|
print("- ✅ Task status filtering by task type and status")
|
|
print("- ✅ Task status sorting (including task status columns)")
|
|
print("- ✅ Category-aware task column display")
|
|
print("- ✅ Combined filtering and sorting")
|
|
print("- ✅ Task status update functionality")
|
|
|
|
if __name__ == "__main__":
|
|
test_complete_task_status_functionality() |