diff options
Diffstat (limited to 'test/units/modules/storage/netapp/test_na_ontap_job_schedule.py')
| -rw-r--r-- | test/units/modules/storage/netapp/test_na_ontap_job_schedule.py | 246 |
1 files changed, 0 insertions, 246 deletions
diff --git a/test/units/modules/storage/netapp/test_na_ontap_job_schedule.py b/test/units/modules/storage/netapp/test_na_ontap_job_schedule.py deleted file mode 100644 index 973474036b..0000000000 --- a/test/units/modules/storage/netapp/test_na_ontap_job_schedule.py +++ /dev/null @@ -1,246 +0,0 @@ -# (c) 2018, NetApp, Inc -# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt) - -''' unit tests for Ansible module: na_ontap_job_schedule ''' - -from __future__ import print_function -import json -import pytest - -from units.compat import unittest -from units.compat.mock import patch, Mock -from ansible.module_utils import basic -from ansible.module_utils._text import to_bytes -import ansible.module_utils.netapp as netapp_utils - -from ansible.modules.storage.netapp.na_ontap_job_schedule \ - import NetAppONTAPJob as job_module # module under test - -if not netapp_utils.has_netapp_lib(): - pytestmark = pytest.mark.skip('skipping as missing required netapp_lib') - - -def set_module_args(args): - """prepare arguments so that they will be picked up during module creation""" - args = json.dumps({'ANSIBLE_MODULE_ARGS': args}) - basic._ANSIBLE_ARGS = to_bytes(args) # pylint: disable=protected-access - - -class AnsibleExitJson(Exception): - """Exception class to be raised by module.exit_json and caught by the test case""" - pass - - -class AnsibleFailJson(Exception): - """Exception class to be raised by module.fail_json and caught by the test case""" - pass - - -def exit_json(*args, **kwargs): # pylint: disable=unused-argument - """function to patch over exit_json; package return data into an exception""" - if 'changed' not in kwargs: - kwargs['changed'] = False - raise AnsibleExitJson(kwargs) - - -def fail_json(*args, **kwargs): # pylint: disable=unused-argument - """function to patch over fail_json; package return data into an exception""" - kwargs['failed'] = True - raise AnsibleFailJson(kwargs) - - -class MockONTAPConnection(object): - ''' mock server connection to ONTAP host ''' - - def __init__(self, kind=None, data=None): - ''' save arguments ''' - self.kind = kind - self.params = data - self.xml_in = None - self.xml_out = None - - def invoke_successfully(self, xml, enable_tunneling): # pylint: disable=unused-argument - ''' mock invoke_successfully returning xml data ''' - self.xml_in = xml - if self.kind == 'job': - xml = self.build_job_schedule_cron_info(self.params) - elif self.kind == 'job_multiple': - xml = self.build_job_schedule_multiple_cron_info(self.params) - # TODO: mock invoke_elem for autosupport calls - elif self.kind == 'vserver': - xml = self.build_vserver_info() - self.xml_out = xml - return xml - - def autosupport_log(self): - ''' Mock autosupport log method, returns None ''' - return None - - @staticmethod - def build_job_schedule_cron_info(job_details): - ''' build xml data for vserser-info ''' - xml = netapp_utils.zapi.NaElement('xml') - attributes = { - 'num-records': 1, - 'attributes-list': { - 'job-schedule-cron-info': { - 'job-schedule-name': job_details['name'], - 'job-schedule-cron-minute': { - 'cron-minute': job_details['minutes'] - } - } - } - } - xml.translate_struct(attributes) - return xml - - @staticmethod - def build_job_schedule_multiple_cron_info(job_details): - ''' build xml data for vserser-info ''' - print("CALLED MULTIPLE BUILD") - xml = netapp_utils.zapi.NaElement('xml') - attributes = { - 'num-records': 1, - 'attributes-list': { - 'job-schedule-cron-info': { - 'job-schedule-name': job_details['name'], - 'job-schedule-cron-minute': [ - {'cron-minute': '25'}, - {'cron-minute': '35'} - ], - 'job-schedule-cron-month': [ - {'cron-month': '5'}, - {'cron-month': '10'} - ] - } - } - } - xml.translate_struct(attributes) - return xml - - -class TestMyModule(unittest.TestCase): - ''' Unit tests for na_ontap_job_schedule ''' - - def setUp(self): - self.mock_module_helper = patch.multiple(basic.AnsibleModule, - exit_json=exit_json, - fail_json=fail_json) - self.mock_module_helper.start() - self.addCleanup(self.mock_module_helper.stop) - self.mock_job = { - 'name': 'test_job', - 'minutes': '25' - } - - def mock_args(self): - return { - 'name': self.mock_job['name'], - 'job_minutes': [self.mock_job['minutes']], - 'hostname': 'test', - 'username': 'test_user', - 'password': 'test_pass!' - } - - def get_job_mock_object(self, kind=None): - """ - Helper method to return an na_ontap_job_schedule object - :param kind: passes this param to MockONTAPConnection() - :return: na_ontap_job_schedule object - """ - job_obj = job_module() - job_obj.autosupport_log = Mock(return_value=None) - if kind is None: - job_obj.server = MockONTAPConnection() - else: - job_obj.server = MockONTAPConnection(kind=kind, data=self.mock_job) - return job_obj - - def test_module_fail_when_required_args_missing(self): - ''' required arguments are reported as errors ''' - with pytest.raises(AnsibleFailJson) as exc: - set_module_args({}) - job_module() - print('Info: %s' % exc.value.args[0]['msg']) - - def test_get_nonexistent_job(self): - ''' Test if get_job_schedule returns None for non-existent job ''' - set_module_args(self.mock_args()) - result = self.get_job_mock_object().get_job_schedule() - assert result is None - - def test_get_existing_job(self): - ''' Test if get_job_schedule retuns job details for existing job ''' - data = self.mock_args() - set_module_args(data) - result = self.get_job_mock_object('job').get_job_schedule() - assert result['name'] == self.mock_job['name'] - assert result['job_minutes'] == data['job_minutes'] - - def test_get_existing_job_multiple_minutes(self): - ''' Test if get_job_schedule retuns job details for existing job ''' - set_module_args(self.mock_args()) - result = self.get_job_mock_object('job_multiple').get_job_schedule() - print(str(result)) - assert result['name'] == self.mock_job['name'] - assert result['job_minutes'] == ['25', '35'] - assert result['job_months'] == ['5', '10'] - - def test_create_error_missing_param(self): - ''' Test if create throws an error if job_minutes is not specified''' - data = self.mock_args() - del data['job_minutes'] - set_module_args(data) - with pytest.raises(AnsibleFailJson) as exc: - self.get_job_mock_object('job').create_job_schedule() - msg = 'Error: missing required parameter job_minutes for create' - assert exc.value.args[0]['msg'] == msg - - def test_successful_create(self): - ''' Test successful create ''' - set_module_args(self.mock_args()) - with pytest.raises(AnsibleExitJson) as exc: - self.get_job_mock_object().apply() - assert exc.value.args[0]['changed'] - - def test_create_idempotency(self): - ''' Test create idempotency ''' - set_module_args(self.mock_args()) - with pytest.raises(AnsibleExitJson) as exc: - self.get_job_mock_object('job').apply() - assert not exc.value.args[0]['changed'] - - def test_successful_delete(self): - ''' Test delete existing job ''' - data = self.mock_args() - data['state'] = 'absent' - set_module_args(data) - with pytest.raises(AnsibleExitJson) as exc: - self.get_job_mock_object('job').apply() - assert exc.value.args[0]['changed'] - - def test_delete_idempotency(self): - ''' Test delete idempotency ''' - data = self.mock_args() - data['state'] = 'absent' - set_module_args(data) - with pytest.raises(AnsibleExitJson) as exc: - self.get_job_mock_object().apply() - assert not exc.value.args[0]['changed'] - - def test_successful_modify(self): - ''' Test successful modify job_minutes ''' - data = self.mock_args() - data['job_minutes'] = ['20'] - set_module_args(data) - with pytest.raises(AnsibleExitJson) as exc: - self.get_job_mock_object('job').apply() - assert exc.value.args[0]['changed'] - - def test_modify_idempotency(self): - ''' Test modify idempotency ''' - data = self.mock_args() - set_module_args(data) - with pytest.raises(AnsibleExitJson) as exc: - self.get_job_mock_object('job').apply() - assert not exc.value.args[0]['changed'] |