mirror of
https://github.com/datahub-project/datahub.git
synced 2025-11-04 04:39:10 +00:00
20 lines
600 B
Python
20 lines
600 B
Python
from datahub.ingestion.api.common import PipelineContext
|
|
from datahub.ingestion.source.sql.mariadb import MariaDBSource
|
|
from datahub.ingestion.source.sql.mysql import MySQLConfig, MySQLSource
|
|
|
|
|
|
def test_platform_correctly_set_mariadb():
|
|
source = MariaDBSource(
|
|
ctx=PipelineContext(run_id="mariadb-source-test"),
|
|
config=MySQLConfig(),
|
|
)
|
|
assert source.platform == "mariadb"
|
|
|
|
|
|
def test_platform_correctly_set_mysql():
|
|
source = MySQLSource(
|
|
ctx=PipelineContext(run_id="mysql-source-test"),
|
|
config=MySQLConfig(),
|
|
)
|
|
assert source.platform == "mysql"
|