import status to avoid surprises in tests

opengisch · Jan 23, 2024 · a5ddf9a · a5ddf9a
1 parent b793159
commit a5ddf9a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/comptages/core/report.py b/comptages/core/report.py
@@ -49,9 +49,12 @@ def _prepare_default_reports(
  sections = models.Section.objects.filter(
  lane__id_installation__count=count
  ).distinct()
+ assert sections
 
  mondays_qty = len(list(_mondays_of_count(count)))
  mondays = _mondays_of_count(count)
+ assert mondays
+
  for section in sections:
  for i, monday in enumerate(mondays):
  progress = int(100 / mondays_qty * (i - 1))

diff --git a/comptages/datamodel/management/commands/importdata.py b/comptages/datamodel/management/commands/importdata.py
@@ -15,6 +15,7 @@
  Category,
  Class,
  ClassCategory,
+ CountDetail,
  Device,
  Installation,
  Model,
@@ -438,3 +439,7 @@ def import_count(self):
  import_file(str(file), count)
 
  print(f"Imported {len(files)} count files!")
+ print(
+ "Setting all count details' status to 'definitive' to dodge a few surprises in tests..."
+ )
+ CountDetail.objects.update(import_status=0)