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 (
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

View File

@ -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

View File

@ -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()

View File

@ -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)