Init Repo

2026-02-28 03:22:04 +08:00
commit de59b57ee7
883 changed files with 156857 additions and 0 deletions
@@ -0,0 +1,137 @@
+"""
+Test script for task status endpoints with custom statuses
+"""
+import requests
+import json
+from sqlalchemy.orm.attributes import flag_modified
+
+BASE_URL = "http://localhost:8000"
+
+def setup_custom_statuses():
+    """Add custom statuses to a project for testing"""
+    from database import SessionLocal
+    from models.project import Project
+    
+    db = SessionLocal()
+    
+    # Get first project
+    project = db.query(Project).first()
+    if not project:
+        print("❌ No project found")
+        db.close()
+        return None
+    
+    # Add custom statuses
+    custom_statuses = [
+        {
+            "id": "custom_review",
+            "name": "In Review",
+            "color": "#8B5CF6",
+            "order": 0,
+            "is_default": False
+        },
+        {
+            "id": "custom_blocked",
+            "name": "Blocked",
+            "color": "#DC2626",
+            "order": 1,
+            "is_default": False
+        },
+        {
+            "id": "custom_ready",
+            "name": "Ready for Delivery",
+            "color": "#059669",
+            "order": 2,
+            "is_default": True  # Set as default
+        }
+    ]
+    
+    project.custom_task_statuses = custom_statuses
+    flag_modified(project, 'custom_task_statuses')
+    
+    project_id = project.id
+    
+    db.commit()
+    db.close()
+    
+    print(f"✅ Added {len(custom_statuses)} custom statuses to project {project_id}")
+    return project_id
+
+
+def test_get_all_task_statuses_with_custom():
+    """Test getting all task statuses including custom ones"""
+    
+    # Setup custom statuses
+    project_id = setup_custom_statuses()
+    if not project_id:
+        return
+    
+    # Login to get a token
+    login_response = requests.post(
+        f"{BASE_URL}/auth/login",
+        json={
+            "email": "admin@vfx.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")
+    
+    # Test GET /projects/{project_id}/task-statuses
+    print(f"\n--- Testing GET /projects/{project_id}/task-statuses ---")
+    response = requests.get(
+        f"{BASE_URL}/projects/{project_id}/task-statuses",
+        headers=headers
+    )
+    
+    print(f"Status Code: {response.status_code}")
+    
+    if response.status_code == 200:
+        data = response.json()
+        print("✅ Successfully retrieved task statuses")
+        print(f"\nResponse:")
+        print(json.dumps(data, indent=2))
+        
+        # Validate response structure
+        assert "statuses" in data, "Missing 'statuses' field"
+        assert "system_statuses" in data, "Missing 'system_statuses' field"
+        assert "default_status_id" in data, "Missing 'default_status_id' field"
+        
+        print(f"\n✅ System statuses count: {len(data['system_statuses'])}")
+        print(f"✅ Custom statuses count: {len(data['statuses'])}")
+        print(f"✅ Default status ID: {data['default_status_id']}")
+        
+        # Verify custom statuses
+        assert len(data['statuses']) == 3, f"Expected 3 custom statuses, got {len(data['statuses'])}"
+        print("✅ Correct number of custom statuses")
+        
+        # Verify default status is custom
+        assert data['default_status_id'] == "custom_ready", f"Expected default to be 'custom_ready', got {data['default_status_id']}"
+        print("✅ Default status correctly set to custom status")
+        
+        # Verify custom status structure
+        for status in data['statuses']:
+            assert "id" in status, f"Custom status missing 'id': {status}"
+            assert "name" in status, f"Custom status missing 'name': {status}"
+            assert "color" in status, f"Custom status missing 'color': {status}"
+            assert "order" in status, f"Custom status missing 'order': {status}"
+            assert "is_default" in status, f"Custom status missing 'is_default': {status}"
+            print(f"✅ Custom status '{status['name']}' has all required fields")
+        
+        print("\n✅ All tests passed!")
+        
+    else:
+        print(f"❌ Failed to get task statuses")
+        print(response.text)
+
+
+if __name__ == "__main__":
+    test_get_all_task_statuses_with_custom()