From 4c4b010f4446914c30ce44f5a3443fb49e205f5c Mon Sep 17 00:00:00 2001
From: Colin Maudry <colin@maudry.com>
Date: Sat, 18 Apr 2026 21:39:36 +0200
Subject: [PATCH] =?UTF-8?q?Correction=20des=20tests=20avec=20donn=C3=A9es?=
 =?UTF-8?q?=20db?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/test_db.py   | 28 ++++++++++++++++++++--------
 tests/test_main.py |  9 +++------
 2 files changed, 23 insertions(+), 14 deletions(-)

diff --git a/tests/test_db.py b/tests/test_db.py
index 6ce740a..2edd1c5 100644
--- a/tests/test_db.py
+++ b/tests/test_db.py
@@ -81,11 +81,15 @@ def built_db(tmp_path, monkeypatch):
                 "uid": "1",
                 "id": "1",
                 "objet": "Travaux",
-                "acheteur_id": "A1",
-                "acheteur_nom": "Mairie",
+                "acheteur_id": "123",
+                "acheteur_nom": "ACHETEUR 1",
                 "acheteur_departement_code": "75",
-                "titulaire_id": "T1",
-                "titulaire_nom": "Entreprise",
+                "acheteur_departement_nom": "Paris",
+                "acheteur_commune_nom": "Paris",
+                "titulaire_commune_nom": "Paris",
+                "titulaire_departement_nom": "Paris",
+                "titulaire_id": "345",
+                "titulaire_nom": "TITULAIRE 1",
                 "titulaire_departement_code": "35",
                 "titulaire_typeIdentifiant": "SIRET",
                 "montant": 1000.0,
@@ -97,10 +101,14 @@ def built_db(tmp_path, monkeypatch):
                 "uid": "2",
                 "id": "2",
                 "objet": "Études",
-                "acheteur_id": "A1",
-                "acheteur_nom": "Mairie",
+                "acheteur_id": "123",
+                "acheteur_nom": "ACHETEUR 1",
                 "acheteur_departement_code": "75",
-                "titulaire_id": "T2",
+                "acheteur_departement_nom": "Paris",
+                "acheteur_commune_nom": "Paris",
+                "titulaire_commune_nom": "Paris",
+                "titulaire_departement_nom": "Paris",
+                "titulaire_id": "567",
                 "titulaire_nom": None,
                 "titulaire_departement_code": "75",
                 "titulaire_typeIdentifiant": "SIRET",
@@ -116,6 +124,10 @@ def built_db(tmp_path, monkeypatch):
                 "acheteur_id": "A2",
                 "acheteur_nom": None,
                 "acheteur_departement_code": "13",
+                "acheteur_departement_nom": "Paris",
+                "acheteur_commune_nom": "Paris",
+                "titulaire_commune_nom": "Paris",
+                "titulaire_departement_nom": "Paris",
                 "titulaire_id": "T3",
                 "titulaire_nom": "Autre",
                 "titulaire_departement_code": "13",
@@ -188,7 +200,7 @@ def test_query_marches_returns_polars_frame(built_db, monkeypatch):
     importlib.reload(src.db)
     from src.db import query_marches
 
-    frame = query_marches("acheteur_id = ?", ("A1",))
+    frame = query_marches("acheteur_id = ?", ("123",))
     assert isinstance(frame, pl.DataFrame)
     assert frame.height == 2
     assert set(frame["uid"].to_list()) == {"1", "2"}
diff --git a/tests/test_main.py b/tests/test_main.py
index 0eaaec1..7f55a02 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -52,7 +52,6 @@ def test_002_filter_persistence(dash_duo: DashComposite):
         return _filter_input
 
     for page in ["tableau", "acheteurs/123", "titulaires/345"]:
-        print("page:", page)
         filter_input = open_page_and_check_filter_input()
         filter_input.send_keys("11")  # a UID that doesn't exist
         filter_input.send_keys(Keys.ENTER)
@@ -91,7 +90,7 @@ def test_003_tableau_download(dash_duo: DashComposite):
 def test_004_add_links_observatoire_acheteur():
     import polars as pl
 
-    from src.utils import add_links
+    from src.utils.table import add_links
 
     dff = pl.DataFrame(
         {
@@ -116,7 +115,7 @@ def test_004_add_links_observatoire_acheteur():
 def test_005_add_links_observatoire_titulaire():
     import polars as pl
 
-    from src.utils import add_links
+    from src.utils.table import add_links
 
     dff = pl.DataFrame(
         {
@@ -219,9 +218,7 @@ def test_008_search_to_observatoire(dash_duo: DashComposite):
 def test_010_observatoire_montant_filter():
     import datetime
 
-    import polars as pl
-
-    from src.utils import prepare_dashboard_data
+    from src.utils.data import prepare_dashboard_data
 
     data = pl.DataFrame(
         {