From 25d4164bae8f51821779131bb4e41a5f6f4cd40b Mon Sep 17 00:00:00 2001
From: Tobias Brunner <tobias.brunner@vshn.ch>
Date: Mon, 23 Jun 2025 14:22:33 +0200
Subject: [PATCH 1/2] only run tests on main and PRs

---
 .forgejo/workflows/test.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.forgejo/workflows/test.yaml b/.forgejo/workflows/test.yaml
index a0cb310..1bc048c 100644
--- a/.forgejo/workflows/test.yaml
+++ b/.forgejo/workflows/test.yaml
@@ -2,7 +2,7 @@ name: Django Tests
 
 on:
   push:
-    branches: ["*"]
+    branches: [main]
   pull_request:
 
 jobs:
@@ -31,4 +31,4 @@ jobs:
             -w /app \
             -e SECRET_KEY=dummysecretkey \
             website:test \
-            sh -c 'uv run --extra dev manage.py migrate --noinput && uv run --extra dev manage.py test hub.services.tests --verbosity=2'
\ No newline at end of file
+            sh -c 'uv run --extra dev manage.py migrate --noinput && uv run --extra dev manage.py test hub.services.tests --verbosity=2'

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 2/2] 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):