diff --git a/tests/test_mediaworker.py b/tests/test_mediaworker.py index 320d1ef97c14f6eb7012eddf8f996db20e8155e3..72ebde0c0c3c9ef6e09257531f12c51036bd29a8 100755 --- a/tests/test_mediaworker.py +++ b/tests/test_mediaworker.py @@ -74,6 +74,18 @@ def check_celerity_connectivity(ip): print('%sFailed to reach tasks server from MediaWorker "%s".%s' % (RED, ip, DEF)) return False +def check_mediaworker_in_whitelist(ip): + nginx_vhosts_path = '/etc/nginx/sites-enabled' + vhosts = os.listdir(nginx_vhosts_path) + for v in vhosts: + if v.endswith('.conf'): + with f as open(os.path.join(nginx_vhosts_path, v), 'r'): + d = f.read() + if 'msuser_whitelist' in d: + if not ip in d: + print('MediaWorker ip is not in %s whitelist' % v) + return False + return True all_ok = True tested = False @@ -90,6 +102,8 @@ for worker_ip in worker_ips.split(','): all_ok = False #if not run_tests(worker_ip): # all_ok = False + if not check_mediaworker_in_whitelist(worker_ip): + all_ok = False if not tested: print('Celerity IP not set or running locally, skipping test.') sys.exit(2)