176 lines
6.8 KiB
Python
176 lines
6.8 KiB
Python
|
|
import unittest
|
|
import getpass
|
|
import os
|
|
import shutil
|
|
import time
|
|
import tempfile
|
|
from nose.plugins.skip import SkipTest
|
|
|
|
from ansible.runner.action_plugins.synchronize import ActionModule as Synchronize
|
|
|
|
class FakeRunner(object):
|
|
def __init__(self):
|
|
self.connection = None
|
|
self.transport = None
|
|
self.basedir = None
|
|
self.sudo = None
|
|
self.remote_user = None
|
|
self.private_key_file = None
|
|
self.check = False
|
|
self.become = False
|
|
self.become_method = 'sudo'
|
|
self.become_user = False
|
|
|
|
def _execute_module(self, conn, tmp, module_name, args,
|
|
async_jid=None, async_module=None, async_limit=None, inject=None,
|
|
persist_files=False, complex_args=None, delete_remote_tmp=True):
|
|
self.executed_conn = conn
|
|
self.executed_tmp = tmp
|
|
self.executed_module_name = module_name
|
|
self.executed_args = args
|
|
self.executed_async_jid = async_jid
|
|
self.executed_async_module = async_module
|
|
self.executed_async_limit = async_limit
|
|
self.executed_inject = inject
|
|
self.executed_persist_files = persist_files
|
|
self.executed_complex_args = complex_args
|
|
self.executed_delete_remote_tmp = delete_remote_tmp
|
|
|
|
def noop_on_check(self, inject):
|
|
return self.check
|
|
|
|
class FakeConn(object):
|
|
def __init__(self):
|
|
self.host = None
|
|
self.delegate = None
|
|
|
|
class TestSynchronize(unittest.TestCase):
|
|
|
|
|
|
def test_synchronize_action_basic(self):
|
|
|
|
""" verify the synchronize action plugin sets
|
|
the delegate to 127.0.0.1 and remote path to user@host:/path """
|
|
|
|
runner = FakeRunner()
|
|
runner.remote_user = "root"
|
|
runner.transport = "ssh"
|
|
conn = FakeConn()
|
|
inject = {
|
|
'inventory_hostname': "el6.lab.net",
|
|
'inventory_hostname_short': "el6",
|
|
'ansible_connection': None,
|
|
'ansible_ssh_user': 'root',
|
|
'delegate_to': None,
|
|
'playbook_dir': '.',
|
|
}
|
|
|
|
x = Synchronize(runner)
|
|
x.setup("synchronize", inject)
|
|
x.run(conn, "/tmp", "synchronize", "src=/tmp/foo dest=/tmp/bar", inject)
|
|
|
|
assert runner.executed_inject['delegate_to'] == "127.0.0.1", "was not delegated to 127.0.0.1"
|
|
assert runner.executed_complex_args == {"dest":"root@el6.lab.net:/tmp/bar", "src":"/tmp/foo"}, "wrong args used"
|
|
assert runner.sudo == None, "sudo was not reset to None"
|
|
|
|
def test_synchronize_action_sudo(self):
|
|
|
|
""" verify the synchronize action plugin unsets and then sets sudo """
|
|
|
|
runner = FakeRunner()
|
|
runner.become = True
|
|
runner.remote_user = "root"
|
|
runner.transport = "ssh"
|
|
conn = FakeConn()
|
|
inject = {
|
|
'inventory_hostname': "el6.lab.net",
|
|
'inventory_hostname_short': "el6",
|
|
'ansible_connection': None,
|
|
'ansible_ssh_user': 'root',
|
|
'delegate_to': None,
|
|
'playbook_dir': '.',
|
|
}
|
|
|
|
x = Synchronize(runner)
|
|
x.setup("synchronize", inject)
|
|
x.run(conn, "/tmp", "synchronize", "src=/tmp/foo dest=/tmp/bar", inject)
|
|
|
|
assert runner.executed_inject['delegate_to'] == "127.0.0.1", "was not delegated to 127.0.0.1"
|
|
assert runner.executed_complex_args == {'dest':'root@el6.lab.net:/tmp/bar',
|
|
'src':'/tmp/foo',
|
|
'rsync_path':'"sudo rsync"'}, "wrong args used"
|
|
assert runner.become == True, "sudo was not reset to True"
|
|
|
|
|
|
def test_synchronize_action_local(self):
|
|
|
|
""" verify the synchronize action plugin sets
|
|
the delegate to 127.0.0.1 and does not alter the dest """
|
|
|
|
runner = FakeRunner()
|
|
runner.remote_user = "jtanner"
|
|
runner.transport = "paramiko"
|
|
conn = FakeConn()
|
|
conn.host = "127.0.0.1"
|
|
conn.delegate = "thishost"
|
|
inject = {
|
|
'inventory_hostname': "thishost",
|
|
'ansible_ssh_host': '127.0.0.1',
|
|
'ansible_connection': 'local',
|
|
'delegate_to': None,
|
|
'playbook_dir': '.',
|
|
}
|
|
|
|
x = Synchronize(runner)
|
|
x.setup("synchronize", inject)
|
|
x.run(conn, "/tmp", "synchronize", "src=/tmp/foo dest=/tmp/bar", inject)
|
|
|
|
assert runner.transport == "paramiko", "runner transport was changed"
|
|
assert runner.remote_user == "jtanner", "runner remote_user was changed"
|
|
assert runner.executed_inject['delegate_to'] == "127.0.0.1", "was not delegated to 127.0.0.1"
|
|
assert "dest_port" not in runner.executed_complex_args, "dest_port should not have been set"
|
|
assert runner.executed_complex_args.get("src") == "/tmp/foo", "source was set incorrectly"
|
|
assert runner.executed_complex_args.get("dest") == "/tmp/bar", "dest was set incorrectly"
|
|
|
|
|
|
def test_synchronize_action_vagrant(self):
|
|
|
|
""" Verify the action plugin accommodates the common
|
|
scenarios for vagrant boxes. """
|
|
|
|
runner = FakeRunner()
|
|
runner.remote_user = "jtanner"
|
|
runner.transport = "ssh"
|
|
conn = FakeConn()
|
|
conn.host = "127.0.0.1"
|
|
conn.delegate = "thishost"
|
|
inject = {
|
|
'inventory_hostname': "thishost",
|
|
'ansible_ssh_user': 'vagrant',
|
|
'ansible_ssh_host': '127.0.0.1',
|
|
'ansible_ssh_port': '2222',
|
|
'delegate_to': None,
|
|
'playbook_dir': '.',
|
|
'hostvars': {
|
|
'thishost': {
|
|
'inventory_hostname': 'thishost',
|
|
'ansible_ssh_port': '2222',
|
|
'ansible_ssh_host': '127.0.0.1',
|
|
'ansible_ssh_user': 'vagrant'
|
|
}
|
|
}
|
|
}
|
|
|
|
x = Synchronize(runner)
|
|
x.setup("synchronize", inject)
|
|
x.run(conn, "/tmp", "synchronize", "src=/tmp/foo dest=/tmp/bar", inject)
|
|
|
|
assert runner.transport == "ssh", "runner transport was changed"
|
|
assert runner.remote_user == "jtanner", "runner remote_user was changed"
|
|
assert runner.executed_inject['delegate_to'] == "127.0.0.1", "was not delegated to 127.0.0.1"
|
|
assert runner.executed_inject['ansible_ssh_user'] == "vagrant", "runner user was changed"
|
|
assert runner.executed_complex_args.get("dest_port") == "2222", "remote port was not set to 2222"
|
|
assert runner.executed_complex_args.get("src") == "/tmp/foo", "source was set incorrectly"
|
|
assert runner.executed_complex_args.get("dest") == "vagrant@127.0.0.1:/tmp/bar", "dest was set incorrectly"
|
|
|