b15c8e9cb1
recieved vs expected so we can be notified if this test breaks, and also to add some more detailed playbooks that use includes & async, etc.
108 lines
3.2 KiB
Python
108 lines
3.2 KiB
Python
|
|
# tests are fairly 'live' (but safe to run)
|
|
# setup authorized_keys for logged in user such
|
|
# that the user can log in as themselves before running tests
|
|
|
|
import unittest
|
|
import getpass
|
|
import ansible.playbook
|
|
import ansible.utils as utils
|
|
import os
|
|
import shutil
|
|
import time
|
|
try:
|
|
import json
|
|
except:
|
|
import simplejson as json
|
|
|
|
class TestCallbacks(object):
|
|
|
|
def __init__(self):
|
|
self.events = []
|
|
|
|
def set_playbook(self, playbook):
|
|
self.playbook = playbook
|
|
|
|
def on_start(self):
|
|
self.events.append('start')
|
|
|
|
def on_task_start(self, name, is_conditional):
|
|
self.events.append([ 'task start', [ name, is_conditional ]])
|
|
|
|
def on_unreachable(self, host, msg):
|
|
self.events.append([ 'unreachable', [ host, msg ]])
|
|
|
|
def on_failed(self, host, results):
|
|
self.events.append([ 'failed', [ host, results ]])
|
|
|
|
# FIXME: this callback should get results too!
|
|
def on_ok(self, host):
|
|
self.events.append([ 'ok', [ host ]])
|
|
|
|
def on_play_start(self, pattern):
|
|
self.events.append([ 'play start', [ pattern ]])
|
|
|
|
def on_async_confused(self, msg):
|
|
self.events.append([ 'async confused', [ msg ]])
|
|
|
|
def on_async_poll(self, jid, host, clock, host_result):
|
|
self.events.append([ 'async poll', [ host, jid ]])
|
|
|
|
def on_dark_host(self, host, msg):
|
|
self.events.append([ 'failed/dark', [ host, msg ]])
|
|
|
|
# FIXME: callbacks need to be fired on notifiers as well
|
|
|
|
|
|
class TestRunner(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.user = getpass.getuser()
|
|
self.cwd = os.getcwd()
|
|
self.test_dir = os.path.join(self.cwd, 'test')
|
|
self.stage_dir = self._prepare_stage_dir()
|
|
|
|
def _prepare_stage_dir(self):
|
|
stage_path = os.path.join(self.test_dir, 'test_data')
|
|
if os.path.exists(stage_path):
|
|
shutil.rmtree(stage_path, ignore_errors=False)
|
|
assert not os.path.exists(stage_path)
|
|
os.makedirs(stage_path)
|
|
assert os.path.exists(stage_path)
|
|
return stage_path
|
|
|
|
def _get_test_file(self, filename):
|
|
# get a file inside the test input directory
|
|
filename = os.path.join(self.test_dir, filename)
|
|
assert os.path.exists(filename)
|
|
return filename
|
|
|
|
def _get_stage_file(self, filename):
|
|
# get a file inside the test output directory
|
|
filename = os.path.join(self.stage_dir, filename)
|
|
return filename
|
|
|
|
def _run(self, test_playbook):
|
|
''' run a module and get the localhost results '''
|
|
self.test_callbacks = TestCallbacks()
|
|
self.playbook = ansible.playbook.PlayBook(
|
|
playbook = test_playbook,
|
|
host_list = 'test/ansible_hosts',
|
|
module_path = 'library/',
|
|
forks = 1,
|
|
timeout = 5,
|
|
remote_user = self.user,
|
|
remote_pass = None,
|
|
verbose = False,
|
|
callbacks = self.test_callbacks
|
|
)
|
|
results = self.playbook.run()
|
|
return dict(
|
|
results = results,
|
|
events = self.test_callbacks.events,
|
|
)
|
|
|
|
def test_one(self):
|
|
pb = os.path.join(self.test_dir, 'playbook1.yml')
|
|
print utils.bigjson(self._run(pb))
|
|
assert False, "this test works, but we need to check the results values to complete it"
|