Correction des tests avec données db

This commit is contained in:
Colin Maudry
2026-04-18 21:39:36 +02:00
parent 9d9760e596
commit 4c4b010f44
2 changed files with 23 additions and 14 deletions
+20 -8
View File
@@ -81,11 +81,15 @@ def built_db(tmp_path, monkeypatch):
"uid": "1", "uid": "1",
"id": "1", "id": "1",
"objet": "Travaux", "objet": "Travaux",
"acheteur_id": "A1", "acheteur_id": "123",
"acheteur_nom": "Mairie", "acheteur_nom": "ACHETEUR 1",
"acheteur_departement_code": "75", "acheteur_departement_code": "75",
"titulaire_id": "T1", "acheteur_departement_nom": "Paris",
"titulaire_nom": "Entreprise", "acheteur_commune_nom": "Paris",
"titulaire_commune_nom": "Paris",
"titulaire_departement_nom": "Paris",
"titulaire_id": "345",
"titulaire_nom": "TITULAIRE 1",
"titulaire_departement_code": "35", "titulaire_departement_code": "35",
"titulaire_typeIdentifiant": "SIRET", "titulaire_typeIdentifiant": "SIRET",
"montant": 1000.0, "montant": 1000.0,
@@ -97,10 +101,14 @@ def built_db(tmp_path, monkeypatch):
"uid": "2", "uid": "2",
"id": "2", "id": "2",
"objet": "Études", "objet": "Études",
"acheteur_id": "A1", "acheteur_id": "123",
"acheteur_nom": "Mairie", "acheteur_nom": "ACHETEUR 1",
"acheteur_departement_code": "75", "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_nom": None,
"titulaire_departement_code": "75", "titulaire_departement_code": "75",
"titulaire_typeIdentifiant": "SIRET", "titulaire_typeIdentifiant": "SIRET",
@@ -116,6 +124,10 @@ def built_db(tmp_path, monkeypatch):
"acheteur_id": "A2", "acheteur_id": "A2",
"acheteur_nom": None, "acheteur_nom": None,
"acheteur_departement_code": "13", "acheteur_departement_code": "13",
"acheteur_departement_nom": "Paris",
"acheteur_commune_nom": "Paris",
"titulaire_commune_nom": "Paris",
"titulaire_departement_nom": "Paris",
"titulaire_id": "T3", "titulaire_id": "T3",
"titulaire_nom": "Autre", "titulaire_nom": "Autre",
"titulaire_departement_code": "13", "titulaire_departement_code": "13",
@@ -188,7 +200,7 @@ def test_query_marches_returns_polars_frame(built_db, monkeypatch):
importlib.reload(src.db) importlib.reload(src.db)
from src.db import query_marches from src.db import query_marches
frame = query_marches("acheteur_id = ?", ("A1",)) frame = query_marches("acheteur_id = ?", ("123",))
assert isinstance(frame, pl.DataFrame) assert isinstance(frame, pl.DataFrame)
assert frame.height == 2 assert frame.height == 2
assert set(frame["uid"].to_list()) == {"1", "2"} assert set(frame["uid"].to_list()) == {"1", "2"}
+3 -6
View File
@@ -52,7 +52,6 @@ def test_002_filter_persistence(dash_duo: DashComposite):
return _filter_input return _filter_input
for page in ["tableau", "acheteurs/123", "titulaires/345"]: for page in ["tableau", "acheteurs/123", "titulaires/345"]:
print("page:", page)
filter_input = open_page_and_check_filter_input() filter_input = open_page_and_check_filter_input()
filter_input.send_keys("11") # a UID that doesn't exist filter_input.send_keys("11") # a UID that doesn't exist
filter_input.send_keys(Keys.ENTER) 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(): def test_004_add_links_observatoire_acheteur():
import polars as pl import polars as pl
from src.utils import add_links from src.utils.table import add_links
dff = pl.DataFrame( dff = pl.DataFrame(
{ {
@@ -116,7 +115,7 @@ def test_004_add_links_observatoire_acheteur():
def test_005_add_links_observatoire_titulaire(): def test_005_add_links_observatoire_titulaire():
import polars as pl import polars as pl
from src.utils import add_links from src.utils.table import add_links
dff = pl.DataFrame( dff = pl.DataFrame(
{ {
@@ -219,9 +218,7 @@ def test_008_search_to_observatoire(dash_duo: DashComposite):
def test_010_observatoire_montant_filter(): def test_010_observatoire_montant_filter():
import datetime import datetime
import polars as pl from src.utils.data import prepare_dashboard_data
from src.utils import prepare_dashboard_data
data = pl.DataFrame( data = pl.DataFrame(
{ {