def idadb_tmp(request):
"""
DataBase connection fixture, to be used by destructive and
semi-destructive functions.
"""
def fin():
try:
idadb_tmp.close()
except:
pass
request.addfinalizer(fin)
jdbc = request.config.getoption('--jdbc')
if jdbc != '':
try:
idadb_tmp = ibmdbpy.IdaDataBase(dsn=jdbc, autocommit=False)
except:
raise
else:
try:
idadb_tmp = ibmdbpy.IdaDataBase(
dsn=request.config.getoption('--dsn'),
uid=request.config.getoption('--uid'),
pwd=request.config.getoption('--pwd'),
autocommit=False)
except:
raise
return idadb_tmp
def idadb(request):
"""
DataBase connection fixture, to be used for the whole testing session.
Hold the main IdaDataBase object. Shall not be closed except by a
pytest finalizer.
"""
def fin():
try:
idadb.close()
except:
pass
request.addfinalizer(fin)
jdbc = request.config.getoption('--jdbc')
if jdbc != '':
try:
idadb = ibmdbpy.IdaDataBase(dsn=jdbc, autocommit=False)
except:
raise
else:
try:
idadb = ibmdbpy.IdaDataBase(dsn=request.config.getoption('--dsn'),
uid=request.config.getoption('--uid'),
pwd=request.config.getoption('--pwd'),
autocommit=False)
except:
raise
return idadb
