nxos_feature fix and unit test (#25200)

* nxos_feature fix

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

* nxos_feature unit test

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

* remove from pep8/legacy-file

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

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

@@ -16,10 +16,11 @@
 # along with Ansible.  If not, see <http://www.gnu.org/licenses/>.
 #
 
-ANSIBLE_METADATA = {'metadata_version': '1.0',
-                    'status': ['preview'],
-                    'supported_by': 'community'}
-
+ANSIBLE_METADATA = {
+    'metadata_version': '1.0',
+    'status': ['preview'],
+    'supported_by': 'community'
+}
 
 DOCUMENTATION = '''
 ---
@@ -28,21 +29,21 @@ extends_documentation_fragment: nxos
 version_added: "2.1"
 short_description: Manage features in NX-OS switches.
 description:
-    - Offers ability to enable and disable features in NX-OS.
+  - Offers ability to enable and disable features in NX-OS.
 author:
-    - Jason Edelman (@jedelman8)
-    - Gabriele Gerbino (@GGabriele)
+  - Jason Edelman (@jedelman8)
+  - Gabriele Gerbino (@GGabriele)
 options:
-    feature:
-        description:
-            - Name of feature.
-        required: true
-    state:
-        description:
-            - Desired state of the feature.
-        required: false
-        default: 'enabled'
-        choices: ['enabled','disabled']
+  feature:
+    description:
+      - Name of feature.
+    required: true
+  state:
+    description:
+      - Desired state of the feature.
+    required: false
+    default: 'enabled'
+    choices: ['enabled','disabled']
 '''
 
 EXAMPLES = '''
@@ -50,20 +51,16 @@ EXAMPLES = '''
   nxos_feature:
     feature: lacp
     state: enabled
-    host: "{{ inventory_hostname }}"
 
 - name: Ensure ospf is disabled
   nxos_feature:
     feature: ospf
     state: disabled
-    host: "{{ inventory_hostname }}"
 
 - name: Ensure vpc is enabled
   nxos_feature:
     feature: vpc
     state: enabled
-    host: "{{ inventory_hostname }}"
-
 '''
 
 RETURN = '''
@@ -73,6 +70,7 @@ commands:
     type: list
     sample: ['nv overlay evpn']
 '''
+
 import re
 
 from ansible.module_utils.basic import AnsibleModule
@@ -80,6 +78,7 @@ from ansible.module_utils.nxos import load_config, run_commands
 from ansible.module_utils.nxos import nxos_argument_spec
 from ansible.module_utils.nxos import check_args as nxos_check_args
 
+
 def check_args(module, warnings):
     nxos_check_args(module, warnings)
 
@@ -91,12 +90,13 @@ def check_args(module, warnings):
 def get_available_features(feature, module):
     available_features = {}
     feature_regex = '(?P<feature>\S+)\s+\d+\s+(?P<state>.*)'
-    command = 'show feature'
+    command = {'command': 'show feature', 'output': 'text'}
 
-    command = {'command': command, 'output': 'text'}
-
-    body = run_commands(module, [command])
-    split_body = body[0].splitlines()
+    try:
+        body = run_commands(module, [command])[0]
+        split_body = body.splitlines()
+    except (KeyError, IndexError):
+        return {}
 
     for line in split_body:
         try:
@@ -115,14 +115,12 @@ def get_available_features(feature, module):
             if feature not in available_features:
                 available_features[feature] = state
             else:
-                if (available_features[feature] == 'disabled' and
-                        state == 'enabled'):
+                if available_features[feature] == 'disabled' and state == 'enabled':
                     available_features[feature] = state
 
     return available_features
 
 
-
 def get_commands(proposed, existing, state, module):
     feature = validate_feature(module, mode='config')
     commands = []
@@ -159,7 +157,7 @@ def validate_feature(module, mode='show'):
             'telnet': 'telnetServer',
             'ethernet-link-oam': 'elo',
             'port-security': 'eth_port_sec'
-            },
+        },
         'config': {
             'nve': 'nv overlay',
             'vnseg_vlan': 'vn-segment-vlan-based',
@@ -174,8 +172,8 @@ def validate_feature(module, mode='show'):
             'telnetServer': 'telnet',
             'elo': 'ethernet-link-oam',
             'eth_port_sec': 'port-security'
-            }
         }
+    }
 
     if feature in feature_to_be_mapped[mode]:
         feature = feature_to_be_mapped[mode][feature]
@@ -196,11 +194,11 @@ def main():
 
     argument_spec.update(nxos_argument_spec)
 
-    module = AnsibleModule(argument_spec=argument_spec,
-                                supports_check_mode=True)
+    module = AnsibleModule(argument_spec=argument_spec, supports_check_mode=True)
 
     warnings = list()
     check_args(module, warnings)
+    results = dict(changed=False, warnings=warnings)
 
     feature = validate_feature(module)
     state = module.params['state'].lower()
@@ -216,25 +214,18 @@ def main():
 
         existing = dict(state=existstate)
         proposed = dict(state=state)
-        changed = False
-        end_state = existing
+        results['changed'] = False
 
         cmds = get_commands(proposed, existing, state, module)
 
         if cmds:
             if not module.check_mode:
                 load_config(module, cmds)
-            changed = True
-
-    results = {
-        'commands': cmds,
-        'changed': changed,
-        'warnings': warnings
-    }
+            results['changed'] = True
 
+    results['commands'] = cmds
     module.exit_json(**results)
 
 
 if __name__ == '__main__':
     main()
-