def dispatch(args):
""" Main entry function. """
if helper.is_terminated():
return
helper.g_log("events.run", 1)
try:
config.read_config()
except Exception:
logger.exception("Unable to read configuration. Skipping processing.")
monitor.send_event(
monitor.h_events.CONFIG_UPDATE,
monitor.severity.WARNING,
"Unable to read configuration (possibly locked)",
)
return
success_folder = Path(config.mercure[mercure_folders.SUCCESS])
error_folder = Path(config.mercure[mercure_folders.ERROR])
retry_max = config.mercure["retry_max"]
retry_delay = config.mercure["retry_delay"]
# TODO: Sort list so that the oldest DICOMs get dispatched first
with os.scandir(config.mercure[mercure_folders.OUTGOING]) as it:
for entry in it:
if entry.is_dir() and not has_been_send(entry.path) and is_ready_for_sending(entry.path):
logger.info(f"Sending folder {entry.path}")
execute(Path(entry.path), success_folder, error_folder, retry_max, retry_delay)
# If termination is requested, stop processing series after the
# active one has been completed
if helper.is_terminated():
break
def test_execute_error_case_delay_is_not_over(fs, mocker):
""" This case simulates a dcmsend error. Should not run anything because
the next_retry_at time has not been passed.
"""
source = "/var/data/source/a"
success = "/var/data/success/"
error = "/var/data/error"
fs.create_dir(source)
fs.create_dir(success)
fs.create_dir(error)
fs.create_file("/var/data/source/a/one.dcm")
target = {
"dispatch": {
"target_ip": "0.0.0.0",
"target_aet_target": "a",
"target_port": 90,
"retries": 5,
"next_retry_at": time.time() + 500
}
}
fs.create_file("/var/data/source/a/"+mercure_names.TASKFILE, contents=json.dumps(target))
mock = mocker.patch(
"dispatch.send.run", side_effect=CalledProcessError("Mock", cmd="None")
)
execute(Path(source), Path(success), Path(error), 5, 1)
assert not mock.called
def test_execute_error_case(fs, mocker):
""" This case simulates a dcmsend error. After that the retry counter
gets increased but the data stays in the folder. """
source = "/var/data/source/a"
success = "/var/data/success/"
error = "/var/data/error"
fs.create_dir(source)
fs.create_dir(success)
fs.create_dir(error)
fs.create_file("/var/data/source/a/one.dcm")
target = { "dispatch": {"target_ip": "0.0.0.0", "target_aet_target": "a", "target_port": 90 } }
fs.create_file("/var/data/source/a/"+mercure_names.TASKFILE, contents=json.dumps(target))
mocker.patch(
"dispatch.send.run", side_effect=CalledProcessError("Mock", cmd="None")
)
execute(Path(source), Path(success), Path(error), 10, 1)
with open("/var/data/source/a/"+mercure_names.TASKFILE, "r") as f:
modified_target = json.load(f)
assert Path(source).exists()
assert (Path(source) / mercure_names.TASKFILE).exists()
assert (Path(source) / "one.dcm").exists()
assert modified_target["dispatch"]["retries"] == 1
assert is_ready_for_sending(source)
def test_execute_error_case_max_retries_reached(fs, mocker):
""" This case simulates a dcmsend error. Max number of retries is reached
and the data is moved to the error folder.
"""
source = "/var/data/source/a"
success = "/var/data/success/"
error = "/var/data/error"
fs.create_dir(source)
fs.create_dir(success)
fs.create_dir(error)
fs.create_file("/var/data/source/a/one.dcm")
target = {
"dispatch": {
"target_ip": "0.0.0.0",
"target_aet_target": "a",
"target_port": 90,
"retries": 5
}
}
fs.create_file("/var/data/source/a/"+mercure_names.TASKFILE, contents=json.dumps(target))
mocker.patch(
"dispatch.send.run", side_effect=CalledProcessError("Mock", cmd="None")
)
execute(Path(source), Path(success), Path(error), 5, 1)
with open("/var/data/error/a/"+mercure_names.TASKFILE, "r") as f:
modified_target = json.load(f)
assert (Path(error) / "a" / mercure_names.TASKFILE).exists()
assert (Path(error) / "a" / "one.dcm").exists()
assert modified_target["dispatch"]["retries"] == 5
def test_execute_successful_case(fs, mocker):
source = "/var/data/source/a"
success = "/var/data/success/"
error = "/var/data/error"
fs.create_dir(source)
fs.create_dir(success)
fs.create_file("/var/data/source/a/one.dcm")
target = { "dispatch": {"target_ip": "0.0.0.0", "target_aet_target": "a", "target_port": 90 } }
fs.create_file("/var/data/source/a/"+mercure_names.TASKFILE, contents=json.dumps(target))
mocker.patch("dispatch.send.run", return_value=0)
execute(Path(source), Path(success), error, 1, 1)
assert not Path(source).exists()
assert (Path(success) / "a").exists()
assert (Path(success) / "a" / mercure_names.TASKFILE).exists()
assert (Path(success) / "a" / "one.dcm").exists()