datahub/metadata-ingestion/tests/unit/test_data_process_instance_urn.py
2022-04-12 23:19:39 -07:00

31 lines
1.3 KiB
Python

import unittest
from datahub.utilities.urns.data_process_instance_urn import DataProcessInstanceUrn
from datahub.utilities.urns.error import InvalidUrnError
class TestDomainUrn(unittest.TestCase):
def test_parse_urn(self) -> None:
dataprocessinstance_urn_str = "urn:li:dataProcessInstance:abc"
dataprocessinstance_urn = DataProcessInstanceUrn.create_from_string(
dataprocessinstance_urn_str
)
assert dataprocessinstance_urn.get_type() == DataProcessInstanceUrn.ENTITY_TYPE
assert dataprocessinstance_urn.get_entity_id() == ["abc"]
assert str(dataprocessinstance_urn) == dataprocessinstance_urn_str
assert dataprocessinstance_urn == DataProcessInstanceUrn(
"dataProcessInstance", ["abc"]
)
assert dataprocessinstance_urn == DataProcessInstanceUrn.create_from_id("abc")
assert "abc" == dataprocessinstance_urn.get_dataprocessinstance_id()
def test_invalid_urn(self) -> None:
with self.assertRaises(InvalidUrnError):
DataProcessInstanceUrn.create_from_string("urn:li:abc:dataProcessInstance")
with self.assertRaises(InvalidUrnError):
DataProcessInstanceUrn.create_from_string(
"urn:li:dataProcessInstance:(part1,part2)"
)