Files
thrillwiki_django_no_react/tests/test_runner.py
pacnpal 7815de158e feat: Complete Company Migration Project and Fix Autocomplete Issues
- Implemented a comprehensive migration from a single Company model to specialized entities (Operators, PropertyOwners, Manufacturers, Designers).
- Resolved critical issues in search suggestions that were returning 404 errors by fixing database queries and reordering URL patterns.
- Conducted extensive testing and validation of the new entity relationships, ensuring all core functionality is operational.
- Updated test suite to reflect changes in entity structure, including renaming fields from `owner` to `operator`.
- Addressed display issues in the user interface related to operator and manufacturer information.
- Completed migration cleanup, fixing references to the removed `companies` app across migration files and test configurations.
- Established a stable testing environment with successful test database creation and functional test infrastructure.
2025-07-05 22:00:21 -04:00

135 lines
3.6 KiB
Python

#!/usr/bin/env python
import os
import sys
import django
from django.conf import settings
from django.test.runner import DiscoverRunner
import coverage # type: ignore
import unittest
def setup_django():
"""Set up Django test environment"""
# Add the project root directory to Python path
project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, project_root)
os***REMOVED***iron.setdefault('DJANGO_SETTINGS_MODULE', 'thrillwiki.settings')
django.setup()
# Use PostGIS for GeoDjango support
settings.DATABASES = {
'default': {
'ENGINE': 'django.contrib.gis.db.backends.postgis',
'NAME': 'test_thrillwiki',
'USER': 'postgres',
'PASSWORD': 'postgres',
'HOST': 'localhost',
'PORT': '5432',
'TEST': {
'NAME': 'test_thrillwiki',
}
}
}
settings.DEBUG = False
# Skip problematic migrations during tests
settings.MIGRATION_MODULES = {
'parks': None,
'operators': None,
'manufacturers': None,
'property_owners': None,
'location': None,
'rides': None,
'reviews': None
}
class CustomTestRunner(DiscoverRunner):
def __init__(self, *args, **kwargs):
self.cov = coverage.Coverage(
source=[
'parks',
'operators',
'manufacturers',
'property_owners',
'location',
'rides',
'reviews'
],
omit=[
'*/migrations/*',
'*/management/*',
'*/admin.py',
'*/apps.py',
'manage.py'
]
)
self.cov.start()
super().__init__(*args, **kwargs)
def setup_databases(self, **kwargs):
"""Set up databases and ensure content types are created"""
old_config = super().setup_databases(**kwargs)
# Create necessary content types
from django.contrib.contenttypes.models import ContentType
from parks.models import Park
ContentType.objects.get_or_create(
app_label='parks',
model='park'
)
return old_config
def run_suite(self, suite, **kwargs):
results = super().run_suite(suite, **kwargs)
self.cov.stop()
self.cov.save()
# Print coverage report
print('\nCoverage Report:')
self.cov.report()
# Generate HTML coverage report
html_dir = os.path.join('tests', 'coverage_html')
self.cov.html_report(directory=html_dir)
print(f'\nDetailed HTML coverage report generated in: {html_dir}')
return results
def run_tests():
# Set up Django
setup_django()
# Initialize test runner
test_runner = CustomTestRunner(
verbosity=2,
interactive=True,
keepdb=True
)
# Define test labels for discovery
test_labels = [
'parks.tests',
'operators.tests',
'manufacturers.tests',
'property_owners.tests',
'location.tests',
'rides.tests',
'reviews.tests'
]
# Run tests and collect results
failures = test_runner.run_tests(test_labels)
return failures
if __name__ == '__main__':
# Create tests directory if it doesn't exist
os.makedirs('tests', exist_ok=True)
os.makedirs(os.path.join('tests', 'coverage_html'), exist_ok=True)
# Run tests and exit with appropriate status code
failures = run_tests()
sys.exit(bool(failures))