ansible/test/units/modules/network/onyx/test_onyx_vxlan.py
anasbadaha 2cc91e26e0 Adding Support For Vxlan In Onyx Switches (#55081)
* Adding Support For Vxlan In Onyx Switches

Signed-off-by: Anas Badaha <anasb@mellanox.com>

* Fix Pep8 Failures in onyx_vxlan.py

Signed-off-by: Anas Badaha <anasb@mellanox.com>

* Fix Pep8 Failures in onyx_vxlan phase 2

Signed-off-by: Anas Badaha <anasb@mellanox.com>

* Fix Shippable failures

Signed-off-by: Anas Badaha <anasb@mellanox.com>

* Fix Samer's Comments on PR

Signed-off-by: Anas Badaha <anasb@mellanox.com>
2019-04-11 20:59:11 +05:30

101 lines
5.1 KiB
Python

#
# 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_vxlan
from units.modules.utils import set_module_args
from .onyx_module import TestOnyxModule, load_fixture
class TestOnyxVxlanModule(TestOnyxModule):
module = onyx_vxlan
arp_suppression = True
def setUp(self):
super(TestOnyxVxlanModule, self).setUp()
self.mock_get_vxlan_config = patch.object(
onyx_vxlan.OnyxVxlanModule, "_show_vxlan_config")
self.get_vxlan_config = self.mock_get_vxlan_config.start()
self.mock_load_config = patch(
'ansible.module_utils.network.onyx.onyx.load_config')
self.load_config = self.mock_load_config.start()
self.mock_get_nve_detail = patch.object(
onyx_vxlan.OnyxVxlanModule, "_show_nve_detail")
self.get_nve_detail = self.mock_get_nve_detail.start()
def tearDown(self):
super(TestOnyxVxlanModule, self).tearDown()
self.mock_get_vxlan_config.stop()
self.mock_load_config.stop()
self.mock_get_nve_detail.stop()
def load_fixtures(self, commands=None, transport='cli'):
interfaces_nve_config_file = 'onyx_show_interfaces_nve.cfg'
interfaces_nve_detail_config_file = 'onyx_show_interfaces_nve_detail.cfg'
self.get_nve_detail.return_value = None
interfaces_nve_detail_data = load_fixture(interfaces_nve_detail_config_file)
interfaces_nv_data = load_fixture(interfaces_nve_config_file)
self.get_nve_detail.return_value = interfaces_nve_detail_data
if self.arp_suppression is False:
interfaces_nve_detail_data[0]["10"][0]["Neigh Suppression"] = "Disable"
interfaces_nve_detail_data[0]["6"][0]["Neigh Suppression"] = "Disable"
self.get_nve_detail.return_value = interfaces_nve_detail_data
self.get_vxlan_config.return_value = interfaces_nv_data
self.load_config.return_value = None
def test_configure_vxlan_no_change(self):
set_module_args(dict(nve_id=1, loopback_id=1, bgp=True, mlag_tunnel_ip='192.10.10.1',
vni_vlan_list=[dict(vlan_id=10, vni_id=10010), dict(vlan_id=6, vni_id=10060)],
arp_suppression=True))
self.execute_module(changed=False)
def test_configure_vxlan_with_change(self):
set_module_args(dict(nve_id=2, loopback_id=1, bgp=True, mlag_tunnel_ip='192.10.10.1',
vni_vlan_list=[dict(vlan_id=10, vni_id=10010), dict(vlan_id=6, vni_id=10060)],
arp_suppression=True))
commands = [
"no interface nve 1", "interface nve 2", "exit",
"interface nve 2 vxlan source interface loopback 1 ",
"interface nve 2 nve controller bgp", "interface nve 2 vxlan mlag-tunnel-ip 192.10.10.1",
"interface nve 2 nve neigh-suppression", "interface nve 2 nve vni 10010 vlan 10",
"interface vlan 10", "exit", "interface nve 2 nve vni 10060 vlan 6", "interface vlan 6", "exit"
]
self.execute_module(changed=True, commands=commands)
def test_loopback_id_with_change(self):
set_module_args(dict(nve_id=1, loopback_id=2, bgp=True, mlag_tunnel_ip='192.10.10.1',
vni_vlan_list=[dict(vlan_id=10, vni_id=10010), dict(vlan_id=6, vni_id=10060)],
arp_suppression=True))
commands = ["interface nve 1 vxlan source interface loopback 2 "]
self.execute_module(changed=True, commands=commands)
def test_mlag_tunnel_ip_with_change(self):
set_module_args(dict(nve_id=1, loopback_id=1, bgp=True, mlag_tunnel_ip='192.10.10.10',
vni_vlan_list=[dict(vlan_id=10, vni_id=10010), dict(vlan_id=6, vni_id=10060)],
arp_suppression=True))
commands = ["interface nve 1 vxlan mlag-tunnel-ip 192.10.10.10"]
self.execute_module(changed=True, commands=commands)
def test_vni_vlan_list_with_change(self):
set_module_args(dict(nve_id=1, loopback_id=1, bgp=True, mlag_tunnel_ip='192.10.10.1',
vni_vlan_list=[dict(vlan_id=11, vni_id=10011), dict(vlan_id=7, vni_id=10061)],
arp_suppression=False))
commands = ["interface nve 1 nve vni 10011 vlan 11", "interface nve 1 nve vni 10061 vlan 7"]
self.execute_module(changed=True, commands=commands)
def test_arp_suppression_with_change(self):
self.arp_suppression = False
set_module_args(dict(nve_id=1, loopback_id=1, bgp=True, mlag_tunnel_ip='192.10.10.1',
vni_vlan_list=[dict(vlan_id=10, vni_id=10010), dict(vlan_id=6, vni_id=10060)],
arp_suppression=True))
commands = ["interface vlan 10", "exit", "interface vlan 6", "exit"]
self.execute_module(changed=True, commands=commands)