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)"
            )