12345678910111213141516171819202122232425262728293031323334353637383940414243444546 |
- import os
- import unittest
- from kalliope.neurons.ansible_playbook import Ansible_playbook
- from kalliope.core.NeuronModule import MissingParameterException
- class TestAnsible_Playbook(unittest.TestCase):
- def setUp(self):
- self.task_file = "task_file"
- self.random = "random"
- self.test_file = "/tmp/kalliope_text_ansible_playbook.txt"
- def testParameters(self):
- def run_test(parameters_to_test):
- with self.assertRaises(MissingParameterException):
- Ansible_playbook(**parameters_to_test)
- # empty
- parameters = dict()
- run_test(parameters)
- # missing task_file
- parameters = {
- "random": self.random
- }
- run_test(parameters)
- def test_create_file_via_ansible_playbook(self):
- """
- This test will use an ansible playbook the create a file. We check that the file has been created
- """
- param = {
- "task_file": "kalliope/neurons/ansible_playbook/tests/test_ansible_playbook_neuron.yml"
- }
- Ansible_playbook(**param)
- self.assertTrue(os.path.isfile(self.test_file))
- if os.path.exists(self.test_file):
- os.remove(self.test_file)
- if __name__ == '__main__':
- unittest.main()
|