ansible/test/TestCallback.py

# (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()