testing locally, fix the tier checks and first test

Author: ajay-sentry

api/internal/owner/serializers.py

@@ -346,7 +346,7 @@ def update(self, instance: Owner, validated_data: Dict[str, Any]) -> object:
             )
 
             sentry_plans = Plan.objects.filter(
-                tier=TierName.SENTRY.value, is_active=True
+                tier__tier_name=TierName.SENTRY.value, is_active=True
             )
 
             if desired_plan["value"] in sentry_plans:

api/internal/tests/test_pagination.py

@@ -1,14 +1,18 @@
 from rest_framework.reverse import reverse
 from rest_framework.test import APITestCase
+from shared.django_apps.codecov_auth.tests.factories import PlanFactory, TierFactory
 from shared.django_apps.core.tests.factories import OwnerFactory
+from shared.plan.constants import TierName
 
 from utils.test_utils import Client
 
 
 class PageNumberPaginationTests(APITestCase):
     def setUp(self):
         self.client = Client()
-        self.owner = OwnerFactory(plan="users-free", plan_user_count=5)
+        tier = TierFactory(tier_name=TierName.BASIC.value)
+        plan = PlanFactory(tier=tier, is_active=True)
+        self.owner = OwnerFactory(plan=plan.name, plan_user_count=5)
         self.users = [
             OwnerFactory(organizations=[self.owner.ownerid]),
             OwnerFactory(organizations=[self.owner.ownerid]),

billing/helpers.py

@@ -8,7 +8,7 @@
 def on_enterprise_plan(owner: Owner) -> bool:
     return settings.IS_ENTERPRISE or (
         owner.plan
-        in Plan.objects.filter(tier=TierName.ENTERPRISE.value).values_list(
+        in Plan.objects.filter(tier__tier_name=TierName.ENTERPRISE.value).values_list(
             "name", flat=True
         )
     )

graphql_api/types/enums/tier_name.graphql

@@ -3,4 +3,6 @@ enum TierName {
   TEAM
   PRO
   ENTERPRISE
+  SENTRY
+  TRIAL
 }

requirements.in

@@ -26,7 +26,7 @@ freezegun
 google-cloud-pubsub
 gunicorn>=22.0.0
 https://github.com/codecov/opentelem-python/archive/refs/tags/v0.0.4a1.tar.gz#egg=codecovopentelem
-https://github.com/codecov/shared/archive/fe16480b3646a616ff412d5c0a28cafd2c7104c1.tar.gz#egg=shared
+https://github.com/codecov/shared/archive/3dcd8efa027e6f45e0cbfd2c08a936ec5a68a57c.tar.gz#egg=shared
 https://github.com/photocrowd/django-cursor-pagination/archive/f560902696b0c8509e4d95c10ba0d62700181d84.tar.gz
 idna>=3.7
 minio

requirements.txt

@@ -416,7 +416,7 @@ sentry-sdk[celery]==2.13.0
     #   shared
 setproctitle==1.1.10
     # via -r requirements.in
-shared @ https://github.com/codecov/shared/archive/fe16480b3646a616ff412d5c0a28cafd2c7104c1.tar.gz
+shared @ https://github.com/codecov/shared/archive/3dcd8efa027e6f45e0cbfd2c08a936ec5a68a57c.tar.gz
     # via -r requirements.in
 simplejson==3.17.2
     # via -r requirements.in