hub/services/tests/test_pricing_edge_cases.py

@@ -1,6 +1,5 @@
 from decimal import Decimal
 from django.test import TestCase
-from django.core.exceptions import ValidationError
 from django.utils import timezone
 from datetime import timedelta
 
@@ -10,16 +9,12 @@ from ..models.services import Service
 from ..models.pricing import (
     ComputePlan,
     ComputePlanPrice,
-    StoragePlan,
-    StoragePlanPrice,
     ProgressiveDiscountModel,
     DiscountTier,
     VSHNAppCatPrice,
     VSHNAppCatBaseFee,
     VSHNAppCatUnitRate,
     VSHNAppCatAddon,
-    VSHNAppCatAddonBaseFee,
-    VSHNAppCatAddonUnitRate,
     ExternalPricePlans,
 )
 
@@ -163,7 +158,8 @@ class PricingEdgeCasesTestCase(TestCase):
         )
 
         # Should return None when price doesn't exist
-        price = addon.get_price(Currency.CHF)
+        # For BASE_FEE addons, service_level is required
+        price = addon.get_price(Currency.CHF, service_level="standard")
         self.assertIsNone(price)
 
     def test_compute_plan_with_validity_dates(self):