mirror of
https://github.com/open-metadata/OpenMetadata.git
synced 2025-12-13 08:37:03 +00:00
Fix import (#13044)
This commit is contained in:
parent
673512d7f6
commit
a5667bb585
@ -22,7 +22,7 @@ from metadata.generated.schema.entity.services.connections.metadata.openMetadata
|
||||
from metadata.generated.schema.metadataIngestion.workflow import (
|
||||
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
|
||||
|
||||
|
||||
|
||||
@ -19,7 +19,7 @@ from unittest import TestCase
|
||||
import pytest
|
||||
|
||||
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 .test_cli import CliBase
|
||||
@ -95,21 +95,17 @@ class CliDashboardBase(TestCase):
|
||||
raise NotImplementedError()
|
||||
|
||||
@abstractmethod
|
||||
def assert_not_including(
|
||||
self, source_status: SourceStatus, sink_status: SinkStatus
|
||||
):
|
||||
def assert_not_including(self, source_status: Status, sink_status: SinkStatus):
|
||||
raise NotImplementedError()
|
||||
|
||||
@abstractmethod
|
||||
def assert_for_vanilla_ingestion(
|
||||
self, source_status: SourceStatus, sink_status: SinkStatus
|
||||
self, source_status: Status, sink_status: SinkStatus
|
||||
) -> None:
|
||||
raise NotImplementedError()
|
||||
|
||||
@abstractmethod
|
||||
def assert_filtered_mix(
|
||||
self, source_status: SourceStatus, sink_status: SinkStatus
|
||||
):
|
||||
def assert_filtered_mix(self, source_status: Status, sink_status: SinkStatus):
|
||||
raise NotImplementedError()
|
||||
|
||||
@staticmethod
|
||||
|
||||
@ -21,7 +21,7 @@ import pytest
|
||||
from metadata.generated.schema.entity.data.table import Table
|
||||
from metadata.generated.schema.tests.testDefinition import TestDefinition, TestPlatform
|
||||
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
|
||||
|
||||
@ -107,12 +107,12 @@ class CliDBTBase(TestCase):
|
||||
|
||||
@abstractmethod
|
||||
def assert_for_vanilla_ingestion(
|
||||
self, source_status: SourceStatus, sink_status: SinkStatus
|
||||
self, source_status: Status, sink_status: SinkStatus
|
||||
) -> None:
|
||||
raise NotImplementedError()
|
||||
|
||||
@abstractmethod
|
||||
def assert_for_dbt_ingestion(
|
||||
self, source_status: SourceStatus, sink_status: SinkStatus
|
||||
self, source_status: Status, sink_status: SinkStatus
|
||||
) -> None:
|
||||
raise NotImplementedError()
|
||||
|
||||
@ -17,7 +17,7 @@ from typing import List
|
||||
import pytest
|
||||
|
||||
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 .common.test_cli_db import CliCommonDB
|
||||
@ -79,7 +79,7 @@ class SnowflakeCliTest(CliCommonDB.TestSuite, SQACommonMethods):
|
||||
return "snowflake"
|
||||
|
||||
def assert_for_vanilla_ingestion(
|
||||
self, source_status: SourceStatus, sink_status: SinkStatus
|
||||
self, source_status: Status, sink_status: SinkStatus
|
||||
) -> None:
|
||||
self.assertTrue(len(source_status.failures) == 0)
|
||||
self.assertTrue(len(source_status.warnings) == 0)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user