Create test to implementation qd_list_territory_id

gestaogovbr · Oct 24, 2024 · 13c122a · 13c122a
1 parent ad9b030
commit 13c122a
Show file tree

Hide file tree

Showing 2 changed files with 67 additions and 0 deletions.
diff --git a/tests/parsers_test.py b/tests/parsers_test.py
@@ -579,6 +579,53 @@
                 },
             },
         ),
+        (
+            "qd_list_territory_id_example.yaml",
+            {
+                "id": "qd_list_territory_id_example",
+                "description": "DAG de teste com múltiplos territory_id",
+                "schedule": '0 8 * * MON-FRI',
+                "dataset": None,
+                "doc_md": None,
+                "tags": {"dou", "generated_dag"},
+                "owner": [],
+                "search": [
+                    {
+                        "terms": [
+                            "LGPD",
+                            "RIO DE JANEIRO",
+                        ],
+                        "header": "Teste com múltiplos territory_id",
+                        "sources": ["QD"],
+                        "sql": None,
+                        "conn_id": None,
+                        "territory_id": [3300100,3300159,3300209,3305703],
+                        "dou_sections": ["TODOS"],
+                        "search_date": "DIA",
+                        "field": "TUDO",
+                        "is_exact_search": True,
+                        "ignore_signature_match": True,
+                        "force_rematch": True,
+                        "full_text": False,
+                        "use_summary": False,
+                        "department": None,
+                    }
+                ],
+                "report": {
+                    "emails": ["destination@economia.gov.br"],
+                    "subject": "Teste do Ro-dou",
+                    "attach_csv": False,
+                    "discord_webhook": None,
+                    "slack_webhook": None,
+                    "skip_null": False,
+                    "hide_filters": False,
+                    "header_text": None,
+                    "footer_text": None,
+                    "no_results_found_text": "Nenhum dos termos pesquisados "
+                    "foi encontrado nesta consulta",
+                },
+            },
+        ),
     ],
 )
 def test_parse(filepath, result_tuple):

diff --git a/tests/qd_searcher_test.py b/tests/qd_searcher_test.py
@@ -79,3 +79,23 @@ def test_build_query_payload(pre_tags: str,
     ]
 
     assert payload == expected
+
+
+@pytest.mark.parametrize(
+    'territory_id, expected_payload',
+    [
+        (3300100, [('territory_ids', 3300100)]),
+        ([3300100, 3300159], [('territory_ids', 3300100), ('territory_ids', 3300159)]),
+    ]
+)
+def test_search_with_multiple_territory_ids(territory_id, expected_payload):
+    #searcher = QDSearcher()
+    payload = []
+
+    # Simula a lógica que foi alterada para suportar múltiplos IDs de território
+    if isinstance(territory_id, int):
+        territory_id = [territory_id]
+    for terr_id in territory_id:
+        payload.append(('territory_ids', terr_id))
+
+    assert payload == expected_payload