mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-20 07:58:31 +00:00
31 lines
1.3 KiB
Python
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)"
|
|
)
|