mirror of
https://github.com/pacnpal/thrillwiki_django_no_react.git
synced 2025-12-20 18:31:09 -05:00
66ed4347a9
- Cleaned up and standardized assertions in ApiTestMixin for API response validation. - Updated ASGI settings to use os.environ for setting the DJANGO_SETTINGS_MODULE. - Removed unused imports and improved formatting in settings.py. - Refactored URL patterns in urls.py for better readability and organization. - Enhanced view functions in views.py for consistency and clarity. - Added .flake8 configuration for linting and style enforcement. - Introduced type stubs for django-environ to improve type checking with Pylance.
137 lines
4.0 KiB
Python
137 lines
4.0 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Basic test script to verify RideLocation and CompanyHeadquarters models work correctly.
|
|
"""
|
|
from rides.models import Ride, RideLocation
|
|
from parks.models import Company, CompanyHeadquarters
|
|
import os
|
|
import sys
|
|
import django
|
|
|
|
# Setup Django
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "thrillwiki.settings")
|
|
django.setup()
|
|
|
|
|
|
def test_company_headquarters():
|
|
"""Test CompanyHeadquarters model functionality"""
|
|
print("Testing CompanyHeadquarters...")
|
|
|
|
# Try to use an existing company or skip this test if none exist
|
|
existing_company = Company.objects.first()
|
|
if not existing_company:
|
|
print("⚠️ No existing companies found, skipping CompanyHeadquarters test")
|
|
return None, None
|
|
|
|
# Check if headquarters already exist
|
|
try:
|
|
headquarters = existing_company.headquarters
|
|
print(f"✓ Found existing headquarters: {headquarters}")
|
|
except CompanyHeadquarters.DoesNotExist:
|
|
# Create headquarters for existing company
|
|
headquarters = CompanyHeadquarters.objects.create(
|
|
company=existing_company,
|
|
city="Orlando",
|
|
state_province="Florida",
|
|
country="USA",
|
|
street_address="123 Theme Park Blvd",
|
|
postal_code="32801",
|
|
)
|
|
print(f"✓ Created new headquarters: {headquarters}")
|
|
|
|
print(f"✓ Created headquarters: {headquarters}")
|
|
print(f"✓ Location display: {headquarters.location_display}")
|
|
print(f"✓ Formatted location: {headquarters.formatted_location}")
|
|
|
|
return existing_company, headquarters
|
|
|
|
|
|
def test_ride_location():
|
|
"""Test RideLocation model functionality"""
|
|
print("\nTesting RideLocation...")
|
|
|
|
# First, we need a ride - let's check if any exist
|
|
if Ride.objects.exists():
|
|
ride = Ride.objects.first()
|
|
print(f"✓ Using existing ride: {ride.name}")
|
|
else:
|
|
print("! No rides found in database - skipping RideLocation test")
|
|
return None, None
|
|
|
|
# Create ride location
|
|
ride_location = RideLocation.objects.create(
|
|
ride=ride,
|
|
park_area="Fantasyland",
|
|
notes="General location information",
|
|
entrance_notes="Queue entrance is to the left of the main attraction sign",
|
|
accessibility_notes="Wheelchair accessible entrance available via side path",
|
|
)
|
|
|
|
print(f"✓ Created ride location: {ride_location}")
|
|
print(f"✓ Has coordinates: {ride_location.has_coordinates}")
|
|
|
|
# Test setting coordinates
|
|
ride_location.set_coordinates(28.3772, -81.5707) # Disney World coordinates
|
|
ride_location.save()
|
|
|
|
print(f"✓ Set coordinates: {ride_location.coordinates}")
|
|
print(f"✓ Latitude: {ride_location.latitude}")
|
|
print(f"✓ Longitude: {ride_location.longitude}")
|
|
|
|
return ride, ride_location
|
|
|
|
|
|
def cleanup_test_data(company=None, headquarters=None, ride_location=None):
|
|
"""Clean up test data"""
|
|
print("\nCleaning up test data...")
|
|
|
|
if ride_location:
|
|
ride_location.delete()
|
|
print("✓ Deleted test ride location")
|
|
|
|
if headquarters:
|
|
headquarters.delete()
|
|
print("✓ Deleted test headquarters")
|
|
|
|
if company:
|
|
company.delete()
|
|
print("✓ Deleted test company")
|
|
|
|
|
|
def main():
|
|
"""Run all tests"""
|
|
print("=" * 50)
|
|
print("Testing Location Models Implementation")
|
|
print("=" * 50)
|
|
|
|
try:
|
|
# Test CompanyHeadquarters
|
|
company, headquarters = test_company_headquarters()
|
|
|
|
# Test RideLocation
|
|
ride, ride_location = test_ride_location()
|
|
|
|
print("\n" + "=" * 50)
|
|
print("✅ ALL TESTS PASSED!")
|
|
print(
|
|
"✅ Both RideLocation and CompanyHeadquarters models are working correctly"
|
|
)
|
|
print("=" * 50)
|
|
|
|
# Clean up
|
|
cleanup_test_data(company, headquarters, ride_location)
|
|
|
|
except Exception as e:
|
|
print(f"\n❌ TEST FAILED: {e}")
|
|
import traceback
|
|
|
|
traceback.print_exc()
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
if __name__ == "__main__":
|
|
success = main()
|
|
sys.exit(0 if success else 1)
|