fix: fix migrations and add new test

Author: kiram15

enterprise_access/apps/api/v1/tests/test_stripe_event_summary_views.py

@@ -110,7 +110,7 @@ def setUp(self):
             expires_at=timezone.now() + timedelta(hours=1),
         )
 
-        # Create twp StripeEventData objects, will each create StripeEventSummary on create
+        # Create two StripeEventData objects, will each create StripeEventSummary on create
         self.stripe_event_data = StripeEventData.objects.create(
             event_id='evt_test_invoice',
             event_type='invoice.paid',
@@ -178,3 +178,82 @@ def test_get_stripe_event_summary_by_subscription_uuid_no_auth(self):
         response = self.client.get(url)
         # trying to fetch a subscription plan for an enterprise customer they are not associated with
         assert response.status_code == 403
+
+
+class StripeEventUpcomingInvoiceAmountDueTests(APITest):
+    """
+    Tests for first_upcoming_invoice_amount_due endpoint.
+    """
+
+    def setUp(self):
+        super().setUp()
+        self.user = UserFactory()
+        self.enterprise_uuid = str(uuid.uuid4())
+        self.stripe_customer_id = 'cus_test_123'
+        self.subscription_plan_uuid = str(uuid.uuid4())
+
+        self.checkout_intent = CheckoutIntent.objects.create(
+            user=self.user,
+            enterprise_uuid=self.enterprise_uuid,
+            enterprise_name='Test Enterprise',
+            enterprise_slug='test-enterprise',
+            stripe_customer_id=self.stripe_customer_id,
+            state=CheckoutIntentState.PAID,
+            quantity=10,
+            expires_at=timezone.now() + timedelta(hours=1),
+        )
+
+        subscription_event_data = {
+            'id': 'evt_test_sub_created',
+            'type': 'customer.subscription.created',
+            'data': {
+                'object': {
+                    'object': 'subscription',
+                    'id': 'sub_test_789',
+                    'status': 'active',
+                    'items': {
+                        'data': [
+                            {
+                                'object': 'subscription_item',
+                                'current_period_start': 1609459200,  # 2021-01-01 00:00:00 UTC
+                                'current_period_end': 1640995200,    # 2022-01-01 00:00:00 UTC
+                            }
+                        ]
+                    },
+                }
+            },
+            'metadata': {
+                'checkout_intent_id': self.checkout_intent.id,
+                'enterprise_customer_name': 'Test Enterprise',
+                'enterprise_customer_slug': 'test-enterprise',
+            }
+        }
+
+        # Creating a StripeEventData record triggers a create of related StripeEventSummary
+        self.stripe_event_data = StripeEventData.objects.create(
+            event_id='evt_test_subscription',
+            event_type='customer.subscription.created',
+            checkout_intent=self.checkout_intent,
+            data=subscription_event_data,
+        )
+
+        test_summary = StripeEventSummary.objects.filter(event_id='evt_test_subscription').first()
+        test_summary.upcoming_invoice_amount_due = 200
+        test_summary.subscription_plan_uuid = self.subscription_plan_uuid
+        test_summary.save(update_fields=['upcoming_invoice_amount_due', 'subscription_plan_uuid'])
+
+    def test_get_first_upcoming_invoice_amount_due(self):
+        self.set_jwt_cookie([{
+            'system_wide_role': SYSTEM_ENTERPRISE_ADMIN_ROLE,
+            'context': self.enterprise_uuid,  # implicit access to this enterprise
+        }])
+
+        query_params = {
+            'subscription_plan_uuid': self.subscription_plan_uuid,
+        }
+
+        url = reverse('api:v1:stripe-event-summary-first-upcoming-invoice-amount-due')
+        url += f"?{urlencode(query_params)}"
+        response = self.client.get(url)
+        assert response.status_code == 200
+        assert response.data == {'upcoming_invoice_amount_due': 200}

enterprise_access/apps/api/v1/views/customer_billing.py

@@ -386,19 +386,16 @@ def create_checkout_portal_session(self, request, pk=None):
             '## Allowed State Transitions\n'
             '```\n'
             'created → paid\n'
-            'created → expired\n'
+            'created → errored_stripe_checkout\n'
             'paid → fulfilled\n'
-            'paid → errored_backoffice\n'
-            'paid → errored_fulfillment_stalled\n'
             'paid → errored_provisioning\n'
-            'errored_provisioning → fulfilled\n'
-            'expired → created\n'
+            'errored_stripe_checkout → paid\n'
+            'errored_provisioning → paid\n'
             '```\n'
             '## Integration Points\n'
             '- **Stripe Webhook**: Transitions from `created` to `paid` after successful payment\n'
             '- **Fulfillment Service**: Transitions from `paid` to `fulfilled` after provisioning\n'
-            '- **Backoffice Integration**: Transitions to `errored_backoffice` on Salesforce failures\n'
-            '- **Error Recovery**: Allows retry from `errored_provisioning` to `fulfilled`\n\n'
+            '- **Error Recovery**: Allows retry from error states back to `paid`\n\n'
         ),
         parameters=[
             OpenApiParameter(
@@ -532,8 +529,14 @@ def list(self, request, *args, **kwargs):
         methods=['get'],
         url_path='first-invoice-upcoming-amount-due',
     )
-    def first_invoice_upcoming_amount_due(self, request, *args, **kwargs):
-        summary = self.get_queryset().filter(event_type='customer.subscription.created').first()
+    def first_upcoming_invoice_amount_due(self, request, *args, **kwargs):
+        subscription_plan_uuid = self.request.query_params.get('subscription_plan_uuid')
+        summary = StripeEventSummary.objects.filter(
+            event_type='customer.subscription.created',
+            subscription_plan_uuid=subscription_plan_uuid,
+        ).first()
+        if not (subscription_plan_uuid and summary):
+            return Response({})
         return Response({
             'upcoming_invoice_amount_due': summary.upcoming_invoice_amount_due,
-        }, content_type='application/json')
+        })

enterprise_access/apps/customer_billing/migrations/0018_merge_20251105_0545.py

@@ -6,7 +6,7 @@
 class Migration(migrations.Migration):
 
     dependencies = [
-        ('customer_billing', '0016_alter_checkoutintent_uuid_and_more'),
+        ('customer_billing', '0018_update_checkout_intent_error_states'),
     ]
 
     operations = [