135 lines
3.3 KiB
Python
135 lines
3.3 KiB
Python
"""
|
|
Test script for category cycle protection
|
|
"""
|
|
import os
|
|
import django
|
|
|
|
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'myproject.settings')
|
|
django.setup()
|
|
|
|
from products.models import ProductCategory
|
|
from django.core.exceptions import ValidationError
|
|
|
|
|
|
print("="*60)
|
|
print("TEST 1: Self-reference (A.parent = A)")
|
|
print("="*60)
|
|
try:
|
|
cat = ProductCategory.objects.first()
|
|
print(f"Category: {cat.name} (id={cat.pk})")
|
|
cat.parent = cat
|
|
cat.save()
|
|
print("[FAIL] Self-reference was NOT blocked!")
|
|
except ValidationError as e:
|
|
print("[PASS] Self-reference blocked successfully")
|
|
print(f"Error: {e.message_dict.get('parent')}")
|
|
|
|
print("\n" + "="*60)
|
|
print("TEST 2: Simple cycle (A -> B -> A)")
|
|
print("="*60)
|
|
try:
|
|
cats = list(ProductCategory.objects.all()[:2])
|
|
cat_a, cat_b = cats[0], cats[1]
|
|
print(f"Category A: {cat_a.name} (id={cat_a.pk})")
|
|
print(f"Category B: {cat_b.name} (id={cat_b.pk})")
|
|
|
|
# Create A -> B
|
|
cat_a.parent = None
|
|
cat_a.save()
|
|
cat_b.parent = cat_a
|
|
cat_b.save()
|
|
print(f"Created: {cat_a.name} -> {cat_b.name}")
|
|
|
|
# Try to create cycle: A.parent = B
|
|
cat_a.parent = cat_b
|
|
cat_a.save()
|
|
print("[FAIL] Cycle was NOT blocked!")
|
|
|
|
# Cleanup
|
|
cat_a.parent = None
|
|
cat_b.parent = None
|
|
cat_a.save()
|
|
cat_b.save()
|
|
|
|
except ValidationError as e:
|
|
print("[PASS] Cycle blocked successfully")
|
|
print(f"Error: {e.message_dict.get('parent')}")
|
|
# Cleanup
|
|
cat_a.parent = None
|
|
cat_b.parent = None
|
|
cat_a.save()
|
|
cat_b.save()
|
|
|
|
print("\n" + "="*60)
|
|
print("TEST 3: Multi-level cycle (A -> B -> C -> A)")
|
|
print("="*60)
|
|
try:
|
|
cats = list(ProductCategory.objects.all()[:3])
|
|
cat_a, cat_b, cat_c = cats[0], cats[1], cats[2]
|
|
print(f"Category A: {cat_a.name} (id={cat_a.pk})")
|
|
print(f"Category B: {cat_b.name} (id={cat_b.pk})")
|
|
print(f"Category C: {cat_c.name} (id={cat_c.pk})")
|
|
|
|
# Create chain A -> B -> C
|
|
cat_a.parent = None
|
|
cat_a.save()
|
|
cat_b.parent = cat_a
|
|
cat_b.save()
|
|
cat_c.parent = cat_b
|
|
cat_c.save()
|
|
print(f"Created: {cat_a.name} -> {cat_b.name} -> {cat_c.name}")
|
|
|
|
# Try to create cycle: A.parent = C
|
|
cat_a.parent = cat_c
|
|
cat_a.save()
|
|
print("[FAIL] Multi-level cycle was NOT blocked!")
|
|
|
|
# Cleanup
|
|
cat_a.parent = None
|
|
cat_b.parent = None
|
|
cat_c.parent = None
|
|
cat_a.save()
|
|
cat_b.save()
|
|
cat_c.save()
|
|
|
|
except ValidationError as e:
|
|
print("[PASS] Multi-level cycle blocked successfully")
|
|
print(f"Error: {e.message_dict.get('parent')}")
|
|
# Cleanup
|
|
cat_a.parent = None
|
|
cat_b.parent = None
|
|
cat_c.parent = None
|
|
cat_a.save()
|
|
cat_b.save()
|
|
cat_c.save()
|
|
|
|
print("\n" + "="*60)
|
|
print("TEST 4: Normal operations should work")
|
|
print("="*60)
|
|
try:
|
|
cats = list(ProductCategory.objects.all()[:2])
|
|
cat_a, cat_b = cats[0], cats[1]
|
|
|
|
# Reset
|
|
cat_a.parent = None
|
|
cat_b.parent = None
|
|
cat_a.save()
|
|
cat_b.save()
|
|
|
|
# Create normal hierarchy A -> B
|
|
cat_b.parent = cat_a
|
|
cat_b.save()
|
|
print(f"[PASS] Normal hierarchy created: {cat_a.name} -> {cat_b.name}")
|
|
|
|
# Cleanup
|
|
cat_b.parent = None
|
|
cat_b.save()
|
|
|
|
except ValidationError as e:
|
|
print("[FAIL] Normal operation was blocked!")
|
|
print(f"Error: {e}")
|
|
|
|
print("\n" + "="*60)
|
|
print("ALL TESTS COMPLETED")
|
|
print("="*60)
|