From 9b34e04d13b4d7ca68f72c6b5e2e602ae6330083 Mon Sep 17 00:00:00 2001
From: Stefan Heese <stefan.heese@hochfrequenz.de>
Date: Wed, 8 Nov 2023 10:28:16 +0100
Subject: [PATCH] =?UTF-8?q?=E2=9C=85Test=20Vertragskonditionen?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/test_vertragskonditionen.py | 56 ++++++++++++-------------------
 1 file changed, 21 insertions(+), 35 deletions(-)

diff --git a/tests/test_vertragskonditionen.py b/tests/test_vertragskonditionen.py
index 90a9aea81..0304f8f71 100644
--- a/tests/test_vertragskonditionen.py
+++ b/tests/test_vertragskonditionen.py
@@ -1,7 +1,10 @@
 from datetime import datetime, timezone
 from decimal import Decimal
 
+import pytest
+
 from bo4e import Vertragskonditionen, Zeiteinheit, Zeitraum
+from tests.serialization_helper import assert_serialization_roundtrip
 
 example_vertragskonditionen = Vertragskonditionen(
     beschreibung="Foobar",
@@ -17,40 +20,23 @@
 
 
 class TestVertragskonditionen:
-    def test_vertragskonditionen_with_optional_attributes(self) -> None:
+    @pytest.mark.parametrize(
+        "vertragskonditionen",
+        [
+            pytest.param(
+                Vertragskonditionen(
+                    example_vertragskonditionen,
+                ),
+            ),
+            pytest.param(
+                Vertragskonditionen(
+                    Vertragskonditionen(),
+                ),
+            ),
+        ],
+    )
+    def test_serialization_roundtrip(self, vertragskonditionen: Vertragskonditionen) -> None:
         """
-        Test de-/serialisation of Vertragskonditionen (only has optional attributes).
+        Test de-/serialisation of Vertragskonditionen.
         """
-        vertragskonditionen = example_vertragskonditionen
-
-        json_string = vertragskonditionen.model_dump_json(by_alias=True)
-
-        assert "Foobar" in json_string
-        assert "3" in json_string
-        assert "2013-10-11T00:00:00Z" in json_string
-        assert "WOCHE" in json_string
-        assert "TAG" in json_string
-        assert "14" in json_string
-
-        vertragskonditionen_deserialized = Vertragskonditionen.model_validate_json(json_string)
-
-        assert isinstance(vertragskonditionen_deserialized.beschreibung, str)
-        assert vertragskonditionen_deserialized.beschreibung == "Foobar"
-        assert isinstance(vertragskonditionen_deserialized.anzahl_abschlaege, Decimal)
-        assert vertragskonditionen_deserialized.anzahl_abschlaege == Decimal(3)
-        assert isinstance(vertragskonditionen_deserialized.vertragslaufzeit, Zeitraum)
-        assert vertragskonditionen_deserialized.vertragslaufzeit == Zeitraum(
-            startdatum=datetime(2012, 9, 21, tzinfo=timezone.utc), enddatum=datetime(2013, 10, 11, tzinfo=timezone.utc)
-        )
-        assert isinstance(vertragskonditionen_deserialized.vertragsverlaengerung, Zeitraum)
-        assert vertragskonditionen_deserialized.vertragsverlaengerung == Zeitraum(
-            einheit=Zeiteinheit.TAG, dauer=Decimal(14)
-        )
-
-    def test_vertragskonditionen_empty(self) -> None:
-        """
-        Test empty Vertragskonditionen (only has optional attributes).
-        """
-        empty_vertragskonditionen = Vertragskonditionen()
-
-        assert isinstance(empty_vertragskonditionen, Vertragskonditionen)
+        assert_serialization_roundtrip(vertragskonditionen)