summaryrefslogtreecommitdiff
path: root/test/units/modules/network/nxos/test_nxos_vlan.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/units/modules/network/nxos/test_nxos_vlan.py')
-rw-r--r--test/units/modules/network/nxos/test_nxos_vlan.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/test/units/modules/network/nxos/test_nxos_vlan.py b/test/units/modules/network/nxos/test_nxos_vlan.py
index 1f0b546b68..eee544e8b9 100644
--- a/test/units/modules/network/nxos/test_nxos_vlan.py
+++ b/test/units/modules/network/nxos/test_nxos_vlan.py
@@ -42,11 +42,16 @@ class TestNxosVlanModule(TestNxosModule):
self.mock_get_config = patch('ansible.modules.network.nxos.nxos_vlan.get_config')
self.get_config = self.mock_get_config.start()
+ self.mock_get_capabilities = patch('ansible.modules.network.nxos.nxos_vlan.get_capabilities')
+ self.get_capabilities = self.mock_get_capabilities.start()
+ self.get_capabilities.return_value = {'device_info': {'network_os_platform': 'N9K-9000v'}, 'network_api': 'cliconf'}
+
def tearDown(self):
super(TestNxosVlanModule, self).tearDown()
self.mock_run_commands.stop()
self.mock_load_config.stop()
self.mock_get_config.stop()
+ self.mock_get_capabilities.stop()
def load_fixtures(self, commands=None, device=''):
def load_from_file(*args, **kwargs):