diff --git a/tests/test_mediaworker.py b/tests/test_mediaworker.py index 6eea127a69fe6579cbd95461a910b8c80d759cc0..86d78daacea9ea10c44494e8c595c0af781d13ff 100755 --- a/tests/test_mediaworker.py +++ b/tests/test_mediaworker.py @@ -32,13 +32,15 @@ def check_ssh(ip): return False return True -def check_apt(ip): - print('Checking apt-get update on MediaWorker') - cmd = 'ssh %s apt-get update' % ip - try: +def run_tests(ip): + print('Running envsetup tests on MediaWorker') + cmd = 'ssh -t %s /root/envsetup/tester.py' % ip + status = os.system(cmd) + if status == 0: subprocess.check_output(cmd, shell=True, timeout=10) - print('%sapt-get update success on MediaWorker%s' % (GREEN, DEF)) - except subprocess.CalledProcessError: + print('%sAll tests completed on MediaWorker%s' % (GREEN, DEF)) + return True + else: print('%apt-get update failed on MediaWorker%s' % (RED, DEF)) return False @@ -48,6 +50,7 @@ worker_ip = conf.get('CELERITY_WORKER_IP') if worker_ip != '127.0.1.1': if not check_ssh(worker_ip): all_ok = False - if not check_apt(worker_ip): - all_ok = False + else: + if not run_tests(worker_ip): + all_ok = False sys.exit(int(not all_ok))