fix tests

Author: johan12345

src/evmap_backend/data_sources/datex2/source.py

@@ -99,29 +99,29 @@ def get_data(self) -> str:
 
 class Datex2MobilithekEcoMovementDatex2DataSource(BaseMobilithekDatex2DataSource):
     id = "mobilithek_ecomovement"
-    subscription_id = os.environ["MOBILITHEK_ECOMOVEMENT_STATIC_SUBSCRIPTION_ID"]
+    subscription_id = os.environ.get("MOBILITHEK_ECOMOVEMENT_STATIC_SUBSCRIPTION_ID")
     ignore_encoding = True
 
 
 class Datex2MobilithekEnbwDataSource(BaseMobilithekDatex2DataSource):
     id = "mobilithek_enbw"
-    subscription_id = os.environ["MOBILITHEK_ENBW_STATIC_SUBSCRIPTION_ID"]
+    subscription_id = os.environ.get("MOBILITHEK_ENBW_STATIC_SUBSCRIPTION_ID")
     parser = Datex2JsonParser()
 
 
 class Datex2MobilithekLadenetzDataSource(BaseMobilithekDatex2DataSource):
     id = "mobilithek_ladenetz"
-    subscription_id = os.environ["MOBILITHEK_LADENETZ_STATIC_SUBSCRIPTION_ID"]
+    subscription_id = os.environ.get("MOBILITHEK_LADENETZ_STATIC_SUBSCRIPTION_ID")
 
 
 class Datex2MobilithekUlmDataSource(BaseMobilithekDatex2DataSource):
     id = "mobilithek_ulm"
-    subscription_id = os.environ["MOBILITHEK_ULM_STATIC_SUBSCRIPTION_ID"]
+    subscription_id = os.environ.get("MOBILITHEK_ULM_STATIC_SUBSCRIPTION_ID")
 
 
 class Datex2MobilithekWirelaneDataSource(BaseMobilithekDatex2DataSource):
     id = "mobilithek_wirelane"
-    subscription_id = os.environ["MOBILITHEK_WIRELANE_STATIC_SUBSCRIPTION_ID"]
+    subscription_id = os.environ.get("MOBILITHEK_WIRELANE_STATIC_SUBSCRIPTION_ID")
     parser = Datex2JsonParser()
 
 

src/tests/data_sources/test_datex2.py

@@ -9,9 +9,7 @@
 from cryptography.x509.verification import VerificationError
 from django.http import HttpRequest
 
-from evmap_backend.data_sources.datex2.source import (
-    Datex2MobilithekEcoMovementDatex2DataSource,
-)
+from evmap_backend.data_sources.datex2.source import BaseMobilithekDatex2DataSource
 
 
 @pytest.fixture
@@ -59,7 +57,11 @@ def random_x509_certificate():
 
 
 def test_verify_push_mobilithek(random_x509_certificate):
-    ds = Datex2MobilithekEcoMovementDatex2DataSource()
+    class DummyMobilithekDatex2DataSource(BaseMobilithekDatex2DataSource):
+        id = "dummy"
+        subscription_id = "dummy"
+
+    ds = DummyMobilithekDatex2DataSource()
 
     # no certificate
     req = HttpRequest()