def test_SQLCredentialYamlHelper_driver(empty_data_context):
helper = SQLCredentialYamlHelper(usage_stats_payload={"foo": "bar"},
driver="stuff")
expected_credentials_snippet = '''\
host = "YOUR_HOST"
port = "YOUR_PORT"
username = "******"
password = "******"
database = "YOUR_DATABASE"'''
assert helper.credentials_snippet() == expected_credentials_snippet
assert (helper.yaml_snippet() == '''f"""
name: {datasource_name}
class_name: SimpleSqlalchemyDatasource
introspection:
whole_table:
data_asset_name_suffix: __whole_table
credentials:
host: {host}
port: '{port}'
username: {username}
password: {password}
database: {database}
drivername: stuff"""''')
renderer = helper.get_notebook_renderer(empty_data_context)
assert renderer.sql_credentials_code_snippet == expected_credentials_snippet
def test_SQLCredentialYamlHelper_defaults(empty_data_context):
helper = SQLCredentialYamlHelper(usage_stats_payload={"foo": "bar"})
expected_credentials_snippet = '''\
host = "YOUR_HOST"
port = "YOUR_PORT"
username = "******"
password = "******"
database = "YOUR_DATABASE"'''
assert helper.credentials_snippet() == expected_credentials_snippet
assert (helper.yaml_snippet() == '''f"""
name: {datasource_name}
class_name: Datasource
execution_engine:
class_name: SqlAlchemyExecutionEngine
credentials:
host: {host}
port: '{port}'
username: {username}
password: {password}
database: {database}
data_connectors:
default_runtime_data_connector_name:
class_name: RuntimeDataConnector
batch_identifiers:
- default_identifier_name
default_inferred_data_connector_name:
class_name: InferredAssetSqlDataConnector
name: whole_table"""''')
renderer = helper.get_notebook_renderer(empty_data_context)
assert renderer.sql_credentials_code_snippet == expected_credentials_snippet