From 06f06c487c6623a02168e538f2f2f10ec6d9bfce Mon Sep 17 00:00:00 2001
From: Sergio Oliveira Campos <seocam@seocam.com>
Date: Tue, 1 Sep 2020 17:04:17 -0300
Subject: [PATCH] Added helpers to config tests for execution on idm-ci.
In order to run the tests in idm-ci we need to configure the our pytest
tests environment variables. This PR configures that automatically if an
environment variable TWD is available and $TWD/config exists.
---
requirements-tests.txt | 1 +
tests/pytests/conftest.py | 50 +++++++++++++++++++++++++++++++++++++++
tests/utils.py | 6 ++++-
3 files changed, 56 insertions(+), 1 deletion(-)
create mode 100644 tests/pytests/conftest.py
diff --git a/requirements-tests.txt b/requirements-tests.txt
index 8cd818c3..4d7ef12d 100644
--- a/requirements-tests.txt
+++ b/requirements-tests.txt
@@ -4,3 +4,4 @@ pytest-sourceorder>=0.5
pytest-split-tests>=1.0.3
testinfra>=5.0
jmespath>=0.9 # needed for the `json_query` filter
+pyyaml>=3
diff --git a/tests/pytests/conftest.py b/tests/pytests/conftest.py
new file mode 100644
index 00000000..942ffc00
--- /dev/null
+++ b/tests/pytests/conftest.py
@@ -0,0 +1,50 @@
+# Authors:
+# Sergio Oliveira Campos <seocam@redhat.com>
+#
+# Copyright (C) 2020 Red Hat
+# see file 'COPYING' for use and warranty information
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+import os
+import yaml
+
+
+def get_inventory(inventory_path):
+ with open(inventory_path) as inventory_yaml:
+ return yaml.safe_load(inventory_yaml)
+
+
+def set_env_if_not_set(envvar, value):
+ if not os.getenv(envvar):
+ os.environ[envvar] = value
+
+
+def pytest_configure(config):
+ test_dir = os.getenv("TWD")
+ if not test_dir:
+ return
+
+ config_dir = os.path.join(test_dir, "config")
+ if os.path.exists(config_dir):
+ inventory_path = os.path.join(config_dir, "test.inventory.yaml")
+ inventory = get_inventory(inventory_path)
+ print("Configuring execution using {}".format(inventory_path))
+ ipaservers = inventory["all"]["children"]["ipaserver"]["hosts"]
+ ipaserver = list(ipaservers.values())[0]
+ private_key = os.path.join(config_dir, "id_rsa")
+
+ set_env_if_not_set("ANSIBLE_PRIVATE_KEY_FILE", private_key)
+ set_env_if_not_set("IPA_SERVER_HOST", ipaserver["ansible_host"])
+ set_env_if_not_set("ANSIBLE_REMOTE_USER", ipaserver["ansible_user"])
diff --git a/tests/utils.py b/tests/utils.py
index b262d720..bd1bbf53 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -220,6 +220,7 @@ class AnsibleFreeIPATestCase(TestCase):
if is_docker_env():
protocol = "docker://"
user = ""
+ ssh_identity_file = None
else:
protocol = "ssh://"
@@ -230,9 +231,12 @@ class AnsibleFreeIPATestCase(TestCase):
current_user = os.getenv("USER")
ansible_user = os.getenv("ANSIBLE_REMOTE_USER", current_user)
user = ansible_user + password + "@"
+ ssh_identity_file = os.getenv("ANSIBLE_PRIVATE_KEY_FILE", None)
host_connection_info = protocol + user + get_server_host()
- self.master = testinfra.get_host(host_connection_info)
+ self.master = testinfra.get_host(
+ host_connection_info, ssh_identity_file=ssh_identity_file,
+ )
def run_playbook(self, playbook, allow_failures=False):
return run_playbook(playbook, allow_failures)
--
GitLab