From 89c89c8ed21137ae4fcdf0770501c2b9a4121bac Mon Sep 17 00:00:00 2001 From: James Tanner Date: Mon, 10 Feb 2014 23:56:32 -0500 Subject: [PATCH] Re-add synchronize unit tests --- test/TestSynchronize.py | 132 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 132 insertions(+) create mode 100644 test/TestSynchronize.py diff --git a/test/TestSynchronize.py b/test/TestSynchronize.py new file mode 100644 index 00000000000..086287409d2 --- /dev/null +++ b/test/TestSynchronize.py @@ -0,0 +1,132 @@ + +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 + + def _execute_module(self, conn, tmp, module_name, args, inject=None): + self.executed_conn = conn + self.executed_tmp = tmp + self.executed_module_name = module_name + self.executed_args = args + self.executed_inject = inject + +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_args == "dest=root@el6.lab.net:/tmp/bar src=/tmp/foo", "wrong args used" + assert runner.sudo == False, "sudo not set to false" + + 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_args, "dest_port should not have been set" + assert "src=/tmp/foo" in runner.executed_args, "source was set incorrectly" + assert "dest=/tmp/bar" in runner.executed_args, "dest was set incorrectly" + + + def test_synchronize_action_vagrant(self): + + """ Verify the action plugin accomodates 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 "dest_port=2222" in runner.executed_args, "remote port was not set to 2222" + assert "src=/tmp/foo" in runner.executed_args, "source was set incorrectly" + assert "dest=vagrant@127.0.0.1:/tmp/bar" in runner.executed_args, "dest was set incorrectly" +