diff --git a/metadata-ingestion/src/datahub/ingestion/source/slack/slack.py b/metadata-ingestion/src/datahub/ingestion/source/slack/slack.py index 21d82c12d98795..1d6fe9342b8060 100644 --- a/metadata-ingestion/src/datahub/ingestion/source/slack/slack.py +++ b/metadata-ingestion/src/datahub/ingestion/source/slack/slack.py @@ -16,11 +16,7 @@ platform_name, support_status, ) -from datahub.ingestion.api.source import ( - SourceReport, - TestableSource, - TestConnectionReport, -) +from datahub.ingestion.api.source import Source, SourceReport from datahub.ingestion.api.workunit import MetadataWorkUnit from datahub.metadata.schema_classes import ( CorpUserEditableInfoClass, @@ -89,10 +85,10 @@ class SlackSourceReport(SourceReport): PLATFORM_NAME = "slack" -@platform_name(PLATFORM_NAME) +@platform_name("Slack") @config_class(SlackSourceConfig) @support_status(SupportStatus.TESTING) -class SlackSource(TestableSource): +class SlackSource(Source): def __init__(self, ctx: PipelineContext, config: SlackSourceConfig): self.ctx = ctx self.config = config @@ -107,10 +103,6 @@ def create(cls, config_dict, ctx): config = SlackSourceConfig.parse_obj(config_dict) return cls(ctx, config) - @staticmethod - def test_connection(config_dict: dict) -> TestConnectionReport: - raise NotImplementedError("This class does not implement this method") - def get_slack_client(self) -> WebClient: return WebClient(token=self.config.bot_token.get_secret_value())