import sys

import pytest

from metadata.generated.schema.entity.data.table import Constraint, Table
from metadata.workflow.metadata import MetadataWorkflow

if not sys.version_info >= (3, 9):
    pytest.skip("requires python 3.9+", allow_module_level=True)


def test_ingest_metadata(
    patch_passwords_for_db_services,
    run_workflow,
    ingestion_config,
    db_service,
    metadata,
):
    run_workflow(MetadataWorkflow, ingestion_config)
    table: Table = metadata.get_by_name(
        Table,
        f"{db_service.fullyQualifiedName.root}.AdventureWorks.HumanResources.Department",
    )
    assert table is not None
    assert table.columns[0].name.root == "DepartmentID"
    assert table.columns[0].constraint == Constraint.PRIMARY_KEY
    assert table.columns[1].name.root == "Name"
    assert table.columns[2].name.root == "GroupName"
    assert table.columns[3].name.root == "ModifiedDate"