update query count

apache · Oct 18, 2024 · ebfa65c · ebfa65c
1 parent b16d951
commit ebfa65c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/models/test_serialized_dag.py b/tests/models/test_serialized_dag.py
@@ -62,6 +62,7 @@ class TestSerializedDagModel:
         ],
     )
     def setup_test_cases(self, request, monkeypatch):
+        db.clear_db_dags()
         db.clear_db_serialized_dags()
         with mock.patch("airflow.models.serialized_dag.COMPRESS_SERIALIZED_DAGS", request.param):
             yield
@@ -191,7 +192,9 @@ def test_bulk_sync_to_db(self):
             DAG("dag_3", schedule=None),
         ]
         DAG.bulk_write_to_db(dags)
-        with assert_queries_count(15):  # we also write to dag_version and dag_code tables
+        # we also write to dag_version and dag_code tables
+        # in dag_version, we search for unique version_name too
+        with assert_queries_count(21):
             SDM.bulk_sync_to_db(dags)
 
     @pytest.mark.skip_if_database_isolation_mode  # Does not work in db isolation mode