reorg tests

dags/sources/tasks/airflow_logic/operators.py

@@ -2,19 +2,20 @@
 
 from airflow import DAG
 from airflow.models.baseoperator import chain
+from sources.tasks.airflow_logic.db_data_prepare_task import db_data_prepare_task
 from sources.tasks.airflow_logic.db_read_acteur_task import db_read_acteur_task
 from sources.tasks.airflow_logic.db_read_propositions_max_id_task import (
     db_read_propositions_max_id_task,
 )
-from sources.tasks.airflow_logic.get_acteur_to_delete_task import (
-    propose_acteur_to_delete_task,
-)
 from sources.tasks.airflow_logic.propose_acteur_changes_task import (
     propose_acteur_changes_task,
 )
 from sources.tasks.airflow_logic.propose_acteur_services_task import (
     propose_acteur_services_task,
 )
+from sources.tasks.airflow_logic.propose_acteur_to_delete_task import (
+    propose_acteur_to_delete_task,
+)
 from sources.tasks.airflow_logic.propose_labels_task import propose_labels_task
 from sources.tasks.airflow_logic.propose_services_sous_categories_task import (
     propose_services_sous_categories_task,
@@ -23,7 +24,6 @@
 from sources.tasks.airflow_logic.read_mapping_from_postgres_task import (
     read_mapping_from_postgres_task,
 )
-from sources.tasks.airflow_logic.serialize_to_json_task import db_data_prepare_task
 from sources.tasks.airflow_logic.source_config_validate_task import (
     source_config_validate_task,
 )

dags_unit_tests/conftest.py

@@ -1,6 +1,115 @@
 import sys
+from datetime import datetime
 from pathlib import Path
 
+import pandas as pd
+import pytest
+
 
 def pytest_configure(config):
     sys.path.insert(0, str(Path(__file__).resolve().parent.parent / "dags"))
+
+
+@pytest.fixture
+def df_proposition_services():
+    return pd.DataFrame(
+        {
+            "action_id": [1, 3, 1, 3],
+            "acteur_id": [1, 1, 2, 2],
+            "action": ["reparer", "trier", "reparer", "trier"],
+            "acteur_service": [
+                "Service de réparation",
+                "Collecte par une structure spécialisée",
+                "Service de réparation",
+                "Collecte par une structure spécialisée",
+            ],
+            "sous_categories": [
+                "téléphones portables",
+                "téléphones portables",
+                "ecrans",
+                "ecrans",
+            ],
+            "id": [1, 2, 3, 4],
+        }
+    )
+
+
+@pytest.fixture
+def df_proposition_services_sous_categories():
+    return pd.DataFrame(
+        {
+            "propositionservice_id": [1, 2, 3, 4],
+            "souscategorieobjet_id": [102, 102, 101, 101],
+            "souscategorie": [
+                "téléphones portables",
+                "téléphones portables",
+                "ecrans",
+                "ecrans",
+            ],
+        }
+    )
+
+
+@pytest.fixture
+def sources_id_by_code():
+    return {
+        "source1": 101,
+        "source2": 102,
+        "cma_reparacteur": 103,
+    }
+
+
+@pytest.fixture
+def acteurtype_id_by_code():
+    return {
+        "ess": 201,
+        "commerce": 202,
+        "artisan": 203,
+        "pav_prive": 204,
+    }
+
+
+@pytest.fixture
+def actions_id_by_code():
+    return {"reparer": 1, "donner": 2, "trier": 3}
+
+
+@pytest.fixture
+def labelqualite_id_by_code():
+    return {"ess": 1, "label_bonus": 2, "reparacteur": 3}
+
+
+@pytest.fixture
+def acteurservice_id_by_code():
+    return {"service_de_reparation": 10, "structure_de_collecte": 20}
+
+
+@pytest.fixture
+def df_empty_acteurs_from_db():
+    return pd.DataFrame(
+        columns=[
+            "identifiant_unique",
+            "source_id",
+            "statut",
+            "cree_le",
+            "modifie_le",
+        ]
+    )
+
+
+@pytest.fixture
+def df_acteurs_from_db():
+    return pd.DataFrame(
+        {
+            "identifiant_unique": ["id1", "id2"],
+            "source_id": [101, 102],
+            "statut": ["ACTIF", "ACTIF"],
+            "cree_le": [datetime(2024, 1, 1), datetime(2024, 1, 1)],
+            "modifie_le": [datetime(2024, 1, 1), datetime(2024, 1, 1)],
+        }
+    )
+
+
+@pytest.fixture
+def souscategorieobjet_code_by_id():
+    return {"ecran": 101, "smartphone, tablette et console": 102}