Skip to content

Commit b7c7349

Browse files
committed
Fix linter issues
1 parent 75d24ae commit b7c7349

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

metadata-ingestion/tests/unit/test_kafka_connect.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)