feat: handle payment error on trial

Author: jono-booth

enterprise_access/apps/api_client/license_manager_client.py

@@ -28,6 +28,61 @@ class LicenseManagerApiClient(BaseOAuthClient):
     subscription_provisioning_endpoint = api_base_url + 'provisioning-admins/subscriptions/'
     subscription_plan_renewal_provisioning_endpoint = api_base_url + 'provisioning-admins/subscription-plan-renewals/'
 
+    def list_subscriptions(self, enterprise_customer_uuid):
+        """
+        List subscription plans for an enterprise.
+
+        Returns a paginated DRF list response: { count, next, previous, results: [...] }
+        """
+        try:
+            params = {
+                'enterprise_customer_uuid': enterprise_customer_uuid,
+            }
+
+            response = self.client.get(
+                self.subscriptions_endpoint,
+                params=params,
+                timeout=settings.LICENSE_MANAGER_CLIENT_TIMEOUT,
+            )
+            response.raise_for_status()
+            return response.json()
+        except requests.exceptions.HTTPError as exc:
+            logger.exception(
+                'Failed to list subscriptions for enterprise %s, response: %s, exc: %s',
+                enterprise_customer_uuid, safe_error_response_content(exc), exc,
+            )
+            raise
+
+    def update_subscription_plan(self, subscription_uuid, **payload):
+        """
+        Partially update a subscription plan via the provisioning-admins endpoint.
+
+        Args:
+            subscription_uuid (str): Subscription plan UUID.
+            **payload: Fields to patch on the subscription (e.g. is_active, change_reason).
+
+        Returns:
+            dict: JSON response from license-manager.
+        """
+        if not payload:
+            raise ValueError('Must supply payload to update subscription plan')
+
+        endpoint = f"{self.subscription_provisioning_endpoint}{subscription_uuid}/"
+        try:
+            response = self.client.patch(
+                endpoint,
+                json=payload,
+                timeout=settings.LICENSE_MANAGER_CLIENT_TIMEOUT,
+            )
+            response.raise_for_status()
+            return response.json()
+        except requests.exceptions.HTTPError as exc:
+            logger.exception(
+                'Failed to update subscription %s, payload=%s, response: %s, exc: %s',
+                subscription_uuid, payload, safe_error_response_content(exc), exc,
+            )
+            raise
+
     def get_subscription_overview(self, subscription_uuid):
         """
         Call license-manager API for data about a SubscriptionPlan.

enterprise_access/apps/api_client/tests/test_license_manager_client.py

@@ -88,6 +88,49 @@ def test_create_customer_agreement(self, mock_oauth_client):
             json=expected_payload,
         )
 
+    @mock.patch('enterprise_access.apps.api_client.base_oauth.OAuthAPIClient', autospec=True)
+    def test_list_subscriptions_params(self, mock_oauth_client):
+        mock_get = mock_oauth_client.return_value.get
+        mock_get.return_value.json.return_value = {'results': []}
+
+        lm_client = LicenseManagerApiClient()
+        enterprise_uuid = 'ec-uuid-123'
+
+        # Should only set enterprise_customer_uuid parameter
+        result = lm_client.list_subscriptions(enterprise_uuid)
+        self.assertEqual(result, {'results': []})
+
+        # Verify URL and params
+        expected_url = (
+            'http://license-manager.example.com'
+            '/api/v1/subscriptions/'
+        )
+        mock_get.assert_called_with(
+            expected_url,
+            params={'enterprise_customer_uuid': enterprise_uuid},
+            timeout=settings.LICENSE_MANAGER_CLIENT_TIMEOUT,
+        )
+
+    @mock.patch('enterprise_access.apps.api_client.base_oauth.OAuthAPIClient', autospec=True)
+    def test_update_subscription_plan_patch(self, mock_oauth_client):
+        mock_patch = mock_oauth_client.return_value.patch
+        mock_patch.return_value.json.return_value = {'uuid': 'plan-uuid', 'is_active': False}
+
+        lm_client = LicenseManagerApiClient()
+        payload = {'is_active': False, 'change_reason': 'delayed_payment'}
+        result = lm_client.update_subscription_plan('plan-uuid', **payload)
+
+        self.assertEqual(result, mock_patch.return_value.json.return_value)
+        expected_url = (
+            'http://license-manager.example.com'
+            '/api/v1/provisioning-admins/subscriptions/plan-uuid/'
+        )
+        mock_patch.assert_called_once_with(
+            expected_url,
+            json=payload,
+            timeout=settings.LICENSE_MANAGER_CLIENT_TIMEOUT,
+        )
+
     @mock.patch('enterprise_access.apps.api_client.base_oauth.OAuthAPIClient', autospec=True)
     def test_create_subscription_plan(self, mock_oauth_client):
         mock_post = mock_oauth_client.return_value.post