nxos_vrf refactor (#24280)

* nxos_vrf refactor

Signed-off-by: Trishna Guha <trishnaguha17@gmail.com>

* Unit test for nxos_vrf
Remove unnecessary keys() method
Signed-off-by: Trishna Guha <trishnaguha17@gmail.com>

lib/ansible/modules/network/nxos/nxos_vrf.py

@@ -16,9 +16,11 @@
 # along with Ansible.  If not, see <http://www.gnu.org/licenses/>.
 #
 
-ANSIBLE_METADATA = {'metadata_version': '1.0',
+ANSIBLE_METADATA = {
+    'metadata_version': '1.0',
     'status': ['preview'],
-                    'supported_by': 'community'}
+    'supported_by': 'community'
+}
 
 
 DOCUMENTATION = '''
@@ -84,40 +86,15 @@ EXAMPLES = '''
 - name: Ensure ntc VRF exists on switch
   nxos_vrf:
     vrf: ntc
-    username: "{{ un }}"
+    state: present
-    password: "{{ pwd }}"
-    host: "{{ inventory_hostname }}"
 '''
 
 RETURN = '''
-proposed:
+commands:
-    description: k/v pairs of parameters passed into module
-    returned: always
-    type: dict
-    sample: {"admin_state": "Up", "description": "Test test",
-            "vrf": "ntc"}
-existing:
-    description: k/v pairs of existing vrf
-    returned: always
-    type: dict
-    sample: {"admin_state": "Up", "description": "Old test",
-            "vrf": "old_ntc"}
-end_state:
-    description: k/v pairs of vrf info after module execution
-    returned: always
-    type: dict
-    sample: {"admin_state": "Up", "description": "Test test",
-            "vrf": "ntc"}
-updates:
     description: commands sent to the device
     returned: always
     type: list
     sample: ["vrf context ntc", "shutdown"]
-changed:
-    description: check to see if a change was made on the device
-    returned: always
-    type: boolean
-    sample: true
 '''
 import re
 
@@ -139,14 +116,10 @@ def execute_show_command(command, module):
 
 def apply_key_map(key_map, table):
     new_dict = {}
-    for key, value in table.items():
+    for key in table:
         new_key = key_map.get(key)
         if new_key:
-            value = table.get(key)
+            new_dict[new_key] = str(table.get(key))
-            if value:
-                new_dict[new_key] = str(value)
-            else:
-                new_dict[new_key] = value
     return new_dict
 
 
@@ -177,14 +150,14 @@ def get_vrf_description(vrf, module):
 
     description = ''
     descr_regex = r".*description\s(?P<descr>[\S+\s]+).*"
-    body = execute_show_command(command, module)
 
     try:
-        body = body[0]
+        body = execute_show_command(command, module)[0]
-        splitted_body = body.split('\n')
+    except IndexError:
-    except (AttributeError, IndexError):
         return description
 
+    if body:
+        splitted_body = body.split('\n')
         for element in splitted_body:
             if 'description' in element:
                 match_description = re.match(descr_regex, element,
@@ -196,10 +169,10 @@ def get_vrf_description(vrf, module):
 
 
 def get_value(arg, config, module):
-    REGEX = re.compile(r'(?:{0}\s)(?P<value>.*)$'.format(arg), re.M)
+    extra_arg_regex = re.compile(r'(?:{0}\s)(?P<value>.*)$'.format(arg), re.M)
     value = ''
     if arg in config:
-        value = REGEX.search(config).group('value')
+        value = extra_arg_regex.search(config).group('value')
     return value
 
 
@@ -210,19 +183,19 @@ def get_vrf(vrf, module):
         'vrf_state': 'admin_state'
         }
 
-    body = execute_show_command(command, module)
     try:
-        vrf_table = body[0]['TABLE_vrf']['ROW_vrf']
+        body = execute_show_command(command, module)[0]
+        vrf_table = body['TABLE_vrf']['ROW_vrf']
     except (TypeError, IndexError):
         return {}
 
     parsed_vrf = apply_key_map(vrf_key, vrf_table)
 
     command = 'show run all | section vrf.context.{0}'.format(vrf)
-    body = execute_show_command(command, module)
+    body = execute_show_command(command, module)[0]
     extra_params = ['vni', 'rd', 'description']
     for param in extra_params:
-        parsed_vrf[param] = get_value(param, body[0], module)
+        parsed_vrf[param] = get_value(param, body, module)
 
     return parsed_vrf
 
@@ -233,10 +206,8 @@ def main():
         description=dict(default=None, required=False),
         vni=dict(required=False, type='str'),
         rd=dict(required=False, type='str'),
-        admin_state=dict(default='up', choices=['up', 'down'],
+        admin_state=dict(default='up', choices=['up', 'down'], required=False),
-                             required=False),
+        state=dict(default='present', choices=['present', 'absent'], required=False),
-        state=dict(default='present', choices=['present', 'absent'],
-                       required=False),
         include_defaults=dict(default=False),
         config=dict(),
         save=dict(type='bool', default=False)
@@ -244,11 +215,11 @@ def main():
 
     argument_spec.update(nxos_argument_spec)
 
-    module = AnsibleModule(argument_spec=argument_spec,
+    module = AnsibleModule(argument_spec=argument_spec, supports_check_mode=True)
-                                supports_check_mode=True)
 
     warnings = list()
     check_args(module, warnings)
+    results = dict(changed=False, warnings=warnings)
 
 
     vrf = module.params['vrf']
@@ -261,27 +232,15 @@ def main():
     if vrf == 'default':
         module.fail_json(msg='cannot use default as name of a VRF')
     elif len(vrf) > 32:
-        module.fail_json(msg='VRF name exceeded max length of 32',
+        module.fail_json(msg='VRF name exceeded max length of 32', vrf=vrf)
-                         vrf=vrf)
 
     existing = get_vrf(vrf, module)
     args = dict(vrf=vrf, description=description, vni=vni,
                 admin_state=admin_state, rd=rd)
 
-    end_state = existing
-    changed = False
     proposed = dict((k, v) for k, v in args.items() if v is not None)
 
-    """Since 'admin_state' is either 'Up' or 'Down' from outputs,
-    we use the following to make sure right letter case is used so that delta
-    results will be consistent to the actual configuration."""
-    if existing:
-        if existing['admin_state'].lower() == admin_state:
-            proposed['admin_state'] = existing['admin_state']
-
     delta = dict(set(proposed.items()).difference(existing.items()))
-    changed = False
-    end_state = existing
     commands = []
     if state == 'absent':
         if existing:
@@ -304,22 +263,14 @@ def main():
             module.exit_json(changed=True, commands=commands)
         else:
             load_config(module, commands)
-            changed = True
+            results['changed'] = True
-            end_state = get_vrf(vrf, module)
             if 'configure' in commands:
                 commands.pop(0)
 
-    results = {}
+    results['commands'] = commands
-    results['proposed'] = proposed
-    results['existing'] = existing
-    results['end_state'] = end_state
-    results['updates'] = commands
-    results['changed'] = changed
-    results['warnings'] = warnings
 
     module.exit_json(**results)
 
 
 if __name__ == '__main__':
     main()
-

test/units/modules/network/nxos/test_nxos_vrf.py

@@ -0,0 +1,64 @@
+# (c) 2016 Red Hat Inc.
+#
+# 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/>.
+
+# Make coding more python3-ish
+from __future__ import (absolute_import, division, print_function)
+__metaclass__ = type
+
+import json
+
+from ansible.compat.tests.mock import patch
+from ansible.modules.network.nxos import nxos_vrf
+from .nxos_module import TestNxosModule, load_fixture, set_module_args
+
+
+class TestNxosVrfModule(TestNxosModule):
+
+    module = nxos_vrf
+
+    def setUp(self):
+        self.mock_run_commands = patch('ansible.modules.network.nxos.nxos_vrf.run_commands')
+        self.run_commands = self.mock_run_commands.start()
+
+        self.mock_load_config = patch('ansible.modules.network.nxos.nxos_vrf.load_config')
+        self.load_config = self.mock_load_config.start()
+
+        self.mock_get_config = patch('ansible.modules.network.nxos.nxos_vrf.get_config')
+        self.get_config = self.mock_get_config.start()
+
+    def tearDown(self):
+        self.mock_run_commands.stop()
+        self.mock_load_config.stop()
+        self.mock_get_config.stop()
+
+    def load_fixtures(self, commands=None):
+        self.load_config.return_value = None
+
+    def test_nxos_vrf_present(self):
+        set_module_args(dict(vrf='ntc', state='present', admin_state='up'))
+        result = self.execute_module(changed=True)
+        self.assertEqual(result['commands'], ['vrf context ntc', 'no shutdown'])
+
+    def test_nxos_vrf_absent(self):
+        set_module_args(dict(vrf='ntc', state='absent'))
+        result = self.execute_module(changed=True)
+        self.assertEqual(result['commands'], ['no vrf context ntc'])
+
+    def test_nxos_vrf_default(self):
+        set_module_args(dict(vrf='default'))
+        result = self.execute_module(failed=True)
+        self.assertEqual(result['msg'], 'cannot use default as name of a VRF')