diff --git a/lib/ansible/modules/network/onyx/onyx_buffer_pool.py b/lib/ansible/modules/network/onyx/onyx_buffer_pool.py new file mode 100644 index 0000000000..98be7ac50a --- /dev/null +++ b/lib/ansible/modules/network/onyx/onyx_buffer_pool.py @@ -0,0 +1,145 @@ +#!/usr/bin/python +# +# Copyright: Ansible Project +# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt) + +from __future__ import absolute_import, division, print_function +__metaclass__ = type + +ANSIBLE_METADATA = {'metadata_version': '1.1', + 'status': ['preview'], + 'supported_by': 'community'} + +DOCUMENTATION = """ +--- +module: onyx_buffer_pool +version_added: "2.8" +author: "Anas Badaha (@anasb)" +short_description: Configures Buffer Pool +description: + - This module provides declarative management of Onyx Buffer Pool configuration + on Mellanox ONYX network devices. +notes: + - Tested on ONYX 3.6.8130 +options: + name: + description: + - pool name. + required: true + pool_type: + description: + - pool type. + choices: ['lossless', 'lossy'] + default: lossy + memory_percent: + description: + - memory percent. + switch_priority: + description: + - switch priority, range 1-7. +""" + +EXAMPLES = """ +- name: configure buffer pool + onyx_buffer_pool: + name: roce + pool_type: lossless + memory_percent: 50.00 + switch_priority: 3 + +""" + +RETURN = """ +commands: + description: The list of configuration mode commands to send to the device. + returned: always + type: list + sample: + - traffic pool roce type lossless + - traffic pool roce memory percent 50.00 + - traffic pool roce map switch-priority 3 +""" + +from ansible.module_utils.basic import AnsibleModule +from ansible.module_utils.network.onyx.onyx import show_cmd +from ansible.module_utils.network.onyx.onyx import BaseOnyxModule + + +class OnyxBufferPoolModule(BaseOnyxModule): + + def init_module(self): + """ initialize module + """ + element_spec = dict( + name=dict(type='str', required=True), + pool_type=dict(choices=['lossless', 'lossy'], default='lossy'), + memory_percent=dict(type='float'), + switch_priority=dict(type='int') + ) + argument_spec = dict() + argument_spec.update(element_spec) + self._module = AnsibleModule( + argument_spec=argument_spec, + supports_check_mode=True) + + def get_required_config(self): + module_params = self._module.params + self._required_config = dict(module_params) + self.validate_param_values(self._required_config) + + def validate_switch_priority(self, value): + if value and not 0 <= int(value) <= 7: + self._module.fail_json(msg='switch_priority value must be between 0 and 7') + + def _set_traffic_pool_config(self, traffic_pool_config): + if traffic_pool_config is None: + return + traffic_pool_config = traffic_pool_config.get(self._required_config.get('name')) + self._current_config['pool_type'] = traffic_pool_config[0].get("Type") + self._current_config['switch_priority'] = int(traffic_pool_config[0].get("Switch Priorities")) + self._current_config['memory_percent'] = float(traffic_pool_config[0].get("Memory [%]")) + + def _show_traffic_pool(self): + cmd = "show traffic pool {0}".format(self._required_config.get("name")) + return show_cmd(self._module, cmd, json_fmt=True, fail_on_error=False) + + def load_current_config(self): + self._current_config = dict() + traffic_pool_config = self._show_traffic_pool() + self._set_traffic_pool_config(traffic_pool_config) + + def generate_commands(self): + name = self._required_config.get("name") + pool_type = self._required_config.get("pool_type") + + if self._current_config is None: + self._add_add_traffic_pool_cmds(name, pool_type) + else: + current_pool_type = self._current_config.get("pool_type") + if pool_type != current_pool_type: + self._add_add_traffic_pool_cmds(name, pool_type) + + memory_percent = self._required_config.get("memory_percent") + if memory_percent is not None: + curr_memory_percent = self._current_config.get("memory_percent") + if curr_memory_percent is None or memory_percent != curr_memory_percent: + self._commands.append('traffic pool {0} memory percent {1}'.format(name, memory_percent)) + + switch_priority = self._required_config.get("switch_priority") + if switch_priority is not None: + curr_switch_priority = self._current_config.get("switch_priority") + if curr_switch_priority is None or switch_priority != curr_switch_priority: + self._commands.append('traffic pool {0} map switch-priority {1}'.format(name, switch_priority)) + + def _add_add_traffic_pool_cmds(self, name, pool_type): + self._commands.append('traffic pool {0} type {1}'.format(name, pool_type)) + + +def main(): + """ main entry point for module execution + """ + OnyxBufferPoolModule.main() + + +if __name__ == '__main__': + main() diff --git a/test/units/modules/network/onyx/fixtures/onyx_buffer_pool.cfg b/test/units/modules/network/onyx/fixtures/onyx_buffer_pool.cfg new file mode 100644 index 0000000000..7172e8dc1a --- /dev/null +++ b/test/units/modules/network/onyx/fixtures/onyx_buffer_pool.cfg @@ -0,0 +1,15 @@ +{ + "roce": [ + { + "Type": "lossless", + "Switch Priorities": "3", + "Max Usage": "0", + "Usage": "0", + "Memory actual": "5.9M", + "Memory [%]": "50.00" + } + ], + "Exception list": { + "message": "N/A" + } +} diff --git a/test/units/modules/network/onyx/test_onyx_buffer_pool.py b/test/units/modules/network/onyx/test_onyx_buffer_pool.py new file mode 100644 index 0000000000..8d55c625c1 --- /dev/null +++ b/test/units/modules/network/onyx/test_onyx_buffer_pool.py @@ -0,0 +1,78 @@ +# +# Copyright: Ansible Project +# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt) + +# Make coding more python3-ish +from __future__ import (absolute_import, division, print_function) +__metaclass__ = type + +from units.compat.mock import patch +from ansible.modules.network.onyx import onyx_buffer_pool +from units.modules.utils import set_module_args +from .onyx_module import TestOnyxModule, load_fixture + + +class TestOnyxBufferPoolModule(TestOnyxModule): + + module = onyx_buffer_pool + buffer_pool_configured = False + + def setUp(self): + super(TestOnyxBufferPoolModule, self).setUp() + self.mock_get_buffer_pool_config = patch.object( + onyx_buffer_pool.OnyxBufferPoolModule, "_show_traffic_pool") + self.get_buffer_pool_config = self.mock_get_buffer_pool_config.start() + + self.mock_load_config = patch( + 'ansible.module_utils.network.onyx.onyx.load_config') + self.load_config = self.mock_load_config.start() + + def tearDown(self): + super(TestOnyxBufferPoolModule, self).tearDown() + self.mock_get_buffer_pool_config.stop() + self.mock_load_config.stop() + + def load_fixtures(self, commands=None, transport='cli'): + buffer_pool_config_file = 'onyx_buffer_pool.cfg' + self.get_buffer_pool_config.return_value = None + + if self.buffer_pool_configured is True: + buffer_pool_data = load_fixture(buffer_pool_config_file) + self.get_buffer_pool_config.return_value = buffer_pool_data + + self.load_config.return_value = None + + def test_buffer_pool_no_change(self): + self.buffer_pool_configured = True + set_module_args(dict(name="roce", pool_type="lossless", + memory_percent=50.0, switch_priority=3)) + self.execute_module(changed=False) + + def test_buffer_pool_with_change(self): + set_module_args(dict(name="roce", pool_type="lossless", + memory_percent=50.0, switch_priority=3)) + commands = ["traffic pool roce type lossless", + "traffic pool roce memory percent 50.0", + "traffic pool roce map switch-priority 3" + ] + self.execute_module(changed=True, commands=commands) + + def test_memory_percent_with_change(self): + self.buffer_pool_configured = True + set_module_args(dict(name="roce", pool_type="lossless", + memory_percent=60.0, switch_priority=3)) + commands = ["traffic pool roce memory percent 60.0"] + self.execute_module(changed=True, commands=commands) + + def test_switch_priority_with_change(self): + self.buffer_pool_configured = True + set_module_args(dict(name="roce", pool_type="lossless", + memory_percent=50.0, switch_priority=5)) + commands = ["traffic pool roce map switch-priority 5"] + self.execute_module(changed=True, commands=commands) + + def test_pool_type_with_change(self): + self.buffer_pool_configured = True + set_module_args(dict(name="roce", memory_percent=50.0, switch_priority=3)) + commands = ["traffic pool roce type lossy"] + self.execute_module(changed=True, commands=commands)