mirror of
https://github.com/open-metadata/OpenMetadata.git
synced 2025-07-19 15:31:59 +00:00
d59b83f9d1
* tests: refactor refactor tests and consolidate common functionality in integrations.conftest this enables writing tests more concisely. demonstrated with postgres and mssql. will migrate more * format * removed helpers * changed scope of fictures * changed scope of fixtures * added profiler test for mssql * fixed import in data_quality test * json safe serialization * format * set MARS_Connection * fix(data-quality): empty test suite do not raise for empty test suite * format * dont need to check length in _get_test_cases_from_test_suite * fix * added warning if no test cases are found
105 lines
3.1 KiB
Python
105 lines
3.1 KiB
Python
from unittest.mock import Mock
|
|
from uuid import UUID
|
|
|
|
import pytest
|
|
|
|
from metadata.data_quality.source.test_suite import TestSuiteSource
|
|
from metadata.generated.schema.entity.data.table import Table
|
|
from metadata.generated.schema.metadataIngestion.workflow import (
|
|
OpenMetadataWorkflowConfig,
|
|
)
|
|
from metadata.generated.schema.tests.testCase import TestCase
|
|
from metadata.generated.schema.tests.testSuite import TestSuite
|
|
from metadata.generated.schema.type.entityReference import EntityReference
|
|
from metadata.ingestion.ometa.ometa_api import OpenMetadata
|
|
|
|
MOCK_ENTITY_REFERENCE = EntityReference(
|
|
id=str(UUID(int=0)), type="test_suite", name="test_suite"
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"parameters,expected",
|
|
[
|
|
(
|
|
{
|
|
"type": "TestSuite",
|
|
"entityFullyQualifiedName": "MyTestSuite",
|
|
},
|
|
["test_case1", "test_case2"],
|
|
),
|
|
(
|
|
{
|
|
"type": "TestSuite",
|
|
"entityFullyQualifiedName": "MyTestSuite",
|
|
"testCases": [
|
|
"test_case1",
|
|
],
|
|
},
|
|
["test_case1"],
|
|
),
|
|
(
|
|
{
|
|
"type": "TestSuite",
|
|
"entityFullyQualifiedName": "MyTestSuite",
|
|
"testCases": [],
|
|
},
|
|
[],
|
|
),
|
|
],
|
|
)
|
|
def test_source_config(parameters, expected, monkeypatch):
|
|
workflow_config = {
|
|
"source": {
|
|
"type": "TestSuite",
|
|
"serviceName": "MyTestSuite",
|
|
"sourceConfig": {"config": parameters},
|
|
"serviceConnection": {
|
|
"config": {
|
|
"type": "Mysql",
|
|
"hostPort": "localhost:3306",
|
|
"username": "root",
|
|
}
|
|
},
|
|
},
|
|
"workflowConfig": {
|
|
"openMetadataServerConfig": {
|
|
"hostPort": "localhost:8585",
|
|
}
|
|
},
|
|
}
|
|
monkeypatch.setattr(TestSuiteSource, "test_connection", Mock())
|
|
|
|
mock_metadata = Mock(spec=OpenMetadata)
|
|
mock_metadata.get_by_name.return_value = Table(
|
|
id=UUID(int=0),
|
|
name="test_table",
|
|
columns=[],
|
|
testSuite=MOCK_ENTITY_REFERENCE,
|
|
)
|
|
mock_metadata.list_all_entities.return_value = [
|
|
TestCase(
|
|
name="test_case1",
|
|
id=UUID(int=0),
|
|
testDefinition=MOCK_ENTITY_REFERENCE,
|
|
testSuite=MOCK_ENTITY_REFERENCE,
|
|
entityLink="<#E::some::link>",
|
|
),
|
|
TestCase(
|
|
name="test_case2",
|
|
id=UUID(int=0),
|
|
testDefinition=MOCK_ENTITY_REFERENCE,
|
|
testSuite=MOCK_ENTITY_REFERENCE,
|
|
entityLink="<#E::some::link>",
|
|
),
|
|
]
|
|
mock_metadata.get_by_id.return_value = TestSuite(
|
|
name="test_suite", executable=True, id=UUID(int=0)
|
|
)
|
|
|
|
source = TestSuiteSource(
|
|
OpenMetadataWorkflowConfig.parse_obj(workflow_config), mock_metadata
|
|
)
|
|
test_cases = list(source._iter())[0].right.test_cases
|
|
assert [t.name.root for t in test_cases] == expected
|