Add ee cleanup tests

* Adds cleanup tests to the live test.
This commit is contained in:
Chris Meyers
2025-01-21 14:38:36 -05:00
committed by Chris Meyers
parent 534c312328
commit ad706d67c2
5 changed files with 127 additions and 43 deletions

View File

@@ -228,22 +228,24 @@ class RemoteJobError(RuntimeError):
pass
def run_until_complete(node, timing_data=None, **kwargs):
def run_until_complete(node, timing_data=None, worktype='ansible-runner', ttl='20s', **kwargs):
"""
Runs an ansible-runner work_type on remote node, waits until it completes, then returns stdout.
"""
config_data = read_receptor_config()
receptor_ctl = get_receptor_ctl(config_data)
use_stream_tls = getattr(get_conn_type(node, receptor_ctl), 'name', None) == "STREAMTLS"
kwargs.setdefault('tlsclient', get_tls_client(config_data, use_stream_tls))
kwargs.setdefault('ttl', '20s')
if ttl is not None:
kwargs['ttl'] = ttl
kwargs.setdefault('payload', '')
if work_signing_enabled(config_data):
kwargs['signwork'] = True
transmit_start = time.time()
result = receptor_ctl.submit_work(worktype='ansible-runner', node=node, **kwargs)
result = receptor_ctl.submit_work(worktype=worktype, node=node, **kwargs)
unit_id = result['unitid']
run_start = time.time()
@@ -371,7 +373,7 @@ def _convert_args_to_cli(vargs):
return args
def worker_cleanup(node_name, vargs, timeout=300.0):
def worker_cleanup(node_name, vargs):
args = _convert_args_to_cli(vargs)
remote_command = ' '.join(args)