Skip to content

Commit 1478d70

Browse files
authored
fix(ingest): Fix breaking smoke test on stateful ingestion (#8128)
1 parent b216caa commit 1478d70

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

smoke-test/tests/test_stateful_ingestion.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from datahub.ingestion.source.sql.mysql import MySQLConfig, MySQLSource
66
from datahub.ingestion.source.state.checkpoint import Checkpoint
77
from datahub.ingestion.source.state.entity_removal_state import GenericCheckpointState
8+
from datahub.ingestion.source.state.stale_entity_removal_handler import StaleEntityRemovalHandler
89
from sqlalchemy import create_engine
910
from sqlalchemy.sql import text
1011

@@ -52,7 +53,9 @@ def get_current_checkpoint_from_pipeline(
5253
# TODO: Refactor to use the helper method in the metadata-ingestion tests, instead of copying it here.
5354
mysql_source = cast(MySQLSource, pipeline.source)
5455
return mysql_source.state_provider.get_current_checkpoint(
55-
mysql_source.stale_entity_removal_handler.job_id
56+
StaleEntityRemovalHandler.compute_job_id(
57+
getattr(mysql_source, "platform", "default")
58+
)
5659
)
5760

5861
source_config_dict: Dict[str, Any] = {

0 commit comments

Comments
 (0)