add search for mixed DOU and QD in example

fix test for multiple_searches
gestaogovbr · Aug 30, 2024 · e447115 · e447115
1 parent 9bcfe72
commit e447115
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 5 deletions.
diff --git a/dag_confs/examples_and_tests/multiple_searchs_example.yaml b/dag_confs/examples_and_tests/multiple_searchs_example.yaml
@@ -2,12 +2,10 @@ dag:
   id: multiple_searchs_example
   description: DAG de teste com múltiplas buscas
   schedule: 0 8 * * MON-FRI
-  tags:
-    - inlabs
   search:
     - header: "Pesquisa no DOU"
       sources:
-        - INLABS
+        - DOU
       terms:
         - dados abertos
         - governo aberto
@@ -23,6 +21,16 @@ dag:
         - lei de acesso à informação
       force_rematch: On
       ignore_signature_match: On
+    - header: "Pesquisa no DOU e QD (misto)"
+      sources:
+        - DOU
+        - QD
+      terms:
+        - dados abertos
+        - governo aberto
+        - lei de acesso à informação
+      force_rematch: On
+      ignore_signature_match: On
   report:
     emails:
       - [email protected]

diff --git a/tests/parsers_test.py b/tests/parsers_test.py
@@ -409,7 +409,7 @@
                 "dataset": None,
                 "description": "DAG de teste com múltiplas buscas",
                 "doc_md": None,
-                "tags": {"dou", "generated_dag", "inlabs"},
+                "tags": {"dou", "generated_dag"},
                 "owner": [],
                 "search": [
                     {
@@ -419,7 +419,7 @@
                             "lei de acesso à informação",
                         ],
                         "header": "Pesquisa no DOU",
-                        "sources": ["INLABS"],
+                        "sources": ["DOU"],
                         "sql": None,
                         "conn_id": None,
                         "territory_id": None,
@@ -454,6 +454,27 @@
                         "use_summary": False,
                         "department": None,
                     },
+                    {
+                        "terms": [
+                            "dados abertos",
+                            "governo aberto",
+                            "lei de acesso à informação",
+                        ],
+                        "header": "Pesquisa no DOU e QD (misto)",
+                        "sources": ["DOU", "QD"],
+                        "sql": None,
+                        "conn_id": None,
+                        "territory_id": None,
+                        "dou_sections": ["TODOS"],
+                        "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": ["[email protected]"],