From 7fc83394183b87b666865d8d49ac1330c540ab4b Mon Sep 17 00:00:00 2001
From: Laurent Indermuehle <laurent.indermuehle@pm.me>
Date: Mon, 9 Jan 2023 18:58:28 +0100
Subject: [PATCH] Add python script to recreate the test matrix from github
 workflow file

---
 run_all_tests.py | 79 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)
 create mode 100755 run_all_tests.py

diff --git a/run_all_tests.py b/run_all_tests.py
new file mode 100755
index 0000000..abc5fef
--- /dev/null
+++ b/run_all_tests.py
@@ -0,0 +1,79 @@
+#!/usr/bin/env python
+
+import yaml
+
+github_workflow_file = '.github/workflows/ansible-test-plugins.yml'
+
+
+def read_github_workflow_file():
+    with open(github_workflow_file, 'r') as gh_file:
+        try:
+            return yaml.safe_load(gh_file)
+        except yaml.YAMLError as exc:
+            print(exc)
+
+
+def extract_value(target, dict_yaml):
+    for key, value in dict_yaml.items():
+        if key == target:
+            return value
+
+
+def extract_matrix(workflow_yaml):
+    jobs = extract_value('jobs', workflow_yaml)
+    integration = extract_value('integration', jobs)
+    strategy = extract_value('strategy', integration)
+    matrix = extract_value('matrix', strategy)
+    return matrix
+
+
+def is_exclude(exclude_list, test_suite):
+    test_is_excluded = False
+    for excl in exclude_list:
+        db_engine_match = False
+        ansible_match = False
+        python_match = False
+        connector_match = False
+
+        if 'db_engine_version' not in excl:
+            continue
+        if 'ansible' not in excl:
+            continue
+        if 'python' not in excl:
+            continue
+        if 'connector' not in excl:
+            continue
+
+        if excl.get('db_engine_version') == test_suite[0]:
+            db_engine_match = True
+        if excl.get('ansible') == test_suite[1]:
+            ansible_match = True
+        if excl.get('python') == test_suite[2]:
+            python_match = True
+        if excl.get('connector') == test_suite[3]:
+            connector_match = True
+
+        if any(db_engine_match, ansible_match, python_match, connector_match):
+            test_is_excluded = True
+
+    return test_is_excluded
+
+
+def main():
+    workflow_yaml = read_github_workflow_file()
+    tests_matrix_yaml = extract_matrix(workflow_yaml)
+
+    matrix = []
+    exclude_list = tests_matrix_yaml.get('exclude')
+    for db_engine in tests_matrix_yaml.get('db_engine_version'):
+        for ansible in tests_matrix_yaml.get('ansible'):
+            for python in tests_matrix_yaml.get('python'):
+                for connector in tests_matrix_yaml.get('connector'):
+                    if not is_exclude(exclude_list, (db_engine, ansible, python, connector)):
+                        matrix.append((db_engine, ansible, python, connector))
+
+    print(len(matrix))
+
+
+if __name__ == '__main__':
+    main()