diff --git a/awx/main/tests/functional/test_licenses.py b/awx/main/tests/functional/test_licenses.py index 4b8ee7a918..9961c7ca96 100644 --- a/awx/main/tests/functional/test_licenses.py +++ b/awx/main/tests/functional/test_licenses.py @@ -26,12 +26,12 @@ def test_python_and_js_licenses(): return (is_gpl, is_lgpl) def find_embedded_source_version(path, name): - for entry in os.listdir(path): - # Check variations of '-' and '_' in filenames due to python - for fname in [name, name.replace('-', '_')]: - if entry.startswith(fname) and entry.endswith('.tar.gz'): - v = entry.split(name + '-')[1].split('.tar.gz')[0] - return v + files = os.listdir(path) + tgz_files = [f for f in files if f.endswith('.tar.gz')] + for tgz in tgz_files: + pkg_name = tgz.split('-')[0].split('_')[0] + if pkg_name == name: + return tgz.split('-')[1].split('.tar.gz')[0] return None list = {}