mirror of
https://github.com/datahub-project/datahub.git
synced 2025-10-31 02:37:05 +00:00
55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
import platform
|
|
|
|
import pytest
|
|
|
|
from datahub.ingestion.api.common import PipelineContext
|
|
from datahub.ingestion.source.sql.hana import HanaConfig, HanaSource
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
platform.machine().lower() == "aarch64",
|
|
reason="The hdbcli dependency is not available for aarch64",
|
|
)
|
|
def test_platform_correctly_set_hana():
|
|
source = HanaSource(
|
|
ctx=PipelineContext(run_id="hana-source-test"),
|
|
config=HanaConfig(),
|
|
)
|
|
assert source.platform == "hana"
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
platform.machine().lower() == "aarch64",
|
|
reason="The hdbcli dependency is not available for aarch64",
|
|
)
|
|
def test_hana_uri_native():
|
|
config = HanaConfig.parse_obj(
|
|
{
|
|
"username": "user",
|
|
"password": "password",
|
|
"host_port": "host:39041",
|
|
"scheme": "hana+hdbcli",
|
|
}
|
|
)
|
|
assert config.get_sql_alchemy_url() == "hana+hdbcli://user:password@host:39041"
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
platform.machine().lower() == "aarch64",
|
|
reason="The hdbcli dependency is not available for aarch64",
|
|
)
|
|
def test_hana_uri_native_db():
|
|
config = HanaConfig.parse_obj(
|
|
{
|
|
"username": "user",
|
|
"password": "password",
|
|
"host_port": "host:39041",
|
|
"scheme": "hana+hdbcli",
|
|
"database": "database",
|
|
}
|
|
)
|
|
assert (
|
|
config.get_sql_alchemy_url()
|
|
== "hana+hdbcli://user:password@host:39041/database"
|
|
)
|