def observer():
observer = TestStateObserver()
runner.register_state_observer(observer)
yield observer
runner.deregister_state_observer(observer)
remove_test_config()
remove_test_db()
def run_app(command, shell=False, state_queue=None):
"""
Run command
:param command: command to run
:param shell: use shell for executing command
:param state_queue: queue for putting new execution states of running jobs
:return: output of the executed command
"""
program.USE_SHELL = shell
# Prevent UnsupportedOperation error: https://github.com/prompt-toolkit/python-prompt-toolkit/issues/1107
prompt_toolkit.output.defaults.create_output = NoFormattingOutput
observer = None
if state_queue:
observer = PutStateToQueueObserver(state_queue)
runner.register_state_observer(observer)
try:
main(command.split())
finally:
prompt_toolkit.output.defaults.create_output = None
program.USE_SHELL = False
if observer:
runner.deregister_state_observer(observer)
def observer():
observer = TestStateObserver()
runner.register_state_observer(observer)
yield observer
runner.deregister_state_observer(observer)