mirror of
https://github.com/kjanat/livegraphs-django.git
synced 2026-01-16 20:12:08 +01:00
70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
#!/usr/bin/env python
|
|
"""
|
|
Test the ExternalDataSource Model Schema
|
|
|
|
This management command tests if the ExternalDataSource schema has been correctly updated.
|
|
"""
|
|
|
|
import logging
|
|
|
|
from data_integration.models import ExternalDataSource
|
|
from django.core.management.base import BaseCommand
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Test ExternalDataSource model fields"
|
|
|
|
def handle(self, *args, **options): # noqa: ARG002
|
|
self.stdout.write("Testing ExternalDataSource schema...")
|
|
|
|
try:
|
|
# Get or create a test source
|
|
source, created = ExternalDataSource.objects.get_or_create(
|
|
name="Test Source",
|
|
defaults={
|
|
"api_url": "https://example.com/api",
|
|
"is_active": False,
|
|
},
|
|
)
|
|
|
|
if created:
|
|
self.stdout.write(f"Created test source with ID: {source.id}")
|
|
else:
|
|
self.stdout.write(f"Using existing test source with ID: {source.id}")
|
|
|
|
# Test setting each field
|
|
fields_to_test = {
|
|
"error_count": 0,
|
|
"last_error": "Test error message",
|
|
"sync_interval": 7200,
|
|
"timeout": 600,
|
|
}
|
|
|
|
for field, value in fields_to_test.items():
|
|
try:
|
|
setattr(source, field, value)
|
|
self.stdout.write(self.style.SUCCESS(f"✅ Successfully set {field} = {value}"))
|
|
except AttributeError:
|
|
self.stdout.write(self.style.ERROR(f"❌ Field {field} doesn't exist on the model"))
|
|
|
|
try:
|
|
source.save()
|
|
self.stdout.write(self.style.SUCCESS("✅ Successfully saved with all fields"))
|
|
except Exception as e:
|
|
self.stdout.write(self.style.ERROR(f"❌ Error saving model: {e}"))
|
|
|
|
# Read back the values to verify
|
|
refreshed_source = ExternalDataSource.objects.get(id=source.id)
|
|
self.stdout.write("\nVerifying saved values:")
|
|
for field, expected_value in fields_to_test.items():
|
|
actual_value = getattr(refreshed_source, field, "MISSING")
|
|
if actual_value == expected_value:
|
|
self.stdout.write(self.style.SUCCESS(f"✅ {field} = {actual_value} (correct)"))
|
|
else:
|
|
self.stdout.write(self.style.ERROR(f"❌ {field} = {actual_value} (expected: {expected_value})"))
|
|
|
|
except Exception as e:
|
|
self.stdout.write(self.style.ERROR(f"❌ Test failed: {e}"))
|