@@ -3353,6 +3353,7 @@ def test_infer_mappings_single_table_mode(self) -> None:
33533353 assert lineages [0 ].source_dataset == "users"
33543354 assert lineages [0 ].target_dataset == "users_topic"
33553355 assert lineages [0 ].source_platform == "postgres"
3356+ assert lineages [0 ].job_property_bag is not None
33563357 assert lineages [0 ].job_property_bag ["mode" ] == "single_table"
33573358
33583359 def test_infer_mappings_multi_table_mode (self ) -> None :
@@ -3381,11 +3382,14 @@ def test_infer_mappings_multi_table_mode(self) -> None:
33813382 )
33823383
33833384 assert len (lineages ) == 3
3384-
3385+
33853386 # Check products lineage
3386- products_lineage = next (lineage for lineage in lineages if "products" in lineage .target_dataset )
3387+ products_lineage = next (
3388+ lineage for lineage in lineages if "products" in lineage .target_dataset
3389+ )
33873390 assert products_lineage .source_dataset == "products"
33883391 assert products_lineage .target_dataset == "db_products"
3392+ assert products_lineage .job_property_bag is not None
33893393 assert products_lineage .job_property_bag ["mode" ] == "multi_table"
33903394
33913395 def test_infer_mappings_with_topic_prefix_only (self ) -> None :
@@ -3416,6 +3420,7 @@ def test_infer_mappings_with_topic_prefix_only(self) -> None:
34163420 assert len (lineages ) == 2
34173421 assert lineages [0 ].source_dataset == "events"
34183422 assert lineages [0 ].target_dataset == "staging_events"
3423+ assert lineages [0 ].job_property_bag is not None
34193424 assert lineages [0 ].job_property_bag ["mode" ] == "inferred"
34203425
34213426 def test_infer_mappings_with_schema_qualified_tables (self ) -> None :
0 commit comments