def test_run_playbooks_list_tasks_arg(self, mock_validate, mock_vars,
mock_run):
mock_vars.return_value = []
parser = argparse.ArgumentParser()
ansible.add_args(parser)
vault.add_args(parser)
args = [
"--list-tasks",
]
parsed_args = parser.parse_args(args)
kwargs = {"list_tasks": False}
ansible.run_playbooks(parsed_args, ["playbook1.yml", "playbook2.yml"],
**kwargs)
expected_cmd = [
"ansible-playbook",
"--inventory",
"/etc/kayobe/inventory",
"playbook1.yml",
"playbook2.yml",
]
expected_env = {"KAYOBE_CONFIG_PATH": "/etc/kayobe"}
mock_run.assert_called_once_with(expected_cmd,
check_output=False,
quiet=False,
env=expected_env)
mock_vars.assert_called_once_with("/etc/kayobe")
def test_run_playbooks_vault_password_helper(self, mock_validate,
mock_vars, mock_run):
mock_vars.return_value = []
parser = argparse.ArgumentParser()
mock_run.return_value = "/path/to/kayobe-vault-password-helper"
ansible.add_args(parser)
vault.add_args(parser)
mock_run.assert_called_once_with(
["which", "kayobe-vault-password-helper"], check_output=True)
mock_run.reset_mock()
parsed_args = parser.parse_args([])
ansible.run_playbooks(parsed_args, ["playbook1.yml"])
expected_cmd = [
"ansible-playbook",
"--vault-password-file",
"/path/to/kayobe-vault-password-helper",
"--inventory",
"/etc/kayobe/inventory",
"playbook1.yml",
]
expected_env = {
"KAYOBE_CONFIG_PATH": "/etc/kayobe",
"KAYOBE_VAULT_PASSWORD": "******"
}
mock_run.assert_called_once_with(expected_cmd,
check_output=False,
quiet=False,
env=expected_env)
def test_run_playbooks(self, mock_validate, mock_vars, mock_run):
mock_vars.return_value = [
"/etc/kayobe/vars-file1.yml", "/etc/kayobe/vars-file2.yaml"
]
parser = argparse.ArgumentParser()
ansible.add_args(parser)
vault.add_args(parser)
parsed_args = parser.parse_args([])
ansible.run_playbooks(parsed_args, ["playbook1.yml", "playbook2.yml"])
expected_cmd = [
"ansible-playbook",
"--inventory",
"/etc/kayobe/inventory",
"-e",
"@/etc/kayobe/vars-file1.yml",
"-e",
"@/etc/kayobe/vars-file2.yaml",
"playbook1.yml",
"playbook2.yml",
]
expected_env = {"KAYOBE_CONFIG_PATH": "/etc/kayobe"}
mock_run.assert_called_once_with(expected_cmd,
check_output=False,
quiet=False,
env=expected_env)
mock_vars.assert_called_once_with("/etc/kayobe")
def test_multiple_inventory_args(self, mock_validate, mock_vars, mock_run):
mock_vars.return_value = []
parser = argparse.ArgumentParser()
ansible.add_args(parser)
vault.add_args(parser)
args = [
"--inventory",
"/etc/kayobe/inventory",
"--inventory",
"/etc/kayobe/environments/foobar/inventory",
]
parsed_args = parser.parse_args(args)
ansible.run_playbooks(parsed_args, ["playbook1.yml", "playbook2.yml"])
expected_cmd = [
"ansible-playbook",
"--inventory",
"/etc/kayobe/inventory",
"--inventory",
"/etc/kayobe/environments/foobar/inventory",
"playbook1.yml",
"playbook2.yml",
]
expected_env = {"KAYOBE_CONFIG_PATH": "/etc/kayobe"}
mock_run.assert_called_once_with(expected_cmd,
check_output=False,
quiet=False,
env=expected_env)
mock_vars.assert_called_once_with(["/etc/kayobe"])
def test_run_playbooks_vault_password_file(self, mock_update,
mock_validate, mock_vars,
mock_run):
mock_vars.return_value = []
parser = argparse.ArgumentParser()
ansible.add_args(parser)
vault.add_args(parser)
args = [
"--vault-password-file",
"/path/to/vault/pw",
]
parsed_args = parser.parse_args(args)
ansible.run_playbooks(parsed_args, ["playbook1.yml"])
expected_cmd = [
"ansible-playbook",
"--vault-password-file",
"/path/to/vault/pw",
"--inventory",
"/etc/kayobe/inventory",
"playbook1.yml",
]
expected_env = {"KAYOBE_CONFIG_PATH": "/etc/kayobe"}
mock_run.assert_called_once_with(expected_cmd,
check_output=False,
quiet=False,
env=expected_env)
mock_update.assert_called_once_with(mock.ANY, expected_env)
def test_run_playbooks_ansible_cfg_env(self, mock_validate, mock_vars,
mock_readable, mock_run):
mock_vars.return_value = []
mock_readable.return_value = {"result": True}
os.environ["ANSIBLE_CONFIG"] = "/path/to/ansible.cfg"
parser = argparse.ArgumentParser()
ansible.add_args(parser)
vault.add_args(parser)
parsed_args = parser.parse_args([])
ansible.run_playbooks(parsed_args, ["playbook1.yml"])
expected_cmd = [
"ansible-playbook",
"--inventory",
"/etc/kayobe/inventory",
"playbook1.yml",
]
expected_env = {
"ANSIBLE_CONFIG": "/path/to/ansible.cfg",
"KAYOBE_CONFIG_PATH": "/etc/kayobe"
}
mock_run.assert_called_once_with(expected_cmd,
check_output=False,
quiet=False,
env=expected_env)
mock_vars.assert_called_once_with(["/etc/kayobe"])
mock_readable.assert_called_once_with("/etc/kayobe/ansible.cfg")
def test_run_playbooks_all_the_args(self, mock_validate, mock_vars,
mock_run):
mock_vars.return_value = [
"/path/to/config/vars-file1.yml", "/path/to/config/vars-file2.yaml"
]
parser = argparse.ArgumentParser()
ansible.add_args(parser)
vault.add_args(parser)
args = [
"-b",
"-C",
"--config-path",
"/path/to/config",
"--environment",
"test-env",
"-e",
"ev_name1=ev_value1",
"-i",
"/path/to/inventory",
"-l",
"group1:host",
"-t",
"tag1,tag2",
"-lt",
]
parsed_args = parser.parse_args(args)
ansible.run_playbooks(parsed_args, ["playbook1.yml", "playbook2.yml"],
verbose_level=2)
expected_cmd = [
"ansible-playbook",
"-vv",
"--list-tasks",
"--inventory",
"/path/to/inventory",
"-e",
"@/path/to/config/vars-file1.yml",
"-e",
"@/path/to/config/vars-file2.yaml",
"-e",
"ev_name1=ev_value1",
"--become",
"--check",
"--limit",
"group1:host",
"--tags",
"tag1,tag2",
"playbook1.yml",
"playbook2.yml",
]
expected_env = {
"KAYOBE_CONFIG_PATH": "/path/to/config",
"KAYOBE_ENVIRONMENT": "test-env"
}
mock_run.assert_called_once_with(expected_cmd,
check_output=False,
quiet=False,
env=expected_env)
mock_vars.assert_called_once_with(
["/path/to/config", "/path/to/config/environments/test-env"])
def test_run_playbooks_all_the_long_args(self, mock_validate, mock_vars,
mock_run):
mock_vars.return_value = [
"/path/to/config/vars-file1.yml", "/path/to/config/vars-file2.yaml"
]
parser = argparse.ArgumentParser()
ansible.add_args(parser)
vault.add_args(parser)
args = [
"--ask-vault-pass",
"--become",
"--check",
"--config-path",
"/path/to/config",
"--extra-vars",
"ev_name1=ev_value1",
"--inventory",
"/path/to/inventory",
"--limit",
"group1:host1",
"--skip-tags",
"tag3,tag4",
"--tags",
"tag1,tag2",
"--list-tasks",
]
parsed_args = parser.parse_args(args)
ansible.run_playbooks(parsed_args, ["playbook1.yml", "playbook2.yml"])
expected_cmd = [
"ansible-playbook",
"--list-tasks",
"--ask-vault-pass",
"--inventory",
"/path/to/inventory",
"-e",
"@/path/to/config/vars-file1.yml",
"-e",
"@/path/to/config/vars-file2.yaml",
"-e",
"ev_name1=ev_value1",
"--become",
"--check",
"--limit",
"group1:host1",
"--skip-tags",
"tag3,tag4",
"--tags",
"tag1,tag2",
"playbook1.yml",
"playbook2.yml",
]
expected_env = {"KAYOBE_CONFIG_PATH": "/path/to/config"}
mock_run.assert_called_once_with(expected_cmd,
quiet=False,
env=expected_env)
mock_vars.assert_called_once_with("/path/to/config")
def test_run_playbooks_func_args(self, mock_validate, mock_vars, mock_run):
mock_vars.return_value = [
"/etc/kayobe/vars-file1.yml", "/etc/kayobe/vars-file2.yaml"
]
parser = argparse.ArgumentParser()
ansible.add_args(parser)
vault.add_args(parser)
args = [
"--extra-vars",
"ev_name1=ev_value1",
"--limit",
"group1:host1",
"--tags",
"tag1,tag2",
]
parsed_args = parser.parse_args(args)
kwargs = {
"extra_vars": {
"ev_name2": "ev_value2"
},
"limit": "group2:host2",
"tags": "tag3,tag4",
"verbose_level": 0,
"check": True,
"diff": True,
}
ansible.run_playbooks(parsed_args, ["playbook1.yml", "playbook2.yml"],
**kwargs)
expected_cmd = [
"ansible-playbook",
"--inventory",
"/etc/kayobe/inventory",
"-e",
"@/etc/kayobe/vars-file1.yml",
"-e",
"@/etc/kayobe/vars-file2.yaml",
"-e",
"ev_name1=ev_value1",
"-e",
"ev_name2='ev_value2'",
"--check",
"--diff",
"--limit",
"group1:host1:&group2:host2",
"--tags",
"tag1,tag2,tag3,tag4",
"playbook1.yml",
"playbook2.yml",
]
expected_env = {"KAYOBE_CONFIG_PATH": "/etc/kayobe"}
mock_run.assert_called_once_with(expected_cmd,
check_output=False,
quiet=False,
env=expected_env)
mock_vars.assert_called_once_with(["/etc/kayobe"])
def test_run_playbooks_all_the_args(self, mock_validate, mock_vars,
mock_run):
mock_vars.return_value = [
"/path/to/config/vars-file1.yml", "/path/to/config/vars-file2.yaml"
]
parser = argparse.ArgumentParser()
ansible.add_args(parser)
vault.add_args(parser)
args = [
"-b",
"-C",
"--config-path",
"/path/to/config",
"-e",
"ev_name1=ev_value1",
"-i",
"/path/to/inventory",
"-l",
"group1:host",
"-t",
"tag1,tag2",
]
parsed_args = parser.parse_args(args)
ansible.run_playbooks(parsed_args, ["playbook1.yml", "playbook2.yml"])
expected_cmd = [
"ansible-playbook",
"--inventory",
"/path/to/inventory",
"-e",
"@/path/to/config/vars-file1.yml",
"-e",
"@/path/to/config/vars-file2.yaml",
"-e",
"ev_name1=ev_value1",
"--become",
"--check",
"--limit",
"group1:host",
"--tags",
"tag1,tag2",
"playbook1.yml",
"playbook2.yml",
]
mock_run.assert_called_once_with(expected_cmd, quiet=False)
mock_vars.assert_called_once_with("/path/to/config")
def test_run_playbooks_vault_password_file(self, mock_validate, mock_vars,
mock_run):
mock_vars.return_value = []
parser = argparse.ArgumentParser()
ansible.add_args(parser)
vault.add_args(parser)
args = [
"--vault-password-file", "/path/to/vault/pw",
]
parsed_args = parser.parse_args(args)
ansible.run_playbooks(parsed_args, ["playbook1.yml"])
expected_cmd = [
"ansible-playbook",
"--vault-password-file", "/path/to/vault/pw",
"--inventory", "/etc/kayobe/inventory",
"playbook1.yml",
]
mock_run.assert_called_once_with(expected_cmd, quiet=False)
def test_multiple_inventories(self, mock_validate, mock_vars, mock_run,
mock_exists):
mock_vars.return_value = []
# os.path.exists gets called three times:
# 1) shared inventory
# 2) environment inventory
# 3) ansible.cfg
mock_exists.side_effect = [True, True, False]
parser = argparse.ArgumentParser()
ansible.add_args(parser)
vault.add_args(parser)
args = [
"--environment",
"test-env",
]
parsed_args = parser.parse_args(args)
ansible.run_playbooks(parsed_args, ["playbook1.yml", "playbook2.yml"])
expected_cmd = [
"ansible-playbook",
"--inventory",
"/etc/kayobe/inventory",
"--inventory",
"/etc/kayobe/environments/test-env/inventory",
"playbook1.yml",
"playbook2.yml",
]
expected_env = {
"KAYOBE_CONFIG_PATH": "/etc/kayobe",
"KAYOBE_ENVIRONMENT": "test-env"
}
expected_calls = [
mock.call("/etc/kayobe/inventory"),
mock.call("/etc/kayobe/environments/test-env/inventory"),
mock.call("/etc/kayobe/ansible.cfg"),
]
self.assertEqual(expected_calls, mock_exists.mock_calls)
mock_run.assert_called_once_with(expected_cmd,
check_output=False,
quiet=False,
env=expected_env)
mock_vars.assert_called_once_with(
["/etc/kayobe", "/etc/kayobe/environments/test-env"])
def test_run_playbooks_all_the_long_args(self, mock_ask, mock_validate,
mock_vars, mock_run):
mock_vars.return_value = [
"/path/to/config/vars-file1.yml", "/path/to/config/vars-file2.yaml"
]
parser = argparse.ArgumentParser()
ansible.add_args(parser)
vault.add_args(parser)
mock_ask.return_value = "test-pass"
args = [
"--ask-vault-pass",
"--become",
"--check",
"--config-path",
"/path/to/config",
"--extra-vars",
"ev_name1=ev_value1",
"--inventory",
"/path/to/inventory",
"--limit",
"group1:host1",
"--skip-tags",
"tag3,tag4",
"--tags",
"tag1,tag2",
"--list-tasks",
]
parsed_args = parser.parse_args(args)
mock_run.return_value = "/path/to/kayobe-vault-password-helper"
ansible.run_playbooks(parsed_args, ["playbook1.yml", "playbook2.yml"])
expected_cmd = [
"ansible-playbook",
"--list-tasks",
"--vault-password-file",
"/path/to/kayobe-vault-password-helper",
"--inventory",
"/path/to/inventory",
"-e",
"@/path/to/config/vars-file1.yml",
"-e",
"@/path/to/config/vars-file2.yaml",
"-e",
"ev_name1=ev_value1",
"--become",
"--check",
"--limit",
"group1:host1",
"--skip-tags",
"tag3,tag4",
"--tags",
"tag1,tag2",
"playbook1.yml",
"playbook2.yml",
]
expected_env = {
"KAYOBE_CONFIG_PATH": "/path/to/config",
"KAYOBE_VAULT_PASSWORD": "******"
}
expected_calls = [
mock.call(["which", "kayobe-vault-password-helper"],
check_output=True),
mock.call(expected_cmd,
check_output=False,
quiet=False,
env=expected_env)
]
self.assertEqual(expected_calls, mock_run.mock_calls)
mock_vars.assert_called_once_with("/path/to/config")
def run_kayobe_playbooks(self, *args, **kwargs):
kwargs.update(self._get_verbosity_args())
return ansible.run_playbooks(*args, **kwargs)
def run_kayobe_playbooks(self, *args, **kwargs):
kwargs.update(self._get_verbosity_args())
return ansible.run_playbooks(*args, **kwargs)
