Skip to content

Commit

Permalink
fix(ingest): Fix breaking smoke test on stateful ingestion (#8128)
Browse files Browse the repository at this point in the history
  • Loading branch information
asikowitz authored May 25, 2023
1 parent b216caa commit 1478d70
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion smoke-test/tests/test_stateful_ingestion.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
from datahub.ingestion.source.sql.mysql import MySQLConfig, MySQLSource
from datahub.ingestion.source.state.checkpoint import Checkpoint
from datahub.ingestion.source.state.entity_removal_state import GenericCheckpointState
from datahub.ingestion.source.state.stale_entity_removal_handler import StaleEntityRemovalHandler
from sqlalchemy import create_engine
from sqlalchemy.sql import text

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

source_config_dict: Dict[str, Any] = {
Expand Down

0 comments on commit 1478d70

Please sign in to comment.