def test_basic_cfg(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxe')
dev2 = Device(testbed=testbed, name='PE2', os='iosxe')
intf1 = Interface(name='GigabitEthernet0/0/1',device=dev1)
intf2 = Interface(name='GigabitEthernet0/0/2',device=dev2)
intf3 = Interface(name='Vlan100',device=dev1)
link = Link(name='1_2_1',testbed=testbed)
link.connect_interface(interface=intf1)
link.connect_interface(interface=intf2)
vlan = Vlan()
link.add_feature(vlan)
vlan.vlan_id = 100
access_map_id = 'ed'
vlan_configuration_id = '3'
vlan.device_attr[dev1]
vlan.device_attr[dev2]
vlan.device_attr[dev1].interface_attr[intf1].switchport_mode = \
'trunk'
vlan.device_attr[dev1].interface_attr[intf1].sw_trunk_allowed_vlan = \
'200-201'
vlan.device_attr[dev1].access_map_attr[access_map_id]
vlan.device_attr[dev2].access_map_attr[access_map_id]
vlan.device_attr[dev1].vlan_configuration_attr[vlan_configuration_id]
vlan.device_attr[dev2].vlan_configuration_attr[vlan_configuration_id]
# Defining attributes section
vlan.shutdown = False
with self.assertRaises(ValueError):
vlan.media = 'invalid'
vlan.media = 'enet'
self.assertIs(type(vlan.media), Vlan.Media)
vlan.accounting_type = 'input'
vlan.device_attr[dev1].access_map_action = 'drop'
vlan.datalink_flow_monitor = True
vlan.private_vlan_type = 'association'
vlan.private_vlan_association_action = 'add'
vlan.private_vlan_association_ids = '201,205'
cfg1 = vlan.build_config(apply=False)
self.assertCountEqual(cfg1.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(cfg1['PE1']),
'\n'.join([
'vlan 100',
' media enet',
' no shutdown',
' private-vlan association add 201,205',
' exit',
'vlan accounting input',
'vlan access-map ed',
' action drop',
' exit',
'vlan configuration 3',
' datalink flow monitor',
' exit',
'interface GigabitEthernet0/0/1',
' switchport mode trunk',
' switchport trunk allowed vlan 200-201',
' exit',
]))
self.assertMultiLineEqual(
str(cfg1['PE2']),
'\n'.join([
'vlan 100',
' media enet',
' no shutdown',
' private-vlan association add 201,205',
' exit',
'vlan accounting input',
'vlan access-map ed',
' exit',
'vlan configuration 3',
' datalink flow monitor',
' exit',
]))
def test_basic_uncfg(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxe')
dev2 = Device(testbed=testbed, name='PE2', os='iosxe')
intf1 = Interface(name='GigabitEthernet0/0/1',device=dev1)
intf1.switchport = True
intf2 = Interface(name='GigabitEthernet0/0/2',device=dev2)
intf3 = Interface(name='Vlan100',device=dev1)
vlan = Vlan()
vlan.device_attr[dev1].vlan_id = 100
vlan.device_attr[dev2].vlan_id = 300
access_map_id = 'ed'
vlan_configuration_id = '3'
vlan.device_attr[dev1]
vlan.device_attr[dev2]
vlan.device_attr[dev1].interface_attr[intf1].switchport_mode = \
'trunk'
vlan.device_attr[dev1].interface_attr[intf1].sw_trunk_allowed_vlan = \
'200-201'
vlan.device_attr[dev1].access_map_attr[access_map_id]
vlan.device_attr[dev2].access_map_attr[access_map_id]
vlan.device_attr[dev1].vlan_configuration_attr[vlan_configuration_id]
vlan.device_attr[dev2].vlan_configuration_attr[vlan_configuration_id]
# Defining attributes section
with self.assertRaises(ValueError):
vlan.media = 'invalid'
vlan.media = 'enet'
self.assertIs(type(vlan.media), Vlan.Media)
vlan.accounting_type = 'input'
vlan.device_attr[dev1].access_map_action = 'drop'
# Unconfig testing
# Set a mock
dev1.configure = Mock()
dev2.configure = Mock()
dev1.add_feature(vlan)
dev2.add_feature(vlan)
# Mock config
uncfg1 = vlan.build_unconfig(apply=False)
self.assertCountEqual(uncfg1.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(uncfg1['PE1']),
'\n'.join([
'no vlan 100',
'no vlan accounting input',
'no vlan access-map ed',
'no vlan configuration 3',
'interface GigabitEthernet0/0/1',
' no switchport mode trunk',
' no switchport trunk allowed vlan 200-201',
' exit',
]))
self.assertMultiLineEqual(
str(uncfg1['PE2']),
'\n'.join([
'no vlan 300',
'no vlan accounting input',
'no vlan access-map ed',
'no vlan configuration 3',
]))
uncfg_intf = intf1.build_unconfig(apply=False, attributes="switchport")
self.assertMultiLineEqual(
str(uncfg_intf),
'\n'.join([
'interface GigabitEthernet0/0/1',
' no switchport',
' exit',
]))
partial_uncfg1 = vlan.build_unconfig(apply=False,
attributes={'device_attr':\
{'*':"media"}})
self.assertCountEqual(partial_uncfg1.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(partial_uncfg1['PE1']),
'\n'.join([
'vlan 100',
' no media enet',
' exit',
]))
partial_unconfigure = vlan.build_unconfig(apply=False,
attributes={'device_attr':\
{'*':{'access_map_attr':\
{'*':"access_map_action"}}}})
self.assertCountEqual(partial_unconfigure.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(partial_unconfigure['PE1']),
'\n'.join([
'vlan access-map ed',
' no action drop',
' exit',
]))
all_vlan_interface_uncfg = vlan.build_unconfig(apply=False,
attributes={'device_attr':\
{'*':{'interface_attr':'*'}}})
self.assertCountEqual(all_vlan_interface_uncfg.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(all_vlan_interface_uncfg['PE1']),
'\n'.join([
'interface GigabitEthernet0/0/1',
' no switchport mode trunk',
' no switchport trunk allowed vlan 200-201',
' exit',
]))
partial_vlan_interface_uncfg = vlan.build_unconfig(apply=False,
attributes={'device_attr':\
{'*':{'interface_attr':\
{'*':"sw_trunk_allowed_vlan"}}}})
self.assertCountEqual(partial_vlan_interface_uncfg.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(partial_vlan_interface_uncfg['PE1']),
'\n'.join([
'interface GigabitEthernet0/0/1',
' no switchport trunk allowed vlan 200-201',
' exit',
]))