From 0a3837d1e197c37062e5f16b94fbb1c251054932 Mon Sep 17 00:00:00 2001
From: Tobias Brunner <tobias.brunner@vshn.ch>
Date: Mon, 23 Jun 2025 14:22:52 +0200
Subject: [PATCH] set service level for test

---
 hub/services/tests/test_pricing_edge_cases.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/hub/services/tests/test_pricing_edge_cases.py b/hub/services/tests/test_pricing_edge_cases.py
index 8a59104..0363ea8 100644
--- a/hub/services/tests/test_pricing_edge_cases.py
+++ b/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):