def setup_session():
"""Cleanup any previous transaction and initialize a new database and
session to work with."""
if models.Session:
models.cleanup_session()
models.Session = None
models.init_session(":memory:")
try:
models.transfer_session.query(models.Unit).one()
except MultipleResultsFound:
assert False
except NoResultFound:
assert True
def setup_session():
"""Cleanup any previous transaction and initialize a new database and
session to work with."""
if models.Session:
models.cleanup_session()
models.Session = None
models.init_session(":memory:")
try:
models.transfer_session.query(models.Unit).one()
except MultipleResultsFound:
assert False
except NoResultFound:
assert True
def manage_automation_execution(pid_file):
"""Cleanup procedures for transfer.py."""
LOGGER.info("Running post-execution clean-up. Exiting script")
os.remove(pid_file)
models.cleanup_session()
def manage_automation_execution(pid_file):
"""Cleanup procedures for transfer.py."""
LOGGER.info("Running post-execution clean-up. Exiting script")
os.remove(pid_file)
models.cleanup_session()