def test_set_default_engine_kwargs_for_mysql_with_other_rdb() -> None:
# Do not change engine_kwargs if database is not MySQL.
engine_kwargs: Dict[str, Any] = {}
RDBStorage._set_default_engine_kwargs_for_mysql("sqlite:///example.db", engine_kwargs)
assert "pool_pre_ping" not in engine_kwargs
RDBStorage._set_default_engine_kwargs_for_mysql("postgres:///example.db", engine_kwargs)
assert "pool_pre_ping" not in engine_kwargs
def test_set_default_engine_kwargs_for_mysql_with_other_rdb():
# type: ()-> None
# Do not change engine_kwargs if database is not MySQL.
engine_kwargs = {} # type: Dict[str, Any]
RDBStorage._set_default_engine_kwargs_for_mysql('sqlite:///example.db', engine_kwargs)
assert 'pool_pre_ping' not in engine_kwargs
RDBStorage._set_default_engine_kwargs_for_mysql('postgres:///example.db', engine_kwargs)
assert 'pool_pre_ping' not in engine_kwargs
def test_set_default_engine_kwargs_for_mysql(url: str,
engine_kwargs: Dict[str, Any],
expected: bool) -> None:
RDBStorage._set_default_engine_kwargs_for_mysql(url, engine_kwargs)
assert engine_kwargs["pool_pre_ping"] is expected
def test_set_default_engine_kwargs_for_mysql(url, engine_kwargs, expected):
# type: (str, Dict[str, Any], bool)-> None
RDBStorage._set_default_engine_kwargs_for_mysql(url, engine_kwargs)
assert engine_kwargs['pool_pre_ping'] is expected
