Add initial Resource Dispatcher tests

.github/workflows/resource-dispatcher-pr-checks.yml

@@ -12,7 +12,19 @@ jobs:
     runs-on: ubuntu-latest
     steps:
     - uses: actions/checkout@v2
-    - uses: vrutkovs/action-s2i@master
+    - uses: actions/setup-python@v2
+      with:
+        python-version: '3.8'
+    - name: Install Pipenv & Run Test
+      run: |
+        export PATH="$PATH:/home/runner/.local/bin"
+        python -m pip install --upgrade pip
+        pip install pipenv
+        cd resource-dispatcher
+        pipenv install
+        pipenv run pytest
+    - name: Build Image
+      uses: vrutkovs/action-s2i@master
       with:
         path: resource-dispatcher
         base: registry.access.redhat.com/ubi8/python-38:latest

resource-dispatcher/.gitignore

@@ -1,2 +1,4 @@
 **/__pycache__/
-tmp/
+tmp/
+.pytest_cache
+*.pyc

resource-dispatcher/Pipfile

@@ -14,6 +14,7 @@ ansible = "*"
 kubernetes = "*"
 inflect = "*"
 six = "*"
+pytest = "*"
 
 [requires]
 python_version = "3.8"

resource-dispatcher/configuration/test/config_exists.yml

@@ -0,0 +1,2 @@
+---
+hello: world

resource-dispatcher/configuration/test/test_configuration.py

@@ -0,0 +1,15 @@
+import pytest
+import os
+import configuration
+
+
+def test_exits_without_configuration():
+    with pytest.raises(SystemExit) as pytest_wrapped_e:
+        configuration.get_config()
+    assert pytest_wrapped_e.type == SystemExit
+
+
+def test_get_config(monkeypatch):
+    monkeypatch.setenv("CONFIG_FILE", os.getcwd() + "/configuration/test/config_exists.yml")
+    config = configuration.get_config()
+    assert {"hello": "world"} == config

resource-dispatcher/execution/test/test_ansible.py

@@ -0,0 +1,82 @@
+import pytest
+import execution
+import os
+
+def test_ansible_basic(capsys):
+    task = {
+        "name": "Ansible Test",
+        "disable_error_handling": True,
+        "steps": [
+            {
+                "plugin": "ansible",
+                "params": {
+                    "playbook_path": os.getcwd() + "/execution/test/test_playbook/site.yml",
+                    "inventory_path": os.getcwd() + "/execution/test/test_playbook/inventory"
+                }
+            }
+        ]
+    }
+
+    execution.task_function(task, context={})
+
+    captured = capsys.readouterr()
+    assert "Hello, world!" in captured.out
+
+
+def test_ansible_reading_context(capsys):
+    task = {
+        "name": "Ansible Test",
+        "disable_error_handling": True,
+        "steps": [
+            {
+                "plugin": "script",
+                "params": {
+                    "script": "context['test_var'] = 'Set via context!'"
+                }
+            },
+            {
+                "plugin": "ansible",
+                "params": {
+                    "playbook_path": os.getcwd() + "/execution/test/test_playbook/site.yml",
+                    "inventory_path": os.getcwd() + "/execution/test/test_playbook/inventory"
+                }
+            }
+        ]
+    }
+
+    execution.task_function(task, context={})
+
+    captured = capsys.readouterr()
+    assert "Set via context!" in captured.out
+
+
+def test_ansible_merging_params_from_context(capsys):
+    task = {
+        "name": "Ansible Test",
+        "disable_error_handling": True,
+        "steps": [
+            {
+                "plugin": "script",
+                "params": {
+                    "script": "context['test_var'] = {'extra_vars': {'test_var_1': 'thing1'}}"
+                }
+            },
+            {
+                "plugin": "ansible",
+                "params_from_context": "test_var",
+                "params": {
+                    "playbook_path": os.getcwd() + "/execution/test/test_playbook/site.yml",
+                    "inventory_path": os.getcwd() + "/execution/test/test_playbook/inventory",
+                    "extra_vars": {
+                        "test_var_2": "thing2"
+                    }
+                }
+            }
+        ]
+    }
+
+    execution.task_function(task, context={})
+
+    captured = capsys.readouterr()
+    assert "thing1" in captured.out
+    assert "thing2" in captured.out

resource-dispatcher/execution/test/test_execution.py

@@ -0,0 +1,9 @@
+import pytest
+from importlib.util import find_spec
+import execution
+
+
+def test_all_plugins_loadable():
+    plugins = ["ansible", "git", "kubernetes", "script"]
+    for plugin in plugins:
+        assert find_spec(f"execution.plugins.{plugin}") is not None

resource-dispatcher/execution/test/test_playbook/inventory/host_vars/localhost.yml

@@ -0,0 +1 @@
+ansible_connection: local

resource-dispatcher/execution/test/test_playbook/inventory/hosts

@@ -0,0 +1,2 @@
+[local]
+localhost

resource-dispatcher/execution/test/test_playbook/site.yml

@@ -0,0 +1,16 @@
+---
+- name: Test Playbook
+  hosts: local
+  gather_facts: false
+  tasks:
+  - debug: 
+      msg: Hello, world!
+  - debug:
+      var: test_var
+    when: test_var is defined
+  - debug:
+      var: test_var_1
+    when: test_var_1 is defined
+  - debug:
+      var: test_var_2
+    when: test_var_2 is defined