From 4ced885619ee9c2893dca57eac25c9e9907e7196 Mon Sep 17 00:00:00 2001
From: David Wilson <dw@botanicus.net>
Date: Tue, 30 Jul 2019 22:11:04 +0100
Subject: [PATCH] issue #543: localhost_ansible scripts.

---
 .ci/localhost_ansible_install.py | 16 ++++++++++++
 .ci/localhost_ansible_tests.py   | 43 ++++++++++++++++++++++++++++++++
 2 files changed, 59 insertions(+)
 create mode 100755 .ci/localhost_ansible_install.py
 create mode 100755 .ci/localhost_ansible_tests.py

diff --git a/.ci/localhost_ansible_install.py b/.ci/localhost_ansible_install.py
new file mode 100755
index 00000000..0cb47374
--- /dev/null
+++ b/.ci/localhost_ansible_install.py
@@ -0,0 +1,16 @@
+#!/usr/bin/env python
+
+import ci_lib
+
+batches = [
+    [
+        # Must be installed separately, as PyNACL indirect requirement causes
+        # newer version to be installed if done in a single pip run.
+        'pip install "pycparser<2.19" "idna<2.7"',
+        'pip install '
+            '-r tests/requirements.txt '
+            '-r tests/ansible/requirements.txt',
+    ]
+]
+
+ci_lib.run_batches(batches)
diff --git a/.ci/localhost_ansible_tests.py b/.ci/localhost_ansible_tests.py
new file mode 100755
index 00000000..71542d21
--- /dev/null
+++ b/.ci/localhost_ansible_tests.py
@@ -0,0 +1,43 @@
+#!/usr/bin/env python
+# Run tests/ansible/all.yml under Ansible and Ansible-Mitogen
+
+import glob
+import os
+import sys
+
+import ci_lib
+from ci_lib import run
+
+
+TESTS_DIR = os.path.join(ci_lib.GIT_ROOT, 'tests/ansible')
+IMAGE_PREP_DIR = os.path.join(ci_lib.GIT_ROOT, 'tests/image_prep')
+HOSTS_DIR = os.path.join(TESTS_DIR, 'hosts')
+
+
+with ci_lib.Fold('unit_tests'):
+    os.environ['SKIP_MITOGEN'] = '1'
+    ci_lib.run('./run_tests -v')
+
+
+with ci_lib.Fold('job_setup'):
+    # Don't set -U as that will upgrade Paramiko to a non-2.6 compatible version.
+    run("pip install -q ansible==%s", ci_lib.ANSIBLE_VERSION)
+
+    os.chdir(TESTS_DIR)
+    os.chmod('../data/docker/mitogen__has_sudo_pubkey.key', int('0600', 7))
+
+    if not ci_lib.exists_in_path('sshpass'):
+        run("brew install sshpass")
+
+
+with ci_lib.Fold('machine_prep'):
+    if os.path.expanduser('~mitogen__user1') == '~mitogen_user1':
+        os.chdir(IMAGE_PREP_DIR)
+        run("ansible-playbook -i localhost, _user_accounts.yml")
+
+
+with ci_lib.Fold('ansible'):
+    os.chdir(TESTS_DIR)
+    playbook = os.environ.get('PLAYBOOK', 'all.yml')
+    run('./run_ansible_playbook.py %s -l target %s',
+        playbook, HOSTS_DIR, ' '.join(sys.argv[1:]))