fix(billing): handle string usage limits TASK-1232

kobo/apps/stripe/tests/test_organization_usage.py

@@ -488,10 +488,24 @@ def test_get_plan_community_limit(self):
     def test_get_suscription_limit(self, usage_type):
         stripe_key = f'{USAGE_LIMIT_MAP[usage_type]}_limit'
         product_metadata = {
-            stripe_key: 1234,
+            stripe_key: '1234',
             'product_type': 'plan',
             'plan_type': 'enterprise',
         }
         generate_plan_subscription(self.organization, metadata=product_metadata)
         limit = get_organization_plan_limit(self.organization, usage_type)
         assert limit == 1234
+
+    # Currently submissions and storage are the only usage types that can be
+    # 'unlimited'
+    @data('submission', 'storage')
+    def test_get_suscription_limit_unlimited(self, usage_type):
+        stripe_key = f'{USAGE_LIMIT_MAP[usage_type]}_limit'
+        product_metadata = {
+            stripe_key: 'unlimited',
+            'product_type': 'plan',
+            'plan_type': 'enterprise',
+        }
+        generate_plan_subscription(self.organization, metadata=product_metadata)
+        limit = get_organization_plan_limit(self.organization, usage_type)
+        assert limit == float('inf')

kobo/apps/stripe/utils.py

@@ -1,4 +1,4 @@
-from math import ceil, floor
+from math import ceil, floor, inf
 
 from django.conf import settings
 from django.db.models import F
@@ -44,7 +44,7 @@ def generate_return_url(product_metadata):
 
 def get_organization_plan_limit(
     organization: Organization, usage_type: UsageType
-) -> int:
+) -> int | float:
     """
     Get organization plan limit for a given usage type
     """
@@ -90,5 +90,8 @@ def get_organization_plan_limit(
     if relevant_limit is None:
         # TODO: get the limits from the community plan, overrides
         relevant_limit = 2000
+    # Limits in Stripe metadata are strings. They may be numbers or 'unlimited'
+    if relevant_limit == 'unlimited':
+        return inf
 
-    return relevant_limit
+    return int(relevant_limit)

kobo/apps/trackers/tests/test_utils.py

@@ -88,7 +88,7 @@ def _make_payment(
     def test_organization_usage_utils(self, usage_type):
         usage_key = f'{USAGE_LIMIT_MAP[usage_type]}_limit'
         sub_metadata = {
-            usage_key: 1000,
+            usage_key: '1000',
             'product_type': 'plan',
             'plan_type': 'enterprise',
         }
@@ -97,7 +97,7 @@ def test_organization_usage_utils(self, usage_type):
         )
         addon_metadata = {
             'product_type': 'addon_onetime',
-            usage_key: 2000,
+            usage_key: '2000',
             'valid_tags': 'all',
         }
         product, price = self._create_product(addon_metadata)