fix fixture terms_from_db_example

gestaogovbr · Aug 27, 2024 · 0e00319 · 0e00319
1 parent a178305
commit 0e00319
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/tests/parsers_test.py b/tests/parsers_test.py
@@ -126,18 +126,22 @@
                 "dataset": None,
                 "search": [
                     {
-                        "terms": [],
+                        "terms": {
+                            "from_airflow_variable": None,
+                            "from_db_select": {
+                                "sql": (
+                                    "SELECT 'cloroquina' as TERMO, 'Ações inefetivas' as GRUPO "
+                                    "UNION SELECT 'ivermectina' as TERMO, 'Ações inefetivas' as GRUPO "
+                                    "UNION SELECT 'vacina contra covid' as TERMO, 'Ações efetivas' as GRUPO "
+                                    "UNION SELECT 'higienização das mãos' as TERMO, 'Ações efetivas' as GRUPO "
+                                    "UNION SELECT 'uso de máscara' as TERMO, 'Ações efetivas' as GRUPO "
+                                    "UNION SELECT 'distanciamento social' as TERMO, 'Ações efetivas' as GRUPO\n"
+                                ),
+                                "conn_id": "example_database_conn",
+                            }
+                        },
                         "header": None,
                         "sources": ["DOU"],
-                        "sql": (
-                            "SELECT 'cloroquina' as TERMO, 'Ações inefetivas' as GRUPO "
-                            "UNION SELECT 'ivermectina' as TERMO, 'Ações inefetivas' as GRUPO "
-                            "UNION SELECT 'vacina contra covid' as TERMO, 'Ações efetivas' as GRUPO "
-                            "UNION SELECT 'higienização das mãos' as TERMO, 'Ações efetivas' as GRUPO "
-                            "UNION SELECT 'uso de máscara' as TERMO, 'Ações efetivas' as GRUPO "
-                            "UNION SELECT 'distanciamento social' as TERMO, 'Ações efetivas' as GRUPO\n"
-                        ),
-                        "conn_id": "example_database_conn",
                         "territory_id": None,
                         "dou_sections": ["TODOS"],
                         "date": "MES",