60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
# (C) 2013, Michael Scherer, <misc@zarb.org>
|
|
|
|
# This file is part of Ansible
|
|
#
|
|
# Ansible is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Ansible is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
import os
|
|
import unittest
|
|
import subprocess
|
|
|
|
# if you change here, also change in the plugin
|
|
FILE_DISABLE = '/tmp/ansible_test_disable'
|
|
FILE_RUN = '/tmp/ansible_test_finish'
|
|
|
|
|
|
class TestInventory(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.cwd = os.getcwd()
|
|
os.chdir(os.path.join(os.path.dirname(__file__), 'test_callbacks'))
|
|
|
|
def clean_file(self):
|
|
if os.path.exists(FILE_RUN):
|
|
os.unlink(FILE_RUN)
|
|
if os.path.exists(FILE_DISABLE):
|
|
os.unlink(FILE_DISABLE)
|
|
|
|
def tearDown(self):
|
|
os.chdir(self.cwd)
|
|
|
|
def run_ansible_playbook(self):
|
|
subprocess.call('source ../../hacking/env-setup 2>&1 >/dev/null; ansible-playbook -i "127.0.0.1," test_playbook.yml 2>&1 >/dev/null', shell=True)
|
|
|
|
def test_callback(self):
|
|
self.clean_file()
|
|
|
|
self.run_ansible_playbook()
|
|
assert os.path.exists(FILE_RUN)
|
|
self.clean_file()
|
|
|
|
def test_callback_disabled(self):
|
|
self.clean_file()
|
|
open(FILE_DISABLE, 'w').close()
|
|
|
|
self.run_ansible_playbook()
|
|
assert not os.path.exists(FILE_RUN)
|
|
|
|
self.clean_file()
|