Fix convention generator tests

MTES-MCT · Jan 28, 2025 · 31da07c · 31da07c
1 parent a1e2cf1
commit 31da07c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/conventions/tests/services/test_convention_generator.py b/conventions/tests/services/test_convention_generator.py
@@ -59,6 +59,9 @@ def convention_context_keys():
         "liste_des_annexes",
         "mixPLUSinf10_10pc",
         "logements",
+        "logements_corrigee",
+        "logements_corrigee_sans_loyer",
+        "logements_sans_loyer",
         "mixPLUSinf10_30pc",
         "nombre_annees_conventionnement",
         "outre_mer",
@@ -70,6 +73,7 @@ def convention_context_keys():
         "convention",
         "sh_totale",
         "sa_totale",
+        "sc_totale",
         "nb_logements_par_type",
         "locaux_collectifs",
         "loyer_m2",
@@ -194,6 +198,7 @@ def test_compute_total_logement(self):
             "loyer_total": Decimal("1500.00"),
             "sa_totale": 0,
             "sar_totale": 0,
+            "sc_totale": 0,
             "sh_totale": 0,
             "su_totale": 0,
         }
@@ -220,30 +225,35 @@ def test_generate_convention_doc(self):
             lot=convention.lot,
             typologie=TypologieLogement.T2,
             designation="Logement 2",
+            loyer="0",
             import_order=3,
         )
         Logement.objects.create(
             lot=convention.lot,
             typologie=TypologieLogement.T2,
             designation="Logement 1",
+            loyer="0",
             import_order=1,
         )
         Logement.objects.create(
             lot=convention.lot,
             typologie=TypologieLogement.T1,
             designation="Logement 3",
+            loyer="0",
             import_order=2,
         )
         Logement.objects.create(
             lot=convention.lot,
             typologie=TypologieLogement.T1BIS,
             designation="Logement 4",
+            loyer="0",
             import_order=4,
         )
         Logement.objects.create(
             lot=convention.lot,
             typologie=TypologieLogement.T1BIS,
             designation="Logement 34",
+            loyer="0",
             import_order=0,
         )
         convention.programme.nature_logement = NatureLogement.RESISDENCESOCIALE