mirror of
https://github.com/open-metadata/OpenMetadata.git
synced 2025-08-31 20:51:26 +00:00
ca812852d6
* Make pytest to user code from src rather than from install package * Fix test_amundsen: missing None * Update pytest configuration to use importlib mode * Fix custom_basemodel_validation to check model_fields on type(values) to prevent noisy warnings * Refactor referencedByQueries validation to use field_validator as per deprecation warning * Update ColumnJson to use model_rebuild rather as replacement for forward reference updates as per deprecation warning * Move superset test to integration test as they are using testcontainers * Update coverage source path * Fix wrong import. * Add install_dev_env target to Makefile for development dependencies * Add test-unit as extra in setup.py * Modify dependencies in dev environment. * Ignore all airflow tests * Remove coverage in unit_ingestion_dev_env. Revert coverage source to prevent broken CI. * Add nox for running unit test * FIx PowerBI integration test to use pathlib for resource paths and not os.getcwd to prevent failures when not executed from the right path * Move test_helpers.py to unit test, as it is not an integration test. * Remove utils empty folder in integration tests * Refactor testcontainers configuration to avoid pitfalls with max_tries setting * Add nox unit testing basic setup * Add format check session * Refactor nox-unit and add plugins tests * Add GHA for py-nox-ci * Add comment to GHA * Restore conftest.py file * Clarify comment * Simplify function * Fix matrix startegy and nox mismatch * Improve python version strategy with nox and GHA --------- Co-authored-by: Pere Menal <pere.menal@getcollate.io>
98 lines
2.8 KiB
Python
98 lines
2.8 KiB
Python
# Copyright 2022 Collate
|
|
# Licensed under the Collate Community License, Version 1.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
# https://github.com/open-metadata/OpenMetadata/blob/main/ingestion/LICENSE
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
"""
|
|
Test suite for the action module implementation
|
|
"""
|
|
|
|
import os
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
from jinja2 import Environment
|
|
from pytest import mark
|
|
|
|
from metadata.great_expectations.utils.ometa_config_handler import render_template
|
|
|
|
_GX_0_18 = "0.18"
|
|
|
|
try:
|
|
import great_expectations as gx
|
|
|
|
from metadata.great_expectations.action import OpenMetadataValidationAction
|
|
|
|
_gx_version_ok = gx.__version__.startswith(_GX_0_18)
|
|
except ImportError:
|
|
_gx_version_ok = False
|
|
|
|
skip_gx = pytest.mark.skipif(
|
|
not _gx_version_ok,
|
|
reason=(
|
|
"Great Expectations not installed or version mismatch "
|
|
f"(required: {_GX_0_18})"
|
|
),
|
|
)
|
|
|
|
|
|
@skip_gx
|
|
@mark.parametrize(
|
|
"input,expected",
|
|
[
|
|
(None, "list_entities"),
|
|
("service_name", "get_by_name"),
|
|
],
|
|
)
|
|
def test_get_table_entity(input, expected, mocked_ometa, mocked_ge_data_context):
|
|
"""Test get table entity"""
|
|
|
|
ometa_validation = OpenMetadataValidationAction(
|
|
data_context=mocked_ge_data_context,
|
|
config_file_path="my/config/path",
|
|
database_service_name=input,
|
|
)
|
|
|
|
res = ometa_validation._get_table_entity("database", "schema", "table")
|
|
assert res._type == expected
|
|
|
|
|
|
@skip_gx
|
|
@mark.parametrize(
|
|
"input,expected",
|
|
[
|
|
(None, "list_entities"),
|
|
("service_name", "get_by_name"),
|
|
],
|
|
)
|
|
def test_get_table_entity_database_service_name(
|
|
input, expected, mocked_ometa, mocked_ge_data_context
|
|
):
|
|
"""Test get table entity"""
|
|
|
|
ometa_validation = OpenMetadataValidationAction(
|
|
data_context=mocked_ge_data_context,
|
|
config_file_path="my/config/path",
|
|
database_service_name=input,
|
|
)
|
|
|
|
res = ometa_validation._get_table_entity("database", "schema", "table")
|
|
assert res._type == expected
|
|
|
|
|
|
def test_create_jinja_environment(fixture_jinja_environment):
|
|
"""Test create jinja environment"""
|
|
assert isinstance(fixture_jinja_environment, Environment)
|
|
|
|
|
|
@mock.patch.dict(os.environ, {"API_VERSION": "v1"})
|
|
def test_render_template(fixture_jinja_environment):
|
|
"""Test create jinja environment"""
|
|
tmplt = render_template(fixture_jinja_environment)
|
|
assert tmplt == "hostPort: http://localhost:8585\napiVersion: v1"
|