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