Test correction

betagouv · Aug 12, 2024 · ed7673e · ed7673e
1 parent 3c7ec91
commit ed7673e
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/data/models/declaration.py b/data/models/declaration.py
@@ -213,11 +213,20 @@ def brevo_parameters(self):
 
     @property
     def expiration_date(self):
-        expirable_statuses = [Declaration.DeclarationStatus.OBJECTION, Declaration.DeclarationStatus.OBSERVATION]
+        expirable_statuses = [
+            Declaration.DeclarationStatus.OBJECTION,
+            Declaration.DeclarationStatus.OBSERVATION,
+            Declaration.DeclarationStatus.ABANDONED,
+        ]
         if self.status not in expirable_statuses:
             return None
         try:
-            latest_snapshot = self.snapshots.latest("creation_date")
+            latest_snapshot = self.snapshots.filter(
+                status__in=[
+                    Declaration.DeclarationStatus.OBJECTION,
+                    Declaration.DeclarationStatus.OBSERVATION,
+                ]
+            ).latest("creation_date")
             expiration_date = latest_snapshot.creation_date + timedelta(days=latest_snapshot.expiration_days)
             return expiration_date
         except Exception as _: