mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-12 03:33:40 +00:00
34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
|
from datahub.utilities.sqlglot_lineage import SchemaResolver, _TableName
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_urn_for_table_lowercase():
|
||
|
|
schema_resolver = SchemaResolver(
|
||
|
|
platform="mssql",
|
||
|
|
platform_instance="Uppercased-Instance",
|
||
|
|
env="PROD",
|
||
|
|
graph=None,
|
||
|
|
)
|
||
|
|
|
||
|
|
table = _TableName(database="Database", db_schema="DataSet", table="Table")
|
||
|
|
|
||
|
|
assert (
|
||
|
|
schema_resolver.get_urn_for_table(table=table, lower=True)
|
||
|
|
== "urn:li:dataset:(urn:li:dataPlatform:mssql,uppercased-instance.database.dataset.table,PROD)"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_urn_for_table_not_lower_should_keep_capital_letters():
|
||
|
|
schema_resolver = SchemaResolver(
|
||
|
|
platform="mssql",
|
||
|
|
platform_instance="Uppercased-Instance",
|
||
|
|
env="PROD",
|
||
|
|
graph=None,
|
||
|
|
)
|
||
|
|
|
||
|
|
table = _TableName(database="Database", db_schema="DataSet", table="Table")
|
||
|
|
|
||
|
|
assert (
|
||
|
|
schema_resolver.get_urn_for_table(table=table, lower=False)
|
||
|
|
== "urn:li:dataset:(urn:li:dataPlatform:mssql,Uppercased-Instance.Database.DataSet.Table,PROD)"
|
||
|
|
)
|