Fix import (#13044)

This commit is contained in:
Pere Miquel Brull 2023-08-31 08:30:18 +02:00 committed by GitHub
parent 673512d7f6
commit a5667bb585
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 10 additions and 14 deletions

View File

@ -22,7 +22,7 @@ from metadata.generated.schema.entity.services.connections.metadata.openMetadata
from metadata.generated.schema.metadataIngestion.workflow import ( from metadata.generated.schema.metadataIngestion.workflow import (
Source as WorkflowSource, Source as WorkflowSource,
) )
from metadata.ingestion.api.source import InvalidSourceException from metadata.ingestion.api.steps import InvalidSourceException
from metadata.ingestion.source.database.query_parser_source import QueryParserSource from metadata.ingestion.source.database.query_parser_source import QueryParserSource

View File

@ -19,7 +19,7 @@ from unittest import TestCase
import pytest import pytest
from metadata.ingestion.api.sink import SinkStatus from metadata.ingestion.api.sink import SinkStatus
from metadata.ingestion.api.source import SourceStatus from metadata.ingestion.api.status import Status
from .e2e_types import E2EType from .e2e_types import E2EType
from .test_cli import CliBase from .test_cli import CliBase
@ -95,21 +95,17 @@ class CliDashboardBase(TestCase):
raise NotImplementedError() raise NotImplementedError()
@abstractmethod @abstractmethod
def assert_not_including( def assert_not_including(self, source_status: Status, sink_status: SinkStatus):
self, source_status: SourceStatus, sink_status: SinkStatus
):
raise NotImplementedError() raise NotImplementedError()
@abstractmethod @abstractmethod
def assert_for_vanilla_ingestion( def assert_for_vanilla_ingestion(
self, source_status: SourceStatus, sink_status: SinkStatus self, source_status: Status, sink_status: SinkStatus
) -> None: ) -> None:
raise NotImplementedError() raise NotImplementedError()
@abstractmethod @abstractmethod
def assert_filtered_mix( def assert_filtered_mix(self, source_status: Status, sink_status: SinkStatus):
self, source_status: SourceStatus, sink_status: SinkStatus
):
raise NotImplementedError() raise NotImplementedError()
@staticmethod @staticmethod

View File

@ -21,7 +21,7 @@ import pytest
from metadata.generated.schema.entity.data.table import Table from metadata.generated.schema.entity.data.table import Table
from metadata.generated.schema.tests.testDefinition import TestDefinition, TestPlatform from metadata.generated.schema.tests.testDefinition import TestDefinition, TestPlatform
from metadata.ingestion.api.sink import SinkStatus from metadata.ingestion.api.sink import SinkStatus
from metadata.ingestion.api.source import SourceStatus from metadata.ingestion.api.status import Status
from .test_cli import CliBase from .test_cli import CliBase
@ -107,12 +107,12 @@ class CliDBTBase(TestCase):
@abstractmethod @abstractmethod
def assert_for_vanilla_ingestion( def assert_for_vanilla_ingestion(
self, source_status: SourceStatus, sink_status: SinkStatus self, source_status: Status, sink_status: SinkStatus
) -> None: ) -> None:
raise NotImplementedError() raise NotImplementedError()
@abstractmethod @abstractmethod
def assert_for_dbt_ingestion( def assert_for_dbt_ingestion(
self, source_status: SourceStatus, sink_status: SinkStatus self, source_status: Status, sink_status: SinkStatus
) -> None: ) -> None:
raise NotImplementedError() raise NotImplementedError()

View File

@ -17,7 +17,7 @@ from typing import List
import pytest import pytest
from metadata.ingestion.api.sink import SinkStatus from metadata.ingestion.api.sink import SinkStatus
from metadata.ingestion.api.source import SourceStatus from metadata.ingestion.api.status import Status
from .base.e2e_types import E2EType from .base.e2e_types import E2EType
from .common.test_cli_db import CliCommonDB from .common.test_cli_db import CliCommonDB
@ -79,7 +79,7 @@ class SnowflakeCliTest(CliCommonDB.TestSuite, SQACommonMethods):
return "snowflake" return "snowflake"
def assert_for_vanilla_ingestion( def assert_for_vanilla_ingestion(
self, source_status: SourceStatus, sink_status: SinkStatus self, source_status: Status, sink_status: SinkStatus
) -> None: ) -> None:
self.assertTrue(len(source_status.failures) == 0) self.assertTrue(len(source_status.failures) == 0)
self.assertTrue(len(source_status.warnings) == 0) self.assertTrue(len(source_status.warnings) == 0)