datahub/metadata-ingestion/tests/unit/urns/test_data_flow_urn.py

18 lines
688 B
Python

import unittest
import pytest
from datahub.utilities.urns.data_flow_urn import DataFlowUrn
@pytest.mark.filterwarnings("ignore::DeprecationWarning")
class TestDataFlowUrn(unittest.TestCase):
def test_parse_urn(self) -> None:
data_flow_urn_str = "urn:li:dataFlow:(airflow,def,prod)"
data_flow_urn = DataFlowUrn.create_from_string(data_flow_urn_str)
assert data_flow_urn.get_orchestrator_name() == "airflow"
assert data_flow_urn.get_flow_id() == "def"
assert data_flow_urn.get_env() == "prod"
assert data_flow_urn.__str__() == "urn:li:dataFlow:(airflow,def,prod)"
assert data_flow_urn == DataFlowUrn("airflow", "def", "prod")