This commit is contained in:
pacnpal
2026-01-08 13:44:37 -05:00
parent 40cba5bdb2
commit fe960e8b62
21 changed files with 2008 additions and 59 deletions

View File

@@ -30,14 +30,14 @@ class RideOpeningWorkflowTests(TestCase):
def _create_ride(self, status="OPERATING", **kwargs):
"""Helper to create a ride with park."""
from apps.parks.models import Company, Park
from apps.rides.models import Ride
from apps.parks.models import Company as ParkCompany, Park
from apps.rides.models import Company as RideCompany, Ride
# Create manufacturer
manufacturer = Company.objects.create(name=f"Manufacturer {timezone.now().timestamp()}", roles=["MANUFACTURER"])
# Create manufacturer (from rides.Company)
manufacturer = RideCompany.objects.create(name=f"Manufacturer {timezone.now().timestamp()}", roles=["MANUFACTURER"])
# Create park with operator
operator = Company.objects.create(name=f"Operator {timezone.now().timestamp()}", roles=["OPERATOR"])
# Create park with operator (from parks.Company)
operator = ParkCompany.objects.create(name=f"Operator {timezone.now().timestamp()}", roles=["OPERATOR"])
park = Park.objects.create(
name=f"Test Park {timezone.now().timestamp()}",
slug=f"test-park-{timezone.now().timestamp()}",
@@ -84,11 +84,11 @@ class RideMaintenanceWorkflowTests(TestCase):
)
def _create_ride(self, status="OPERATING", **kwargs):
from apps.parks.models import Company, Park
from apps.rides.models import Ride
from apps.parks.models import Company as ParkCompany, Park
from apps.rides.models import Company as RideCompany, Ride
manufacturer = Company.objects.create(name=f"Mfr Maint {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = Company.objects.create(name=f"Op Maint {timezone.now().timestamp()}", roles=["OPERATOR"])
manufacturer = RideCompany.objects.create(name=f"Mfr Maint {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = ParkCompany.objects.create(name=f"Op Maint {timezone.now().timestamp()}", roles=["OPERATOR"])
park = Park.objects.create(
name=f"Park Maint {timezone.now().timestamp()}",
slug=f"park-maint-{timezone.now().timestamp()}",
@@ -140,11 +140,11 @@ class RideSBNOWorkflowTests(TestCase):
)
def _create_ride(self, status="OPERATING", **kwargs):
from apps.parks.models import Company, Park
from apps.rides.models import Ride
from apps.parks.models import Company as ParkCompany, Park
from apps.rides.models import Company as RideCompany, Ride
manufacturer = Company.objects.create(name=f"Mfr SBNO {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = Company.objects.create(name=f"Op SBNO {timezone.now().timestamp()}", roles=["OPERATOR"])
manufacturer = RideCompany.objects.create(name=f"Mfr SBNO {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = ParkCompany.objects.create(name=f"Op SBNO {timezone.now().timestamp()}", roles=["OPERATOR"])
park = Park.objects.create(
name=f"Park SBNO {timezone.now().timestamp()}",
slug=f"park-sbno-{timezone.now().timestamp()}",
@@ -234,11 +234,11 @@ class RideScheduledClosureWorkflowTests(TestCase):
)
def _create_ride(self, status="OPERATING", **kwargs):
from apps.parks.models import Company, Park
from apps.rides.models import Ride
from apps.parks.models import Company as ParkCompany, Park
from apps.rides.models import Company as RideCompany, Ride
manufacturer = Company.objects.create(name=f"Mfr Closing {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = Company.objects.create(name=f"Op Closing {timezone.now().timestamp()}", roles=["OPERATOR"])
manufacturer = RideCompany.objects.create(name=f"Mfr Closing {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = ParkCompany.objects.create(name=f"Op Closing {timezone.now().timestamp()}", roles=["OPERATOR"])
park = Park.objects.create(
name=f"Park Closing {timezone.now().timestamp()}",
slug=f"park-closing-{timezone.now().timestamp()}",
@@ -324,11 +324,11 @@ class RideDemolitionWorkflowTests(TestCase):
)
def _create_ride(self, status="CLOSED_PERM", **kwargs):
from apps.parks.models import Company, Park
from apps.rides.models import Ride
from apps.parks.models import Company as ParkCompany, Park
from apps.rides.models import Company as RideCompany, Ride
manufacturer = Company.objects.create(name=f"Mfr Demo {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = Company.objects.create(name=f"Op Demo {timezone.now().timestamp()}", roles=["OPERATOR"])
manufacturer = RideCompany.objects.create(name=f"Mfr Demo {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = ParkCompany.objects.create(name=f"Op Demo {timezone.now().timestamp()}", roles=["OPERATOR"])
park = Park.objects.create(
name=f"Park Demo {timezone.now().timestamp()}",
slug=f"park-demo-{timezone.now().timestamp()}",
@@ -383,11 +383,11 @@ class RideRelocationWorkflowTests(TestCase):
)
def _create_ride(self, status="CLOSED_PERM", **kwargs):
from apps.parks.models import Company, Park
from apps.rides.models import Ride
from apps.parks.models import Company as ParkCompany, Park
from apps.rides.models import Company as RideCompany, Ride
manufacturer = Company.objects.create(name=f"Mfr Reloc {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = Company.objects.create(name=f"Op Reloc {timezone.now().timestamp()}", roles=["OPERATOR"])
manufacturer = RideCompany.objects.create(name=f"Mfr Reloc {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = ParkCompany.objects.create(name=f"Op Reloc {timezone.now().timestamp()}", roles=["OPERATOR"])
park = Park.objects.create(
name=f"Park Reloc {timezone.now().timestamp()}",
slug=f"park-reloc-{timezone.now().timestamp()}",
@@ -445,11 +445,11 @@ class RideWrapperMethodTests(TestCase):
)
def _create_ride(self, status="OPERATING", **kwargs):
from apps.parks.models import Company, Park
from apps.rides.models import Ride
from apps.parks.models import Company as ParkCompany, Park
from apps.rides.models import Company as RideCompany, Ride
manufacturer = Company.objects.create(name=f"Mfr Wrapper {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = Company.objects.create(name=f"Op Wrapper {timezone.now().timestamp()}", roles=["OPERATOR"])
manufacturer = RideCompany.objects.create(name=f"Mfr Wrapper {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = ParkCompany.objects.create(name=f"Op Wrapper {timezone.now().timestamp()}", roles=["OPERATOR"])
park = Park.objects.create(
name=f"Park Wrapper {timezone.now().timestamp()}",
slug=f"park-wrapper-{timezone.now().timestamp()}",
@@ -573,11 +573,11 @@ class RidePostClosingStatusAutomationTests(TestCase):
)
def _create_ride(self, status="CLOSING", **kwargs):
from apps.parks.models import Company, Park
from apps.rides.models import Ride
from apps.parks.models import Company as ParkCompany, Park
from apps.rides.models import Company as RideCompany, Ride
manufacturer = Company.objects.create(name=f"Mfr Auto {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = Company.objects.create(name=f"Op Auto {timezone.now().timestamp()}", roles=["OPERATOR"])
manufacturer = RideCompany.objects.create(name=f"Mfr Auto {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = ParkCompany.objects.create(name=f"Op Auto {timezone.now().timestamp()}", roles=["OPERATOR"])
park = Park.objects.create(
name=f"Park Auto {timezone.now().timestamp()}",
slug=f"park-auto-{timezone.now().timestamp()}",
@@ -659,11 +659,11 @@ class RideStateLogTests(TestCase):
)
def _create_ride(self, status="OPERATING", **kwargs):
from apps.parks.models import Company, Park
from apps.rides.models import Ride
from apps.parks.models import Company as ParkCompany, Park
from apps.rides.models import Company as RideCompany, Ride
manufacturer = Company.objects.create(name=f"Mfr Log {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = Company.objects.create(name=f"Op Log {timezone.now().timestamp()}", roles=["OPERATOR"])
manufacturer = RideCompany.objects.create(name=f"Mfr Log {timezone.now().timestamp()}", roles=["MANUFACTURER"])
operator = ParkCompany.objects.create(name=f"Op Log {timezone.now().timestamp()}", roles=["OPERATOR"])
park = Park.objects.create(
name=f"Park Log {timezone.now().timestamp()}",
slug=f"park-log-{timezone.now().timestamp()}",