fix all tests except account_viewset

Author: ajay-sentry

billing/helpers.py

@@ -1,8 +1,8 @@
 from django.conf import settings
 from django.db.models import QuerySet
-from shared.plan.constants import TierName, PlanName, PlanPrice
-from shared.django_apps.codecov_auth.tests.factories import PlanFactory, TierFactory
 from shared.django_apps.codecov_auth.models import BillingRate
+from shared.django_apps.codecov_auth.tests.factories import PlanFactory, TierFactory
+from shared.plan.constants import PlanName, PlanPrice, TierName
 
 from codecov_auth.models import Owner, Plan
 
@@ -30,7 +30,6 @@ def get_all_admins_for_owners(owners: QuerySet[Owner]):
     return admins
 
 
-
 def mock_all_plans_and_tiers():
     trial_tier = TierFactory(tier_name=TierName.TRIAL.value)
     PlanFactory(
@@ -179,4 +178,4 @@ def mock_all_plans_and_tiers():
         billing_rate=BillingRate.ANNUALLY.value,
         base_unit_price=PlanPrice.YEARLY.value,
         paid_plan=True,
-    )
+    )

billing/tests/test_helpers.py

@@ -1,7 +1,12 @@
 from django.test import TestCase, override_settings
-from shared.django_apps.core.tests.factories import OwnerFactory
-from shared.plan.constants import ENTERPRISE_CLOUD_USER_PLAN_REPRESENTATIONS, TierName, PlanName
 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 (
+    ENTERPRISE_CLOUD_USER_PLAN_REPRESENTATIONS,
+    PlanName,
+    TierName,
+)
+
 from billing.helpers import on_enterprise_plan
 
 

billing/tests/test_views.py

@@ -11,6 +11,8 @@
 from shared.django_apps.core.tests.factories import OwnerFactory, RepositoryFactory
 from shared.plan.constants import PlanName
 
+from billing.helpers import mock_all_plans_and_tiers
+
 from ..constants import StripeHTTPHeaders
 
 
@@ -70,16 +72,19 @@ def __getitem__(self, key):
 
 class StripeWebhookHandlerTests(APITestCase):
     def setUp(self):
+        mock_all_plans_and_tiers()
         self.owner = OwnerFactory(
-            stripe_customer_id="cus_123", stripe_subscription_id="sub_123"
+            stripe_customer_id="cus_123",
+            stripe_subscription_id="sub_123",
         )
 
     # Creates a second owner that shares billing details with self.owner.
     # This is used to test the case where owners are manually set to share a
     # subscription in Stripe.
     def add_second_owner(self):
         self.other_owner = OwnerFactory(
-            stripe_customer_id="cus_123", stripe_subscription_id="sub_123"
+            stripe_customer_id="cus_123",
+            stripe_subscription_id="sub_123",
         )
 
     def _send_event(self, payload, errorSig=None):
@@ -467,7 +472,7 @@ def test_invoice_payment_failed_sends_email_to_admins_no_card(
         mocked_send_email.assert_has_calls(expected_calls)
 
     def test_customer_subscription_deleted_sets_plan_to_free(self):
-        self.owner.plan = "users-inappy"
+        self.owner.plan = PlanName.CODECOV_PRO_YEARLY.value
         self.owner.plan_user_count = 20
         self.owner.save()
 
@@ -492,10 +497,10 @@ def test_customer_subscription_deleted_sets_plan_to_free(self):
 
     def test_customer_subscription_deleted_sets_plan_to_free_mutliple_owner(self):
         self.add_second_owner()
-        self.owner.plan = "users-inappy"
+        self.owner.plan = PlanName.CODECOV_PRO_YEARLY.value
         self.owner.plan_user_count = 20
         self.owner.save()
-        self.other_owner.plan = "users-inappy"
+        self.other_owner.plan = PlanName.CODECOV_PRO_YEARLY.value
         self.other_owner.plan_user_count = 20
         self.other_owner.save()
 
@@ -540,7 +545,7 @@ def test_customer_subscription_deleted_deactivates_all_repos(self):
                     "object": {
                         "id": self.owner.stripe_subscription_id,
                         "customer": self.owner.stripe_customer_id,
-                        "plan": {"name": "users-inappm"},
+                        "plan": {"name": PlanName.CODECOV_PRO_MONTHLY.value},
                     }
                 },
             }
@@ -577,7 +582,7 @@ def test_customer_subscription_deleted_deactivates_all_repos_multiple_owner(self
                     "object": {
                         "id": self.owner.stripe_subscription_id,
                         "customer": self.owner.stripe_customer_id,
-                        "plan": {"name": "users-inappm"},
+                        "plan": {"name": PlanName.CODECOV_PRO_MONTHLY.value},
                     }
                 },
             }
@@ -598,7 +603,7 @@ def test_customer_subscription_deleted_deactivates_all_repos_multiple_owner(self
 
     @patch("logging.Logger.info")
     def test_customer_subscription_deleted_no_customer(self, log_info_mock):
-        self.owner.plan = "users-inappy"
+        self.owner.plan = PlanName.CODECOV_PRO_MONTHLY.value
         self.owner.plan_user_count = 20
         self.owner.save()
 
@@ -688,7 +693,7 @@ def test_customer_subscription_created_sets_plan_info(self):
 
         stripe_subscription_id = "FOEKDCDEQ"
         stripe_customer_id = "sdo050493"
-        plan_name = "users-pr-inappy"
+        plan_name = PlanName.CODECOV_PRO_YEARLY.value
         quantity = 20
 
         self._send_event(