mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-21 08:32:05 +00:00
36 lines
1.4 KiB
Python
36 lines
1.4 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
from datahub.utilities.urns.data_flow_urn import DataFlowUrn
|
||
|
|
from datahub.utilities.urns.data_job_urn import DataJobUrn
|
||
|
|
from datahub.utilities.urns.error import InvalidUrnError
|
||
|
|
|
||
|
|
|
||
|
|
class TestDataJobUrn(unittest.TestCase):
|
||
|
|
def test_parse_urn(self) -> None:
|
||
|
|
data_job_urn_str = (
|
||
|
|
"urn:li:dataJob:(urn:li:dataFlow:(airflow,flow_id,prod),job_id)"
|
||
|
|
)
|
||
|
|
data_job_urn = DataJobUrn.create_from_string(data_job_urn_str)
|
||
|
|
assert data_job_urn.get_data_flow_urn() == DataFlowUrn.create_from_string(
|
||
|
|
"urn:li:dataFlow:(airflow,flow_id,prod)"
|
||
|
|
)
|
||
|
|
assert data_job_urn.get_job_id() == "job_id"
|
||
|
|
assert data_job_urn.__str__() == data_job_urn_str
|
||
|
|
assert data_job_urn == DataJobUrn(
|
||
|
|
"dataJob", ["urn:li:dataFlow:(airflow,flow_id,prod)", "job_id"]
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_invalid_urn(self) -> None:
|
||
|
|
with self.assertRaises(InvalidUrnError):
|
||
|
|
DataJobUrn.create_from_string(
|
||
|
|
"urn:li:abc:(urn:li:dataFlow:(airflow,flow_id,prod),job_id)"
|
||
|
|
)
|
||
|
|
|
||
|
|
with self.assertRaises(InvalidUrnError):
|
||
|
|
DataJobUrn.create_from_string("urn:li:dataJob:(urn:li:user:abc,job_id)")
|
||
|
|
|
||
|
|
with self.assertRaises(InvalidUrnError):
|
||
|
|
DataJobUrn.create_from_string(
|
||
|
|
"urn:li:dataJob:(urn:li:dataFlow:(airflow,flow_id,prod))"
|
||
|
|
)
|