def setUp(self):
pass
testbed = Genie.testbed = Testbed()
self.dev1 = Device(testbed=testbed, name='PE1', os='iosxe')
self.dev2 = Device(testbed=testbed, name='PE2', os='iosxe')
self.intf1 = Interface(name='GigabitEthernet0/0/1', device=self.dev1)
self.intf2 = Interface(name='GigabitEthernet0/0/2', device=self.dev2)
self.link = Link(name='1_2_1', testbed=testbed)
self.link.connect_interface(interface=self.intf1)
self.link.connect_interface(interface=self.intf2)
self.assertSetEqual(set(self.link.find_interfaces()),
set([self.intf1, self.intf2]))
self.assertSetEqual(set(self.dev1.find_interfaces()),
set([self.intf1]))
self.assertSetEqual(set(self.dev2.find_interfaces()),
set([self.intf2]))
isis = Isis("core")
self.assertSetEqual(set(isis.devices), set([]))
self.assertSetEqual(set(isis.links), set([]))
self.link.add_feature(isis)
for intf_iter in self.link.interfaces:
intf_iter.add_feature(isis)
self.assertCountEqual(isis.devices, [self.dev1, self.dev2])
self.assertSetEqual(set(isis.links), set([self.link]))
self.assertSetEqual(set(isis.interfaces), set([self.intf1,
self.intf2]))
def test_init(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxe', context='yang')
dev2 = Device(testbed=testbed, name='PE2', os='iosxe', context='yang')
intf1 = Interface(name='GigabitEthernet0/0/1', device=dev1)
intf2 = Interface(name='GigabitEthernet0/0/2', device=dev2)
link = Link(name='1_2_1', testbed=testbed)
link.connect_interface(interface=intf1)
link.connect_interface(interface=intf2)
self.assertSetEqual(set(link.find_interfaces()), set([intf1, intf2]))
self.assertSetEqual(set(dev1.find_interfaces()), set([intf1]))
self.assertSetEqual(set(dev2.find_interfaces()), set([intf2]))
vrf = Vrf(name='test')
self.assertSetEqual(set(vrf.devices), set([]))
self.assertSetEqual(set(vrf.interfaces), set([]))
intf1.vrf = vrf
intf2.vrf = vrf
dev1.add_feature(vrf)
dev2.add_feature(vrf)
self.assertCountEqual(vrf.devices, [dev1, dev2])
self.assertSetEqual(set(vrf.interfaces), set([intf1, intf2]))
with self.assertRaises(AttributeError):
vrf.address_families = set([AddressFamily.ipv8_unicast])
def test_init_vpls(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxe')
lo1 = Interface(device=dev1, name='Loopback0', ipv4='101.0.0.1/32')
intf1 = Interface(device=dev1,
name='GigabitEthernet0/0/1',
ipv4='10.1.0.1/24')
intf2 = Interface(device=dev1,
name='GigabitEthernet0/0/2',
ipv4='10.2.0.1/24')
dev2 = Device(testbed=testbed, name='PE2', os='iosxe')
lo2 = Interface(device=dev2, name='Loopback0', ipv4='102.0.0.1/32')
intf3 = Interface(device=dev2,
name='GigabitEthernet0/0/3',
ipv4='10.1.0.2/24')
intf4 = Interface(device=dev2,
name='GigabitEthernet0/0/4',
ipv4='10.2.0.2/24')
link1 = Link(testbed=testbed, name='link1', interfaces=(intf1, intf3))
link2 = Link(testbed=testbed, name='link2', interfaces=(intf2, intf4))
dev3 = Device(testbed=testbed, name='PE3', os='iosxe')
bd1 = BridgeDomain(name='200')
dev1.add_feature(bd1)
dev2.add_feature(bd1)
self.assertCountEqual(bd1.pseudowires, [])
nbr1 = bd1.create_pseudowire_neighbor(device=dev1, ip=lo2.ipv4.ip)
nbr2 = bd1.device_attr[dev2].create_pseudowire_neighbor(ip=lo1.ipv4.ip)
pw1 = Pseudowire(neighbors=(nbr1, nbr2), pw_id=1)
self.assertCountEqual(bd1.pseudowires, [pw1])
self.assertCountEqual(bd1.pseudowire_neighbors, [nbr1, nbr2])
self.assertCountEqual(bd1.device_attr[dev1].pseudowires, [pw1])
self.assertCountEqual(bd1.device_attr[dev1].pseudowire_neighbors,
[nbr1])
self.assertCountEqual(bd1.device_attr[dev2].pseudowires, [pw1])
self.assertCountEqual(bd1.device_attr[dev2].pseudowire_neighbors,
[nbr2])
self.assertCountEqual(bd1.device_attr[dev3].pseudowires, [])
self.assertCountEqual(bd1.device_attr[dev3].pseudowire_neighbors, [])
self.assertCountEqual(bd1.segments, [pw1])
self.assertCountEqual(bd1.device_attr[dev1].segments, [pw1])
self.assertCountEqual(bd1.device_attr[dev2].segments, [pw1])
self.assertCountEqual(bd1.device_attr[dev3].segments, [])
cfgs = bd1.build_config(apply=False)
self.assertMultiLineDictEqual(
cfgs, {
dev1.name: '\n'.join([
'bridge-domain 200',
' exit',
]),
dev2.name: '\n'.join([
'bridge-domain 200',
' exit',
]),
})
def test_init_vpws(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxr')
lo1 = Interface(device=dev1, name='Loopback0', ipv4='101.0.0.1/32')
intf1 = Interface(device=dev1, name='GigabitEthernet0/0/0/1', ipv4='10.1.0.1/24')
intf2 = Interface(device=dev1, name='GigabitEthernet0/0/0/2', ipv4='10.2.0.1/24')
dev2 = Device(testbed=testbed, name='PE2', os='iosxr')
lo2 = Interface(device=dev2, name='Loopback0', ipv4='102.0.0.1/32')
intf3 = Interface(device=dev2, name='GigabitEthernet0/0/0/3', ipv4='10.1.0.2/24')
intf4 = Interface(device=dev2, name='GigabitEthernet0/0/0/4', ipv4='10.2.0.2/24')
link1 = Link(testbed=testbed, name='link1', interfaces=(intf1, intf3))
link2 = Link(testbed=testbed, name='link2', interfaces=(intf2, intf4))
dev3 = Device(testbed=testbed, name='PE3', os='iosxr')
xc1 = Xconnect(name='xc1')
dev1.add_feature(xc1)
dev2.add_feature(xc1)
self.assertCountEqual(xc1.pseudowires, [])
nbr1 = xc1.create_pseudowire_neighbor(device=dev1, ip=lo2.ipv4.ip)
nbr2 = xc1.device_attr[dev2].create_pseudowire_neighbor(ip=lo1.ipv4.ip)
pw1 = Pseudowire(neighbors=(nbr1, nbr2), pw_id=1)
self.assertCountEqual(xc1.pseudowires, [pw1])
self.assertCountEqual(xc1.device_attr[dev1].pseudowires, [pw1])
self.assertCountEqual(xc1.device_attr[dev2].pseudowires, [pw1])
self.assertCountEqual(xc1.device_attr[dev3].pseudowires, [])
self.assertCountEqual(xc1.segments, [pw1])
self.assertCountEqual(xc1.device_attr[dev1].segments, [pw1])
self.assertCountEqual(xc1.device_attr[dev2].segments, [pw1])
self.assertCountEqual(xc1.device_attr[dev3].segments, [])
cfgs = xc1.build_config(apply=False)
self.assertMultiLineDictEqual(
cfgs,
{
dev1.name: '\n'.join([
'l2vpn',
' xconnect group xc1g',
' p2p xc1',
' neighbor ipv4 102.0.0.1 pw-id 1',
' exit',
' exit',
' exit',
' exit',
]),
dev2.name: '\n'.join([
'l2vpn',
' xconnect group xc1g',
' p2p xc1',
' neighbor ipv4 101.0.0.1 pw-id 1',
' exit',
' exit',
' exit',
' exit',
]),
})
def _write_connections_to_testbed(self, connection_dict, testbed):
'''Writes the connections found in the connection_dict into the testbed
Args:
connection_dict ('dict'): Dictionary with connections found earlier
testbed ('testbed'): testbed to write connections into
'''
# filter to strip out the numbers from an interface to create a type name
# example: ethernet0/3 becomes ethernet
interface_filter = re.compile(r'[a-zA-Z]+')
log.debug('Adding connections to testbed')
for device in connection_dict:
log.debug(' Writing connections found in {}'.format(device))
for interface_name in connection_dict[device]:
#if connecting interface is not in the testbed, create the interface
if interface_name not in testbed.devices[device].interfaces:
type_name = interface_filter.match(interface_name)
interface = Interface(interface_name,
type=type_name[0].lower())
interface.device = testbed.devices[device]
else:
# get the interface found in the connection on the device searched
interface = testbed.devices[device].interfaces[
interface_name]
# if the interface is not already part of a link get a list of
# all interfaces involved in the link and create a new link
# object with the associated interfaces
if interface.link is None:
int_list = [interface]
for entry in connection_dict[device][interface_name]:
dev = entry['dest_host']
dest_int = entry['dest_port']
if testbed.devices[dev].interfaces[
dest_int] not in int_list:
int_list.append(
testbed.devices[dev].interfaces[dest_int])
if len(int_list) > 1:
link = Link(
'Link_{num}'.format(num=len(testbed.links)),
interfaces=int_list)
# if the interface is already part of the link go over the
# other interfaces found in the connection_dict and add them to the link
# if they are not already there
else:
link = interface.link
for entry in connection_dict[device][interface_name]:
dev = entry['dest_host']
dest_int = entry['dest_port']
if testbed.devices[dev].interfaces[
dest_int] not in link.interfaces:
link.connect_interface(
testbed.devices[dev].interfaces[dest_int])
def test_init(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxr')
lo1 = Interface(device=dev1, name='Loopback0', ipv4='101.0.0.1/32')
intf1 = Interface(device=dev1, name='GigabitEthernet0/0/0/1', ipv4='10.1.0.1/24')
intf2 = Interface(device=dev1, name='GigabitEthernet0/0/0/2', ipv4='10.2.0.1/24')
dev2 = Device(testbed=testbed, name='PE2', os='iosxr')
lo2 = Interface(device=dev2, name='Loopback0', ipv4='102.0.0.1/32')
intf3 = Interface(device=dev2, name='GigabitEthernet0/0/0/3', ipv4='10.1.0.2/24')
intf4 = Interface(device=dev2, name='GigabitEthernet0/0/0/4', ipv4='10.2.0.2/24')
link1 = Link(testbed=testbed, name='link1', interfaces=(intf1, intf3))
link2 = Link(testbed=testbed, name='link2', interfaces=(intf2, intf4))
dev3 = Device(testbed=testbed, name='PE3', os='iosxr')
container = Mock()
container.add_pseudowire = MagicMock()
nbr1 = PseudowireNeighbor(container=container, device=dev1, ip='1.2.3.4')
nbr2 = PseudowireNeighbor(container=container, device=dev2, ip='1.2.3.4')
nbr3 = PseudowireNeighbor(container=container, device=dev3, ip='1.2.3.4')
with self.assertRaises(TypeError):
pw1 = Pseudowire()
with self.assertRaises(TypeError):
pw1 = Pseudowire(pw_id=1)
with self.assertRaises(ValueError):
pw1 = Pseudowire(neighbors=())
with self.assertRaises(ValueError):
pw1 = Pseudowire(neighbors=(nbr1,))
with self.assertRaises(ValueError):
pw1 = Pseudowire(neighbors=(nbr1, nbr1))
with self.assertRaises(ValueError):
pw1 = Pseudowire(neighbors=(nbr1, nbr2, nbr3))
with self.assertRaises(ValueError):
pw1 = Pseudowire(neighbors=(nbr1, intf2))
pw1 = Pseudowire(neighbors=(nbr1, nbr2))
self.assertCountEqual(pw1.neighbors, [nbr1, nbr2])
self.assertEqual(nbr1.pw_id, None)
self.assertEqual(nbr2.pw_id, None)
container.add_pseudowire.assert_has_calls([
unittest.mock.call(pw1),
unittest.mock.call(pw1),
])
container.add_pseudowire.reset_mock()
pw1 = Pseudowire(neighbors=(nbr1, nbr2), pw_id=1)
self.assertCountEqual(pw1.neighbors, [nbr1, nbr2])
self.assertEqual(nbr1.pw_id, 1)
self.assertEqual(nbr2.pw_id, 1)
container.add_pseudowire.assert_has_calls([
unittest.mock.call(pw1),
unittest.mock.call(pw1),
])
container.add_pseudowire.reset_mock()
def setUp(self):
testbed = Genie.testbed = Testbed()
self.dev1 = Device(testbed=testbed, name='PE1', os='iosxr')
self.dev2 = Device(testbed=testbed, name='PE2', os='iosxr')
self.intf1 = Interface(name='GigabitEthernet0/0/0/1', device=self.dev1)
self.intf2 = Interface(name='GigabitEthernet0/0/0/2', device=self.dev2)
self.link = Link(name='1_2_1', testbed=testbed)
self.link.connect_interface(interface=self.intf1)
self.link.connect_interface(interface=self.intf2)
def setUp(self):
tb = Genie.testbed = Testbed()
self.dev1 = Device(testbed=tb, name='PE1', os='iosxe')
self.dev2 = Device(testbed=tb, name='PE2', os='iosxe')
self.i1 = Interface(name='GigabitEthernet0/0/1', device=self.dev1)
self.i2 = Interface(name='GigabitEthernet0/0/2', device=self.dev2)
self.i3 = Interface(name='GigabitEthernet0/0/3', device=self.dev1)
self.i4 = Interface(name='GigabitEthernet0/0/4', device=self.dev2)
self.i5 = Interface(name='GigabitEthernet0/0/5', device=self.dev1)
self.i6 = Interface(name='GigabitEthernet0/0/6', device=self.dev2)
self.i7 = Interface(name='GigabitEthernet0/0/7', device=self.dev1)
self.i8 = Interface(name='GigabitEthernet0/0/8', device=self.dev2)
self.link = Link(name='1_2_1')
self.link.connect_interface(interface=self.i1)
self.link.connect_interface(interface=self.i2)
self.link2 = Link(name='1_2_2')
self.link2.connect_interface(interface=self.i3)
self.link2.connect_interface(interface=self.i4)
self.link3 = Link(name='1_2_3')
self.link3.connect_interface(interface=self.i5)
self.link3.connect_interface(interface=self.i6)
self.link4 = Link(name='1_2_4')
self.link4.connect_interface(interface=self.i7)
self.link4.connect_interface(interface=self.i8)
self.assertCountEqual(self.link.find_interfaces(), [self.i1, self.i2])
self.assertCountEqual(self.dev1.find_interfaces(),
[self.i1, self.i3, self.i5, self.i7])
self.assertCountEqual(self.dev2.find_interfaces(),
[self.i2, self.i4, self.i6, self.i8])
def test_init(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)
self.assertSetEqual(
set(link.find_interfaces()),
set([intf1, intf2]))
self.assertSetEqual(
set(dev1.find_interfaces()),
set([intf1, intf3]))
self.assertSetEqual(
set(dev2.find_interfaces()),
set([intf2]))
vlan = Vlan()
self.assertSetEqual(set(vlan.devices), set([]))
self.assertSetEqual(set(vlan.links), set([]))
link.add_feature(vlan)
self.assertCountEqual(vlan.devices, [dev1, dev2])
self.assertSetEqual(set(vlan.links), set([link]))
self.assertSetEqual(set(vlan.interfaces), set([intf1, intf2]))
with self.assertRaises(AttributeError):
vlan.access_map_id
with self.assertRaises(AttributeError):
vlan.vlan_configuration_id
def test_basic_cfg(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxe', context='yang')
dev2 = Device(testbed=testbed, name='PE2', os='iosxe', context='yang')
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()
for dev in testbed.devices:
dev.connections = Mock()
dev.connections = {'netconf': NetconfConnectionInfo()}
link.add_feature(vlan)
vlan.vlan_id = 100
cfg1 = vlan.build_config(apply=False)
self.assertCountEqual(cfg1.keys(), ['PE1', 'PE2'])
compare = ""
for i in cfg1['PE1']:
compare += str(i)
self.assertMultiLineEqual(
compare, '\n'.join([
'<edit-config xmlns="urn:ietf:params:xml:ns:netconf:base:1.0">\n'
' <target>\n'
' <running></running>\n'
' </target>\n'
' <config>\n'
' <vlan xmlns="http://cisco.com/ns/yang/Cisco-IOS-XE-native">\n'
' <vlan-list xmlns="http://cisco.com/ns/yang/Cisco-IOS-XE-vlan">\n'
' <id>100</id>\n'
' </vlan-list>\n'
' </vlan>\n'
' </config>\n'
'</edit-config>\n'
]))
compare = ""
for i in cfg1['PE2']:
compare += str(i)
self.assertMultiLineEqual(
compare, '\n'.join([
'<edit-config xmlns="urn:ietf:params:xml:ns:netconf:base:1.0">\n'
' <target>\n'
' <running></running>\n'
' </target>\n'
' <config>\n'
' <vlan xmlns="http://cisco.com/ns/yang/Cisco-IOS-XE-native">\n'
' <vlan-list xmlns="http://cisco.com/ns/yang/Cisco-IOS-XE-vlan">\n'
' <id>100</id>\n'
' </vlan-list>\n'
' </vlan>\n'
' </config>\n'
'</edit-config>\n'
]))
def test_vlan_interface_configuration(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxr')
dev2 = Device(testbed=testbed, name='PE2', os='iosxr')
intf1 = Interface(name='GigabitEthernet0/0/1', device=dev1)
intf2 = Interface(name='GigabitEthernet0/0/2', device=dev2)
intf3 = Interface(name='GigabitEthernet0/0/3', 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.device_attr[dev1]
vlan.device_attr[dev2]
vlan.device_attr[dev1].interface_attr[intf1]
vlan.device_attr[dev1].interface_attr[intf1].eth_encap_type1 = 'dot1q'
vlan.device_attr[dev1].interface_attr[intf1].eth_encap_val1 = 2
vlan.device_attr[dev1].interface_attr[
intf1].eth_encap_type2 = 'second-dot1q'
vlan.device_attr[dev1].interface_attr[intf1].eth_encap_val2 = 5
cfg1 = vlan.build_config(apply=False)
self.assertCountEqual(cfg1.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(cfg1['PE1']), '\n'.join([
'interface GigabitEthernet0/0/1',
' encapsulation dot1q 2 second-dot1q 5',
' exit',
]))
def test_vlan_interface_configuration(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxr', context='yang')
dev2 = Device(testbed=testbed, name='PE2', os='iosxr', context='yang')
intf1 = Interface(name='GigabitEthernet0/0/1',
device=dev1,
context='yang')
intf2 = Interface(name='GigabitEthernet0/0/2',
device=dev2,
context='yang')
intf3 = Interface(name='GigabitEthernet0/0/3',
device=dev1,
context='yang')
link = Link(name='1_2_1', testbed=testbed)
link.connect_interface(interface=intf1)
link.connect_interface(interface=intf2)
vlan = Vlan()
for dev in testbed.devices:
dev.connections = Mock()
dev.connections = {'netconf': NetconfConnectionInfo()}
link.add_feature(vlan)
vlan.device_attr[dev1]
vlan.device_attr[dev2]
vlan.device_attr[dev1].interface_attr[intf1]
vlan.device_attr[dev1].interface_attr[intf1].eth_encap_type1 = 'dot1q'
vlan.device_attr[dev1].interface_attr[intf1].eth_encap_val1 = 2
vlan.device_attr[dev1].interface_attr[
intf1].eth_encap_type2 = 'second-dot1q'
vlan.device_attr[dev1].interface_attr[intf1].eth_encap_val2 = 5
class test_te(unittest.TestCase):
def setUp(self):
self.tb = Genie.testbed = Testbed()
self.dev1 = Device(testbed=self.tb, name='PE1', os='iosxe')
self.dev2 = Device(testbed=self.tb, name='PE2', os='iosxe')
self.i1 = Interface(name='GigabitEthernet0/0/1', device=self.dev1)
self.i2 = Interface(name='GigabitEthernet0/0/2', device=self.dev2)
self.i3 = Interface(name='GigabitEthernet0/0/3', device=self.dev1)
self.i4 = Interface(name='GigabitEthernet0/0/4', device=self.dev2)
self.i5 = Interface(name='GigabitEthernet0/0/5', device=self.dev1)
self.i6 = Interface(name='GigabitEthernet0/0/6', device=self.dev2)
self.i7 = Interface(name='GigabitEthernet0/0/7', device=self.dev1)
self.i8 = Interface(name='GigabitEthernet0/0/8', device=self.dev2)
self.link = Link(name='1_2_1', testbed=self.tb)
self.link.connect_interface(interface=self.i1)
self.link.connect_interface(interface=self.i2)
self.link2 = Link(name='1_2_2', testbed=self.tb)
self.link2.connect_interface(interface=self.i3)
self.link2.connect_interface(interface=self.i4)
self.link3 = Link(name='1_2_3', testbed=self.tb)
self.link3.connect_interface(interface=self.i5)
self.link3.connect_interface(interface=self.i6)
self.link4 = Link(name='1_2_4', testbed=self.tb)
self.link4.connect_interface(interface=self.i7)
self.link4.connect_interface(interface=self.i8)
self.assertSetEqual(set(self.link.find_interfaces()),
set([self.i1, self.i2]))
self.assertSetEqual(set(self.dev1.find_interfaces()),
set([self.i1, self.i3, self.i5, self.i7]))
self.assertSetEqual(set(self.dev2.find_interfaces()),
set([self.i2, self.i4, self.i6, self.i8]))
def test_MplsTe(self):
te = Te()
self.assertSetEqual(set(te.devices), set([]))
self.assertSetEqual(set(te.links), set([]))
self.link.add_feature(te)
self.link2.add_feature(te)
self.link3.add_feature(te)
self.link4.add_feature(te)
self.assertCountEqual(te.devices, [self.dev1, self.dev2])
self.assertSetEqual(
set(te.links), set([self.link, self.link2, self.link3,
self.link4]))
self.assertSetEqual(
set(te.interfaces),
set([
self.i1, self.i2, self.i3, self.i4, self.i5, self.i6, self.i7,
self.i8
]))
out = te.build_config(apply=False)
self.assertMultiLineEqual(
str(out['PE1']), '\n'.join([
'mpls traffic-eng tunnels',
'interface GigabitEthernet0/0/1',
' mpls traffic-eng tunnels',
' exit',
'interface GigabitEthernet0/0/3',
' mpls traffic-eng tunnels',
' exit',
'interface GigabitEthernet0/0/5',
' mpls traffic-eng tunnels',
' exit',
'interface GigabitEthernet0/0/7',
' mpls traffic-eng tunnels',
' exit',
]))
self.assertMultiLineEqual(
str(out['PE2']), '\n'.join([
'mpls traffic-eng tunnels',
'interface GigabitEthernet0/0/2',
' mpls traffic-eng tunnels',
' exit',
'interface GigabitEthernet0/0/4',
' mpls traffic-eng tunnels',
' exit',
'interface GigabitEthernet0/0/6',
' mpls traffic-eng tunnels',
' exit',
'interface GigabitEthernet0/0/8',
' mpls traffic-eng tunnels',
' exit',
]))
out = te.build_unconfig(apply=False)
self.assertMultiLineEqual(
str(out['PE1']), '\n'.join([
'default mpls traffic-eng tunnels',
'interface GigabitEthernet0/0/1',
' default mpls traffic-eng tunnels',
' exit',
'interface GigabitEthernet0/0/3',
' default mpls traffic-eng tunnels',
' exit',
'interface GigabitEthernet0/0/5',
' default mpls traffic-eng tunnels',
' exit',
'interface GigabitEthernet0/0/7',
' default mpls traffic-eng tunnels',
' exit',
]))
self.assertMultiLineEqual(
str(out['PE2']), '\n'.join([
'default mpls traffic-eng tunnels',
'interface GigabitEthernet0/0/2',
' default mpls traffic-eng tunnels',
' exit',
'interface GigabitEthernet0/0/4',
' default mpls traffic-eng tunnels',
' exit',
'interface GigabitEthernet0/0/6',
' default mpls traffic-eng tunnels',
' exit',
'interface GigabitEthernet0/0/8',
' default mpls traffic-eng tunnels',
' exit',
]))
def test_init(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxr')
intf1 = Interface(device=dev1, name='GigabitEthernet0/0/0/1')
intf2 = Interface(device=dev1, name='GigabitEthernet0/0/0/2')
dev2 = Device(testbed=testbed, name='PE2', os='iosxr')
intf3 = Interface(device=dev2, name='GigabitEthernet0/0/0/3')
intf4 = Interface(device=dev2, name='GigabitEthernet0/0/0/4')
link1 = Link(testbed=testbed, name='link1', interfaces=(intf1, intf3))
link2 = Link(testbed=testbed, name='link2', interfaces=(intf2, intf4))
with self.assertRaises(TypeError):
xc1 = Xconnect()
with self.assertRaises(TypeError):
xc1 = Xconnect(group_name='bg1')
xc1 = Xconnect(name='xc1', group_name='bg1')
self.assertIs(xc1.xconnect_type, Xconnect.Type.p2p)
self.assertEqual(xc1.name, 'xc1')
self.assertEqual(xc1.group_name, 'bg1')
xc1 = Xconnect(name='xc1')
self.assertEqual(xc1.name, 'xc1')
self.assertEqual(xc1.group_name, 'xc1g')
self.assertCountEqual(xc1.devices, [])
self.assertCountEqual(xc1.interfaces, [])
self.assertCountEqual(xc1.segments, [])
self.assertCountEqual(xc1.link.interfaces, [])
dev1.add_feature(xc1)
self.assertCountEqual(xc1.devices, [dev1])
self.assertCountEqual(xc1.interfaces, [])
self.assertCountEqual(xc1.segments, [])
self.assertCountEqual(xc1.link.interfaces, [])
cfgs = xc1.build_config(apply=False)
self.assertCountEqual(cfgs.keys(), [dev1.name])
self.assertMultiLineEqual(str(cfgs[dev1.name]), '\n'.join([
'l2vpn',
' xconnect group xc1g',
' p2p xc1',
' exit',
' exit',
' exit',
]))
#xc1.add_interface(intf1)
intf1.l2transport.enabled = True
#self.assertCountEqual(xc1.interfaces, [intf1])
#self.assertCountEqual(xc1.devices, [dev1])
#self.assertCountEqual(xc1.segments, [intf1])
#self.assertCountEqual(xc1.link.interfaces, [intf3])
#self.assertCountEqual(xc1.device_attr[dev1].interfaces, [intf1])
#self.assertCountEqual(xc1.device_attr[dev2].interfaces, [])
#self.assertCountEqual(xc1.device_attr[dev1].segments, [intf1])
self.assertCountEqual(xc1.device_attr[dev2].segments, [])
cfgs = xc1.build_config(apply=False)
self.assertCountEqual(cfgs.keys(), [dev1.name])
if False:
self.assertMultiLineEqual(str(cfgs[dev1.name]), '\n'.join([
'l2vpn',
' xconnect group xc1g',
' p2p xc1',
' interface GigabitEthernet0/0/0/1',
' exit',
' exit',
' exit',
]))
dev2.add_feature(xc1)
xc1.xconnect_type = Xconnect.Type.mp2mp
xc1.autodiscovery_bgp.enabled = True
xc1.autodiscovery_bgp.signaling_protocol_bgp.enabled = True
xc1.autodiscovery_bgp.export_route_targets = [RouteTarget.ImportExport('1.1.1.1:1')]
xc1.autodiscovery_bgp.import_route_targets = [RouteTarget.ImportExport('1.1.1.1:1')]
xc1.autodiscovery_bgp.rd = '1000:1'
xc1.device_attr[dev1].vpn_id = 100
xc1.device_attr[dev2].vpn_id = 101
ce_id1 = 1001
xc1.device_attr[dev1].autodiscovery_bgp.signaling_protocol_bgp.add_ce_id(ce_id1)
xc1.device_attr[dev1].autodiscovery_bgp.signaling_protocol_bgp.ce_attr[ce_id1].add_interface(intf1)
ce_id2 = 1000
xc1.device_attr[dev2].autodiscovery_bgp.signaling_protocol_bgp.add_ce_id(ce_id1)
xc1.device_attr[dev2].autodiscovery_bgp.signaling_protocol_bgp.ce_attr[ce_id2].add_interface(intf2)
xc1.device_attr[dev1].autodiscovery_bgp.signaling_protocol_bgp.ce_attr[ce_id1].interface_attr[intf1].remote_ce_id = ce_id2
xc1.device_attr[dev2].autodiscovery_bgp.signaling_protocol_bgp.ce_attr[ce_id2].interface_attr[intf2].remote_ce_id = ce_id1
cfgs = xc1.build_config(apply=False)
def test_vlan_interface_confoguration(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxr')
dev2 = Device(testbed=testbed, name='PE2', os='iosxr')
intf1 = Interface(name='GigabitEthernet0/0/1', device=dev1)
intf2 = Interface(name='GigabitEthernet0/0/2', device=dev2)
intf3 = Interface(name='GigabitEthernet0/0/3', 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.device_attr[dev1]
vlan.device_attr[dev2]
# Defining attributes section
intf1.transceiver_permit_pid_all = True
intf1.duplex = 'full'
intf1.mac_address = 'aaaa.bbbb.cccc'
intf1.eth_encap_type1 = 'dot1q'
intf1.eth_encap_val1 = 2
intf1.eth_encap_val2 = 5
intf1.eth_dot1q_type = 'tunneling ethertype'
intf1.eth_dot1q_value = '0x9100'
intf1.ipv4 = '201.0.12.1'
intf2.eth_dot1q_type = 'native vlan'
intf2.eth_dot1q_value = '15'
cfg1 = intf1.build_config(apply=False)
self.assertMultiLineEqual(
str(cfg1), '\n'.join([
'interface GigabitEthernet0/0/1',
' dot1q tunneling ethertype 0x9100',
' encapsulation dot1q 2 second-dot1q 5',
' ipv4 address 201.0.12.1/32',
' transceiver permit pid all',
' duplex full',
' mac-address aaaa.bbbb.cccc',
' exit',
]))
cfg2 = intf2.build_config(apply=False)
self.assertMultiLineEqual(
str(cfg2), '\n'.join([
'interface GigabitEthernet0/0/2',
' dot1q native vlan 15',
' exit',
]))
uncfg = intf1.build_unconfig(apply=False,
attributes={
'mac_address': True,
'duplex': True,
'ipv4': True,
'eth_dot1q_value': True,
'eth_encap_type1': True,
})
self.assertMultiLineEqual(
str(uncfg), '\n'.join([
'interface GigabitEthernet0/0/1',
' no encapsulation dot1q 2 second-dot1q 5',
' no ipv4 address 201.0.12.1/32',
' no duplex full',
' no mac-address',
' exit',
]))
class test_te(unittest.TestCase):
def setUp(self):
self.tb = Genie.testbed = Testbed()
self.dev1 = Device(testbed=self.tb, name='PE1', os='iosxr')
self.dev2 = Device(testbed=self.tb, name='PE2', os='iosxr')
self.i1 = Interface(name='GigabitEthernet0/0/0/1', device=self.dev1)
self.i2 = Interface(name='GigabitEthernet0/0/0/2', device=self.dev2)
self.i3 = Interface(name='GigabitEthernet0/0/0/3', device=self.dev1)
self.i4 = Interface(name='GigabitEthernet0/0/0/4', device=self.dev2)
self.i5 = Interface(name='GigabitEthernet0/0/0/5', device=self.dev1)
self.i6 = Interface(name='GigabitEthernet0/0/0/6', device=self.dev2)
self.i7 = Interface(name='GigabitEthernet0/0/0/7', device=self.dev1)
self.i8 = Interface(name='GigabitEthernet0/0/0/8', device=self.dev2)
self.link = Link(name='1_2_1', testbed=self.tb)
self.link.connect_interface(interface=self.i1)
self.link.connect_interface(interface=self.i2)
self.link2 = Link(name='1_2_2', testbed=self.tb)
self.link2.connect_interface(interface=self.i3)
self.link2.connect_interface(interface=self.i4)
self.link3 = Link(name='1_2_3', testbed=self.tb)
self.link3.connect_interface(interface=self.i5)
self.link3.connect_interface(interface=self.i6)
self.link4 = Link(name='1_2_4', testbed=self.tb)
self.link4.connect_interface(interface=self.i7)
self.link4.connect_interface(interface=self.i8)
self.assertSetEqual(set(self.link.find_interfaces()),
set([self.i1, self.i2]))
self.assertSetEqual(set(self.dev1.find_interfaces()),
set([self.i1, self.i3, self.i5, self.i7]))
self.assertSetEqual(set(self.dev2.find_interfaces()),
set([self.i2, self.i4, self.i6, self.i8]))
def test_MplsTe(self):
te = Te()
self.assertSetEqual(set(te.devices), set([]))
self.assertSetEqual(set(te.links), set([]))
if 0:
print("before")
print(te.devices)
print(te.links)
print(te.interfaces)
self.link.add_feature(te)
self.link2.add_feature(te)
self.link3.add_feature(te)
self.link4.add_feature(te)
self.assertCountEqual(te.devices, [self.dev1, self.dev2])
self.assertSetEqual(
set(te.links), set([self.link, self.link2, self.link3,
self.link4]))
self.assertSetEqual(
set(te.interfaces),
set([
self.i1, self.i2, self.i3, self.i4, self.i5, self.i6, self.i7,
self.i8
]))
if 0:
print("after")
print(te.links)
print(te.devices)
print(te.interfaces)
te.log_events_preemption = True
te.log_events_frr_protection = True
te.device_attr['PE1'].log_events_frr_protection_type = 'backup-tunnel'
te.device_attr['PE2'].log_events_frr_protection_type = 'primary-lsp'
te.device_attr[
'PE2'].log_events_frr_protection_primary_lsp_type = 'active-state'
te.srlg_admin_weight = 20000
te.backup_auto_tun_tunid_min = 210
te.backup_auto_tun_tunid_max = 600
te.auto_tun_backup_affinity_ignore = True
te.affinity_map_val_dict['RED'] = "0x1"
te.soft_preempt_timeout = 5
te.reoptimize_delay_cleanup = 10
te.reoptimize_delay_install = 1
te.flooding_threshold_up = 1
te.flooding_threshold_down = 1
te.p2mp_auto_tun_tunid_min = 100
te.p2mp_auto_tun_tunid_max = 200
te.device_attr['PE1'].p2mp_auto_tun_tunid_min = 200
te.device_attr['PE1'].p2mp_auto_tun_tunid_max = 300
te.auto_tun_backup_affinity_ignore = True
te.auto_tun_backup_timers_rem_unused = 100
te.auto_tun_backup_attr_set = "backup"
te.device_attr['PE1'].interface_attr[
self.i1].auto_tun_backup_exclude_srlg = True
te.device_attr['PE1'].interface_attr[
self.i3].auto_tun_backup_exclude_srlg = True
te.device_attr['PE1'].interface_attr[
self.i3].auto_tun_backup_exclude_srlg_type = 'preferred'
te.device_attr['PE1'].interface_attr[
self.i5].auto_tun_backup_exclude_srlg = True
te.device_attr['PE1'].interface_attr[
self.i5].auto_tun_backup_exclude_srlg_type = 'weighted'
te.device_attr['PE1'].affinity_map_val_dict = {}
te.device_attr['PE1'].affinity_map_val_dict['RED'] = "0x2"
te.device_attr['PE2'].affinity_map_bitpos_dict = {}
te.device_attr['PE2'].affinity_map_bitpos_dict['BLUE'] = 94
te.device_attr['PE2'].affinity_map_bitpos_dict['EDGE'] = 27
te.device_attr['PE2'].affinity_map_bitpos_dict['PINK'] = 95
te.device_attr['PE2'].affinity_map_bitpos_dict['GREEN'] = 91
te.device_attr['PE2'].affinity_map_bitpos_dict['METRO'] = 29
out = te.build_config(apply=False)
self.maxDiff = None
self.assertCountEqual(out.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(out['PE1']), '\n'.join([
'mpls traffic-eng',
' auto-tunnel backup affinity ignore',
' auto-tunnel backup timers removal unused 100',
' auto-tunnel backup tunnel-id min 210 max 600',
' auto-tunnel p2mp tunnel-id min 200 max 300',
' logging events preemption',
' logging events frr-protection backup-tunnel',
' reoptimize timers delay cleanup 10',
' reoptimize timers delay installation 1',
' flooding threshold up 1 down 1',
' affinity-map RED 0x2',
' srlg admin-weight 20000',
' soft-preemption timeout 5',
' interface GigabitEthernet0/0/0/1',
' auto-tunnel backup attribute-set backup',
' auto-tunnel backup exclude srlg',
' exit',
' interface GigabitEthernet0/0/0/3',
' auto-tunnel backup attribute-set backup',
' auto-tunnel backup exclude srlg preferred',
' exit',
' interface GigabitEthernet0/0/0/5',
' auto-tunnel backup attribute-set backup',
' auto-tunnel backup exclude srlg weighted',
' exit',
' interface GigabitEthernet0/0/0/7',
' auto-tunnel backup attribute-set backup',
' exit',
' exit',
]))
if 1:
self.assertMultiLineEqual(
str(out['PE2']), '\n'.join([
'mpls traffic-eng',
' auto-tunnel backup affinity ignore',
' auto-tunnel backup timers removal unused 100',
' auto-tunnel backup tunnel-id min 210 max 600',
' auto-tunnel p2mp tunnel-id min 100 max 200',
' logging events preemption',
' logging events frr-protection primary-lsp active-state',
' reoptimize timers delay cleanup 10',
' reoptimize timers delay installation 1',
' flooding threshold up 1 down 1',
' affinity-map BLUE bit-position 94',
' affinity-map EDGE bit-position 27',
' affinity-map GREEN bit-position 91',
' affinity-map METRO bit-position 29',
' affinity-map PINK bit-position 95',
' affinity-map RED 0x1',
' srlg admin-weight 20000',
' soft-preemption timeout 5',
' interface GigabitEthernet0/0/0/2',
' auto-tunnel backup attribute-set backup',
' exit',
' interface GigabitEthernet0/0/0/4',
' auto-tunnel backup attribute-set backup',
' exit',
' interface GigabitEthernet0/0/0/6',
' auto-tunnel backup attribute-set backup',
' exit',
' interface GigabitEthernet0/0/0/8',
' auto-tunnel backup attribute-set backup',
' exit',
' exit',
]))
def test_UnnumInterfaces(self):
# Test unnum interface output
te = Te()
self.link.add_feature(te)
te.ipv4_unnum_interfaces = {self.i1, self.i2, self.i3, self.i4}
out = te.build_config(apply=False)
self.assertRegex(
str(out['PE1']),
'ipv4 unnumbered mpls traffic-eng GigabitEthernet0/0/0/1')
self.assertRegex(
str(out['PE1']),
'ipv4 unnumbered mpls traffic-eng GigabitEthernet0/0/0/3')
self.assertRegex(
str(out['PE2']),
'ipv4 unnumbered mpls traffic-eng GigabitEthernet0/0/0/2')
self.assertRegex(
str(out['PE2']),
'ipv4 unnumbered mpls traffic-eng GigabitEthernet0/0/0/4')
def test_Srlg(self):
srlg = Srlg()
self.dev1.add_feature(srlg)
self.dev2.add_feature(srlg)
srlg.name_value_dict['R13'] = 10
srlg.name_value_dict['R11'] = 20
srlg.name_value_dict['R23'] = 30
srlg.name_value_dict['R25'] = 40
srlg.name_value_dict['R34'] = 50
# if per-device dict is not initialized, base class dict will be over-written
srlg.device_attr['PE1'].name_value_dict = {}
srlg.device_attr['PE1'].name_value_dict['R13'] = 10
srlg.device_attr['PE1'].name_value_dict['R11'] = 20
srlg.device_attr['PE1'].name_value_dict['R23'] = 30
srlg.device_attr['PE1'].name_value_dict['R25'] = 40
srlg.device_attr['PE1'].name_value_dict['R34'] = 50
srlg.device_attr['PE1'].name_value_dict['R35'] = 60
srlg.device_attr['PE1'].name_value_dict['R45'] = 70
srlg.device_attr['PE1'].interface_attr[self.i1].intf_name = 'R13'
srlg.device_attr['PE1'].interface_attr[self.i3].intf_name = 'R11'
srlg.device_attr['PE1'].interface_attr[self.i5].intf_name = 'R23'
out = srlg.build_config(apply=False)
self.assertCountEqual(out.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(out['PE1']), '\n'.join([
'srlg',
' name R11 value 20',
' name R13 value 10',
' name R23 value 30',
' name R25 value 40',
' name R34 value 50',
' name R35 value 60',
' name R45 value 70',
' interface GigabitEthernet0/0/0/1',
' name R13',
' exit',
' interface GigabitEthernet0/0/0/3',
' name R11',
' exit',
' interface GigabitEthernet0/0/0/5',
' name R23',
' exit',
' exit',
]))
self.assertMultiLineEqual(
str(out['PE2']), '\n'.join([
'srlg',
' name R11 value 20',
' name R13 value 10',
' name R23 value 30',
' name R25 value 40',
' name R34 value 50',
' exit',
]))
def test_init(self):
# import pdb; pdb.set_trace()
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxr')
intf1 = Interface(device=dev1, name='GigabitEthernet0/0/0/1')
intf2 = Interface(device=dev1, name='GigabitEthernet0/0/0/2')
dev2 = Device(testbed=testbed, name='PE2', os='iosxr')
intf3 = Interface(device=dev2, name='GigabitEthernet0/0/0/3')
intf4 = Interface(device=dev2, name='GigabitEthernet0/0/0/4')
link1 = Link(testbed=testbed, name='link1', interfaces=(intf1, intf3))
link2 = Link(testbed=testbed, name='link2', interfaces=(intf2, intf4))
with self.assertRaises(TypeError):
bd1 = BridgeDomain()
with self.assertRaises(TypeError):
bd1 = BridgeDomain(group_name='bg1')
bd1 = BridgeDomain(name='bd1', group_name='mybg1')
self.assertEqual(bd1.name, 'bd1')
self.assertEqual(bd1.group_name, 'mybg1')
bd1 = BridgeDomain(name='bd1')
self.assertEqual(bd1.name, 'bd1')
self.assertEqual(bd1.group_name, 'bd1g')
self.assertCountEqual(bd1.devices, [])
self.assertCountEqual(bd1.interfaces, [])
self.assertCountEqual(bd1.vnis, [])
self.assertCountEqual(bd1.vfis, [])
self.assertCountEqual(bd1.evis, [])
self.assertCountEqual(bd1.segments, [])
self.assertCountEqual(bd1.link.interfaces, [])
dev1.add_feature(bd1)
self.assertCountEqual(bd1.devices, [dev1])
self.assertCountEqual(bd1.interfaces, [])
self.assertCountEqual(bd1.vnis, [])
self.assertCountEqual(bd1.vfis, [])
self.assertCountEqual(bd1.evis, [])
self.assertCountEqual(bd1.segments, [])
self.assertCountEqual(bd1.link.interfaces, [])
cfgs = bd1.build_config(apply=False)
self.assertCountEqual(cfgs.keys(), [dev1.name])
self.assertMultiLineEqual(
str(cfgs[dev1.name]), '\n'.join([
'l2vpn',
' bridge group bd1g',
' bridge-domain bd1',
' exit',
' exit',
' exit',
]))
bd1.add_interface(intf1)
intf1.l2transport.enabled = True
self.assertCountEqual(bd1.interfaces, [intf1])
self.assertCountEqual(bd1.devices, [dev1])
self.assertCountEqual(bd1.vnis, [])
self.assertCountEqual(bd1.vfis, [])
self.assertCountEqual(bd1.evis, [])
self.assertCountEqual(bd1.segments, [intf1])
# Links under Genie Interface object is deprecated
# Placed the below workaround to bypass the Unittest (commented out)
# self.assertCountEqual(bd1.link.interfaces, [intf3])
self.assertCountEqual(bd1.device_attr[dev1].interfaces, [intf1])
self.assertCountEqual(bd1.device_attr[dev2].interfaces, [])
self.assertCountEqual(bd1.device_attr[dev1].segments, [intf1])
self.assertCountEqual(bd1.device_attr[dev2].segments, [])
cfgs = bd1.build_config(apply=False)
self.assertCountEqual(cfgs.keys(), [dev1.name])
self.assertMultiLineEqual(
str(cfgs[dev1.name]), '\n'.join([
'l2vpn',
' bridge group bd1g',
' bridge-domain bd1',
' interface GigabitEthernet0/0/0/1',
' exit',
' exit',
' exit',
' exit',
]))
def test_init(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxe')
intf1 = Interface(device=dev1, name='GigabitEthernet0/0/1')
intf2 = Interface(device=dev1, name='GigabitEthernet0/0/2')
dev2 = Device(testbed=testbed, name='PE2', os='iosxe')
intf3 = Interface(device=dev2, name='GigabitEthernet0/0/3')
intf4 = Interface(device=dev2, name='GigabitEthernet0/0/4')
link1 = Link(testbed=testbed, name='link1', interfaces=(intf1, intf3))
link2 = Link(testbed=testbed, name='link2', interfaces=(intf2, intf4))
bd1 = BridgeDomain(name='200')
self.assertEqual(bd1.name, '200')
self.assertCountEqual(bd1.devices, [])
self.assertCountEqual(bd1.interfaces, [])
self.assertCountEqual(bd1.vnis, [])
self.assertCountEqual(bd1.vfis, [])
self.assertCountEqual(bd1.evis, [])
self.assertCountEqual(bd1.segments, [])
self.assertCountEqual(bd1.link.interfaces, [])
dev1.add_feature(bd1)
self.assertCountEqual(bd1.devices, [dev1])
self.assertCountEqual(bd1.interfaces, [])
self.assertCountEqual(bd1.vnis, [])
self.assertCountEqual(bd1.vfis, [])
self.assertCountEqual(bd1.evis, [])
self.assertCountEqual(bd1.segments, [])
self.assertCountEqual(bd1.link.interfaces, [])
cfgs = bd1.build_config(apply=False)
self.assertCountEqual(cfgs.keys(), [dev1.name])
self.assertMultiLineEqual(str(cfgs[dev1.name]), '\n'.join([
'bridge-domain 200',
' exit',
]))
efp = Interface(device=dev1,
name='GigabitEthernet0/0/1.20',
service_instance=20)
efp.encapsulation = 'dot1q'
efp.rewrite_ingress = 'pop 1 symmetric'
bd1.add_segment(efp)
self.assertCountEqual(bd1.interfaces, [efp])
self.assertCountEqual(bd1.devices, [dev1])
self.assertCountEqual(bd1.vnis, [])
self.assertCountEqual(bd1.vfis, [])
self.assertCountEqual(bd1.evis, [])
self.assertCountEqual(bd1.segments, [efp])
self.assertCountEqual(bd1.device_attr[dev1].interfaces, [efp])
self.assertCountEqual(bd1.device_attr[dev2].interfaces, [])
self.assertCountEqual(bd1.device_attr[dev1].segments, [efp])
self.assertCountEqual(bd1.device_attr[dev2].segments, [])
cfgs = bd1.build_config(apply=False)
self.assertCountEqual(cfgs.keys(), [dev1.name])
self.assertMultiLineEqual(
str(cfgs[dev1.name]), '\n'.join([
'bridge-domain 200',
' member GigabitEthernet0/0/1 service-instance 20',
' exit',
]))
class test_ldp(unittest.TestCase):
def setUp(self):
self.maxDiff = None
tb = Genie.testbed = Testbed()
self.dev1 = Device(testbed=tb, name='PE1', os='iosxe')
self.dev2 = Device(testbed=tb, name='PE2', os='iosxe')
self.i1 = Interface(name='GigabitEthernet0/0/1', device=self.dev1)
self.i2 = Interface(name='GigabitEthernet0/0/2', device=self.dev2)
self.i3 = Interface(name='GigabitEthernet0/0/3', device=self.dev1)
self.i4 = Interface(name='GigabitEthernet0/0/4', device=self.dev2)
self.i5 = Interface(name='GigabitEthernet0/0/5', device=self.dev1)
self.i6 = Interface(name='GigabitEthernet0/0/6', device=self.dev2)
self.i7 = Interface(name='GigabitEthernet0/0/7', device=self.dev1)
self.i8 = Interface(name='GigabitEthernet0/0/8', device=self.dev2)
self.link = Link(name='1_2_1')
self.link.connect_interface(interface=self.i1)
self.link.connect_interface(interface=self.i2)
self.link2 = Link(name='1_2_2')
self.link2.connect_interface(interface=self.i3)
self.link2.connect_interface(interface=self.i4)
self.link3 = Link(name='1_2_3')
self.link3.connect_interface(interface=self.i5)
self.link3.connect_interface(interface=self.i6)
self.link4 = Link(name='1_2_4')
self.link4.connect_interface(interface=self.i7)
self.link4.connect_interface(interface=self.i8)
self.assertSetEqual(set(self.link.find_interfaces()),
set([self.i1, self.i2]))
self.assertSetEqual(set(self.dev1.find_interfaces()),
set([self.i1, self.i3, self.i5, self.i7]))
self.assertSetEqual(set(self.dev2.find_interfaces()),
set([self.i2, self.i4, self.i6, self.i8]))
def test_01_interface_only(self):
ldp = Ldp()
self.assertSetEqual(set(ldp.devices), set([]))
self.assertSetEqual(set(ldp.links), set([]))
self.link.add_feature(ldp)
self.link2.add_feature(ldp)
self.link3.add_feature(ldp)
self.link4.add_feature(ldp)
self.assertCountEqual(ldp.devices, [self.dev1, self.dev2])
self.assertSetEqual(
set(ldp.links),
set([self.link, self.link2, self.link3, self.link4]))
self.assertSetEqual(
set(ldp.interfaces),
set([
self.i1, self.i2, self.i3, self.i4, self.i5, self.i6, self.i7,
self.i8
]))
# ==Test interface-only config==
out = ldp.build_config(apply=False)
if 1:
self.assertRegex(str(out['PE1']), 'interface GigabitEthernet0/0/1')
self.assertRegex(str(out['PE1']), 'interface GigabitEthernet0/0/3')
self.assertRegex(str(out['PE1']), 'interface GigabitEthernet0/0/5')
self.assertRegex(str(out['PE1']), 'interface GigabitEthernet0/0/7')
self.assertRegex(str(out['PE2']), 'interface GigabitEthernet0/0/2')
self.assertRegex(str(out['PE2']), 'interface GigabitEthernet0/0/4')
self.assertRegex(str(out['PE2']), 'interface GigabitEthernet0/0/6')
self.assertRegex(str(out['PE2']), 'interface GigabitEthernet0/0/8')
def test_1_top_level(self):
acl1 = AccessList(name='1')
acl2 = AccessList(name='2')
acl3 = AccessList(name='3')
acl4 = AccessList(name='4')
acl5 = AccessList(name='5')
acl6 = AccessList(name='6')
acl7 = AccessList(name='7')
acl8 = AccessList(name='8')
acl9 = AccessList(name='9')
acl11 = AccessList(name='11')
acl22 = AccessList(name='22')
acl33 = AccessList(name='33')
acl44 = AccessList(name='44')
ldp = Ldp()
self.link.add_feature(ldp)
# ==Test top-level config==
ldp.hello_holdtime = 100
ldp.hello_interval = 200
ldp.targeted_hello_accept = True
ldp.gr = True
ldp.gr_fwdstate_holdtime = 60
ldp.nsr = True
ldp.device_attr['PE1'].session_holdtime = 444
ldp.session_protection = True
ldp.session_protection_dur = 222
ldp.device_attr['PE2'].session_protection_dur = 333
ldp.session_protection_for_acl = acl1
out = ldp.build_config(apply=False)
if 1:
self.assertCountEqual(out.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(out['PE1']), '\n'.join([
'mpls label protocol ldp',
'mpls ip',
'mpls ldp nsr',
'mpls ldp graceful-restart',
'mpls ldp graceful-restart timers forwarding-holding 60',
'mpls ldp discovery hello interval 200',
'mpls ldp discovery hello holdtime 100',
'mpls ldp discovery targeted-hello accept',
'mpls ldp session protection for 1 222',
'interface GigabitEthernet0/0/1',
' mpls ip',
' exit',
]))
self.assertMultiLineEqual(
str(out['PE2']), '\n'.join([
'mpls label protocol ldp',
'mpls ip',
'mpls ldp nsr',
'mpls ldp graceful-restart',
'mpls ldp graceful-restart timers forwarding-holding 60',
'mpls ldp discovery hello interval 200',
'mpls ldp discovery hello holdtime 100',
'mpls ldp discovery targeted-hello accept',
'mpls ldp session protection for 1 333',
'interface GigabitEthernet0/0/2',
' mpls ip',
' exit',
]))
if 1:
# set the per-attr variables that are inherited by VRF
ldp.device_attr['PE1'].vrf_attr['default'].router_id = self.i1
ldp.device_attr['PE2'].vrf_attr['default'].router_id = self.i2
out = ldp.build_config(apply=False)
self.assertRegex(str(out['PE1']), 'router-id GigabitEthernet0/0/1')
self.assertRegex(str(out['PE2']), 'router-id GigabitEthernet0/0/2')
# Check unconfig - <nsr> config filter
out = ldp.build_unconfig(apply=False, attributes='device_attr__*__nsr')
if 1:
self.assertRegex(str(out['PE1']), 'no mpls ldp nsr')
self.assertRegex(str(out['PE2']), 'no mpls ldp nsr')
# Check unconfig - <gr> config filter=
out = ldp.build_unconfig(apply=False, attributes='device_attr__*__gr')
if 1:
self.assertRegex(str(out['PE1']), 'no mpls ldp graceful-restart')
self.assertRegex(str(out['PE2']), 'no mpls ldp graceful-restart')
def test_2_per_vrf(self):
acl1 = AccessList(name='1')
acl2 = AccessList(name='2')
acl3 = AccessList(name='3')
acl4 = AccessList(name='4')
acl5 = AccessList(name='5')
acl6 = AccessList(name='6')
acl7 = AccessList(name='7')
acl8 = AccessList(name='8')
acl9 = AccessList(name='9')
acl11 = AccessList(name='11')
acl22 = AccessList(name='22')
acl33 = AccessList(name='33')
acl44 = AccessList(name='44')
ldp = Ldp()
self.link.add_feature(ldp)
vrf = Vrf(name='vrf1')
ldp.add_force_vrf(vrf)
vrf2 = Vrf(name='vrf2')
ldp.add_force_vrf(vrf2)
ldp.device_attr['PE1'].router_id = self.i1
ldp.device_attr['PE2'].router_id = self.i2
ldp.device_attr['PE1'].vrf_attr['vrf1'].router_id = self.i3
ldp.device_attr['PE2'].vrf_attr['vrf1'].router_id = self.i4
# Test per vrf config
# Check vrf config - full config
out = ldp.build_config(apply=False)
self.assertMultiLineEqual(
str(out['PE1']), '\n'.join([
'mpls label protocol ldp',
'mpls ip',
'mpls ldp router-id GigabitEthernet0/0/1',
'mpls ldp router-id vrf vrf1 GigabitEthernet0/0/3',
'interface GigabitEthernet0/0/1',
' mpls ip',
' exit',
]))
self.assertMultiLineEqual(
str(out['PE2']), '\n'.join([
'mpls label protocol ldp',
'mpls ip',
'mpls ldp router-id GigabitEthernet0/0/2',
'mpls ldp router-id vrf vrf1 GigabitEthernet0/0/4',
'interface GigabitEthernet0/0/2',
' mpls ip',
' exit',
]))
# Test VRF unconfig
out = ldp.build_unconfig(apply=False,
attributes='device_attr__*__vrf_attr')
self.assertTrue(
'no mpls ldp router-id GigabitEthernet0/0/1' in str(out['PE1']))
self.assertTrue(
'no mpls ldp router-id GigabitEthernet0/0/2' in str(out['PE2']))
self.assertTrue('no mpls ldp router-id vrf vrf1 GigabitEthernet0/0/3'
in str(out['PE1']))
self.assertTrue('no mpls ldp router-id vrf vrf1 GigabitEthernet0/0/4'
in str(out['PE2']))
out2 = ldp.build_config(apply=False,
attributes='device_attr__*__vrf_attr')
self.assertTrue(
'mpls ldp router-id GigabitEthernet0/0/1' in str(out2['PE1']))
self.assertTrue(
'mpls ldp router-id GigabitEthernet0/0/2' in str(out2['PE2']))
self.assertTrue('mpls ldp router-id vrf vrf1 GigabitEthernet0/0/3' in
str(out2['PE1']))
self.assertTrue('mpls ldp router-id vrf vrf1 GigabitEthernet0/0/4' in
str(out2['PE2']))
# Check vrf config - vrf <vrf1> config filter
out3 = ldp.build_config(apply=False,
attributes='device_attr__*__vrf_attr__vrf1')
self.assertTrue('mpls ldp router-id vrf vrf1 GigabitEthernet0/0/3' in
str(out3['PE1']))
self.assertTrue('mpls ldp router-id vrf vrf1 GigabitEthernet0/0/4' in
str(out3['PE2']))
# interface output from default vrf should not be built
self.assertNotRegex(str(out['PE1']), '(?s)interface Gig')
# Check vrf unconfig - vrf <vrf1> config filter
out = ldp.build_unconfig(apply=False,
attributes='device_attr__*__vrf_attr__vrf1')
self.assertEqual(
str(out['PE1']),
'no mpls ldp router-id vrf vrf1 GigabitEthernet0/0/3')
self.assertEqual(
str(out['PE2']),
'no mpls ldp router-id vrf vrf1 GigabitEthernet0/0/4')
# Check vrf unconfig - vrf <vrf1> config filter
out = ldp.build_unconfig(
apply=False, attributes='device_attr__*__vrf_attr__*__router_id')
self.assertTrue(
'no mpls ldp router-id GigabitEthernet0/0/1' in str(out['PE1']))
self.assertTrue(
'no mpls ldp router-id GigabitEthernet0/0/2' in str(out['PE2']))
self.assertTrue('no mpls ldp router-id vrf vrf1 GigabitEthernet0/0/3'
in str(out['PE1']))
self.assertTrue('no mpls ldp router-id vrf vrf1 GigabitEthernet0/0/4'
in str(out['PE2']))
# Check vrf unconfig - vrf <vrf1> config filter after adding router-id to vrf2
ldp.device_attr['PE2'].vrf_attr['vrf2'].router_id = self.i6
out = ldp.build_unconfig(
apply=False, attributes='device_attr__*__vrf_attr__*__router_id')
self.assertTrue(
'mpls ldp router-id GigabitEthernet0/0/1' in str(out['PE1']))
self.assertTrue(
'mpls ldp router-id GigabitEthernet0/0/2' in str(out['PE2']))
self.assertTrue('mpls ldp router-id vrf vrf1 GigabitEthernet0/0/3' in
str(out['PE1']))
self.assertTrue('mpls ldp router-id vrf vrf1 GigabitEthernet0/0/4' in
str(out['PE2']))
self.assertTrue('mpls ldp router-id vrf vrf2 GigabitEthernet0/0/6' in
str(out['PE2']))
# Check vrf config - vrf <vrf2> config filter
out3 = ldp.build_config(apply=False,
attributes='device_attr__*__vrf_attr__vrf2')
self.assertEqual(str(out3['PE2']),
'mpls ldp router-id vrf vrf2 GigabitEthernet0/0/6')
def test_3_per_intf(self):
acl1 = AccessList(name='1')
acl2 = AccessList(name='2')
acl3 = AccessList(name='3')
acl4 = AccessList(name='4')
acl5 = AccessList(name='5')
acl6 = AccessList(name='6')
acl7 = AccessList(name='7')
acl8 = AccessList(name='8')
acl9 = AccessList(name='9')
acl11 = AccessList(name='11')
acl22 = AccessList(name='22')
acl33 = AccessList(name='33')
acl44 = AccessList(name='44')
ldp = Ldp()
self.link.add_feature(ldp)
self.link4.add_feature(ldp)
ldp.hello_interval = 88
ldp.device_attr['PE2'].hello_holdtime = 99
# ==Test per interface config===
out = ldp.build_config(apply=False)
if 1:
self.assertMultiLineEqual(
str(out['PE1']), '\n'.join([
'mpls label protocol ldp',
'mpls ip',
'mpls ldp discovery hello interval 88',
'interface GigabitEthernet0/0/1',
' mpls ip',
' exit',
'interface GigabitEthernet0/0/7',
' mpls ip',
' exit',
]))
self.assertMultiLineEqual(
str(out['PE2']), '\n'.join([
'mpls label protocol ldp',
'mpls ip',
'mpls ldp discovery hello interval 88',
'mpls ldp discovery hello holdtime 99',
'interface GigabitEthernet0/0/2',
' mpls ip',
' exit',
'interface GigabitEthernet0/0/8',
' mpls ip',
' exit',
]))
# Check intf unconfig - intf <GigabitEthernet0/0/0/8> config filter
out = ldp.build_unconfig(
apply=False,
attributes='device_attr__*__interface_attr__GigabitEthernet0/0/8')
self.assertMultiLineEqual(
str(out['PE2']), '\n'.join([
'interface GigabitEthernet0/0/8',
' no mpls ip',
' exit',
]))
def test_4_per_neighbor(self):
ldp = Ldp()
self.link.add_feature(ldp)
nbr1 = IPv4LsrNeighbor('1.2.3.4:0')
nbr2 = IPv4LsrNeighbor('1.2.3.5:0')
nbr3 = IPv4LsrNeighbor('1.2.3.6:0')
ldp.device_attr['PE1'].password = '******'
ldp.device_attr['PE2'].password = '******'
ldp.device_attr['PE1'].vrf_attr['default'].neighbors = [
nbr1, nbr2, nbr3
]
ldp.device_attr['PE1'].vrf_attr['default'].neighbor_attr[
nbr2].password = '******'
ldp.device_attr['PE1'].vrf_attr['default'].neighbor_attr[
nbr3].password = '******'
# ==Test per neighbor config==
out = ldp.build_config(apply=False)
self.assertMultiLineEqual(
str(out['PE1']), '\n'.join([
'mpls label protocol ldp',
'mpls ip',
'mpls ldp neighbor 1.2.3.4 password password1',
'mpls ldp neighbor 1.2.3.5 password blah',
'mpls ldp neighbor 1.2.3.6 password 060506324F41',
'interface GigabitEthernet0/0/1',
' mpls ip',
' exit',
]))
self.assertMultiLineEqual(
str(out['PE2']), '\n'.join([
'mpls label protocol ldp',
'mpls ip',
'interface GigabitEthernet0/0/2',
' mpls ip',
' exit',
]))
# Check nbr config - nbr <1.2.3.5:0'> config filter
out = ldp.build_config(
apply=False,
attributes='device_attr__*__vrf_attr__*__neighbor_attr__1.2.3.5:0')
self.assertMultiLineEqual(
str(out['PE1']), '\n'.join([
'mpls ldp neighbor 1.2.3.5 password blah',
]))
out = ldp.build_unconfig(
apply=False,
attributes=
'device_attr__*__vrf_attr__*__neighbor_attr__1.2.3.5:0__password')
self.assertMultiLineEqual(
str(out['PE1']), '\n'.join([
'no mpls ldp neighbor 1.2.3.5 password blah',
]))
out = ldp.build_unconfig(
apply=False,
attributes=
'device_attr__*__vrf_attr__*__neighbor_attr__1.2.3.6:0__password')
self.assertMultiLineEqual(
str(out['PE1']), '\n'.join([
'no mpls ldp neighbor 1.2.3.6 password 060506324F41',
]))
def test_8_per_vrf_per_neighbor(self):
ldp = Ldp()
self.link.add_feature(ldp)
vrf = Vrf(name='vrf1')
ldp.add_force_vrf(vrf)
nbr1 = IPv4LsrNeighbor('1.2.3.4:0')
nbr2 = IPv4LsrNeighbor('1.2.3.5:0')
nbr3 = IPv4LsrNeighbor('1.2.3.6:0')
ldp.device_attr['PE1'].vrf_attr['vrf1'].neighbors = [nbr1, nbr2, nbr3]
ldp.device_attr['PE1'].vrf_attr['vrf1'].neighbor_attr[
'1.2.3.5:0'].password = '******'
ldp.device_attr['PE1'].vrf_attr['vrf1'].neighbor_attr[
'1.2.3.6:0'].password = '******'
# ==Test per vrf per neighbor config==
out = ldp.build_config(apply=False)
self.assertMultiLineEqual(
str(out['PE1']), '\n'.join([
'mpls label protocol ldp',
'mpls ip',
'mpls ldp neighbor vrf vrf1 1.2.3.5 password blah22',
'mpls ldp neighbor vrf vrf1 1.2.3.6 password 060506324F41',
'interface GigabitEthernet0/0/1',
' mpls ip',
' exit',
]))
def test_init(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxr')
intf1 = Interface(device=dev1, name='GigabitEthernet0/0/0/1')
intf2 = Interface(device=dev1, name='GigabitEthernet0/0/0/2')
dev2 = Device(testbed=testbed, name='PE2', os='iosxr')
intf3 = Interface(device=dev2, name='GigabitEthernet0/0/0/3')
intf4 = Interface(device=dev2, name='GigabitEthernet0/0/0/4')
link1 = Link(testbed=testbed, name='link1', interfaces=(intf1, intf3))
link2 = Link(testbed=testbed, name='link2', interfaces=(intf2, intf4))
with self.assertRaises(TypeError):
grp1 = IccpGroup()
grp1 = IccpGroup(group_id=1)
self.assertEqual(grp1.group_id, 1)
self.assertCountEqual(grp1.devices, [])
self.assertCountEqual(grp1.interfaces, [])
dev1.add_feature(grp1)
self.assertCountEqual(grp1.devices, [dev1])
self.assertCountEqual(grp1.interfaces, [])
cfgs = grp1.build_config(apply=False)
self.assertCountEqual(cfgs.keys(), [dev1.name])
self.assertMultiLineEqual(
str(cfgs[dev1.name]), '\n'.join([
'redundancy',
' iccp group 1',
' exit',
' exit',
]))
grp1.add_interface(intf1)
self.assertCountEqual(grp1.interfaces, [intf1])
self.assertCountEqual(grp1.devices, [dev1])
self.assertCountEqual(grp1.device_attr[dev1].interfaces, [intf1])
self.assertCountEqual(grp1.device_attr[dev2].interfaces, [])
cfgs = grp1.build_config(apply=False)
self.assertCountEqual(cfgs.keys(), [dev1.name])
self.assertMultiLineEqual(
str(cfgs[dev1.name]), '\n'.join([
'redundancy',
' iccp group 1',
' interface GigabitEthernet0/0/0/1',
' exit',
' exit',
' exit',
]))
dev2.add_feature(grp1)
grp1.add_interface(intf2)
grp1.add_interface(intf3)
grp1.mac_flush = 'stp-tcn'
grp1.device_attr[dev1].recovery_delay = 100
grp1.device_attr[intf1.device].interface_attr[intf1].primary_vlan = 1
grp1.device_attr[intf1.device].interface_attr[intf1].mac_flush = None
self.assertCountEqual(grp1.interfaces, [intf1, intf2, intf3])
self.assertCountEqual(grp1.devices, [dev1, dev2])
self.assertCountEqual(grp1.device_attr[dev1].interfaces,
[intf1, intf2])
self.assertCountEqual(grp1.device_attr[dev2].interfaces, [intf3])
cfgs = grp1.build_config(apply=False)
self.assertCountEqual(cfgs.keys(), [dev1.name, dev2.name])
self.assertMultiLineEqual(
str(cfgs[dev1.name]), '\n'.join([
'redundancy',
' iccp group 1',
' interface GigabitEthernet0/0/0/1',
' primary vlan 1',
' recovery delay 100',
' exit',
' interface GigabitEthernet0/0/0/2',
' mac-flush stp-tcn',
' recovery delay 100',
' exit',
' exit',
' exit',
]))
self.assertMultiLineEqual(
str(cfgs[dev2.name]), '\n'.join([
'redundancy',
' iccp group 1',
' interface GigabitEthernet0/0/0/3',
' mac-flush stp-tcn',
' exit',
' exit',
' exit',
]))
class test_rsvp(TestCase):
def setUp(self):
tb = Genie.testbed = Testbed()
self.dev1 = Device(testbed=tb, name='PE1', os='iosxr')
self.dev2 = Device(testbed=tb, name='PE2', os='iosxr')
self.i1 = Interface(name='GigabitEthernet0/0/0/1', device=self.dev1)
self.i2 = Interface(name='GigabitEthernet0/0/0/2', device=self.dev2)
self.i3 = Interface(name='GigabitEthernet0/0/0/3', device=self.dev1)
self.i4 = Interface(name='GigabitEthernet0/0/0/4', device=self.dev2)
self.i5 = Interface(name='GigabitEthernet0/0/0/5', device=self.dev1)
self.i6 = Interface(name='GigabitEthernet0/0/0/6', device=self.dev2)
self.i7 = Interface(name='GigabitEthernet0/0/0/7', device=self.dev1)
self.i8 = Interface(name='GigabitEthernet0/0/0/8', device=self.dev2)
self.link = Link(name='1_2_1')
self.link.connect_interface(interface=self.i1)
self.link.connect_interface(interface=self.i2)
self.link2 = Link(name='1_2_2')
self.link2.connect_interface(interface=self.i3)
self.link2.connect_interface(interface=self.i4)
self.link3 = Link(name='1_2_3')
self.link3.connect_interface(interface=self.i5)
self.link3.connect_interface(interface=self.i6)
self.link4 = Link(name='1_2_4')
self.link4.connect_interface(interface=self.i7)
self.link4.connect_interface(interface=self.i8)
self.assertCountEqual(self.link.find_interfaces(), [self.i1, self.i2])
self.assertCountEqual(self.dev1.find_interfaces(),
[self.i1, self.i3, self.i5, self.i7])
self.assertCountEqual(self.dev2.find_interfaces(),
[self.i2, self.i4, self.i6, self.i8])
def test_0_interface_only(self):
rsvp = Rsvp()
self.assertCountEqual(rsvp.devices, [])
self.assertCountEqual(rsvp.links, [])
self.assertCountEqual(rsvp.interfaces, [])
self.link.add_feature(rsvp)
self.link2.add_feature(rsvp)
self.link3.add_feature(rsvp)
self.link4.add_feature(rsvp)
self.assertCountEqual(rsvp.devices, [self.dev1, self.dev2])
self.assertCountEqual(rsvp.links,
[self.link, self.link2, self.link3, self.link4])
self.assertCountEqual(rsvp.interfaces, [
self.i1, self.i2, self.i3, self.i4, self.i5, self.i6, self.i7,
self.i8
])
rsvp.device_attr[self.i1.device].interface_attr[
self.i1].rdm_bw_total = None
rsvp.device_attr[self.i1.device].interface_attr[
self.i3].rdm_bw_cli_rdm_kw = False
rsvp.device_attr[self.i1.device].interface_attr[
self.i5].rdm_bw_cli_style = Rsvp.RdmBwCliStyle.bc0_bc1
rsvp.device_attr[self.i1.device].interface_attr[
self.i7].rdm_bw_cli_style = Rsvp.RdmBwCliStyle.global_subpool
cfgs = rsvp.build_config(apply=False)
self.assertMultiLineDictEqual(
cfgs, {
self.dev1.name:
'\n'.join([
'rsvp',
' interface GigabitEthernet0/0/0/1',
' exit',
' interface GigabitEthernet0/0/0/3',
' bandwidth 20000 20000',
' exit',
' interface GigabitEthernet0/0/0/5',
' bandwidth rdm bc0 20000 20000',
' exit',
' interface GigabitEthernet0/0/0/7',
' bandwidth rdm global-pool 20000 20000',
' exit',
' exit',
]),
self.dev2.name:
'\n'.join([
'rsvp',
' interface GigabitEthernet0/0/0/2',
' bandwidth rdm 20000 20000',
' exit',
' interface GigabitEthernet0/0/0/4',
' bandwidth rdm 20000 20000',
' exit',
' interface GigabitEthernet0/0/0/6',
' bandwidth rdm 20000 20000',
' exit',
' interface GigabitEthernet0/0/0/8',
' bandwidth rdm 20000 20000',
' exit',
' exit',
]),
})
def test_init(self):
Genie.testbed = Testbed()
dev1 = Device(name='PE1', os='iosxr')
intf1 = Interface(device=dev1, name='GigabitEthernet0/0/0/1')
intf2 = Interface(device=dev1, name='GigabitEthernet0/0/0/2')
dev2 = Device(name='PE2', os='iosxr')
intf3 = Interface(device=dev2, name='GigabitEthernet0/0/0/3')
intf4 = Interface(device=dev2, name='GigabitEthernet0/0/0/4')
tgen = Device(testbed=Genie.testbed, name='TGEN', os='spirent')
intf5 = Interface(device=tgen, name='1/1', intf_mode='ethernet')
intf6 = Interface(device=tgen, name='1/2', intf_mode='ethernet')
Link(name='PE1-TGEN-1', interfaces=(intf1, intf5))
Link(name='PE1-PE2-1', interfaces=(intf2, intf3))
Link(name='PE2-TGEN-1', interfaces=(intf4, intf6))
with self.assertRaises(AttributeError):
stream1 = Stream()
stream1 = Stream(source=intf5, destination=intf6)
self.assertIs(stream1.source_tgen_interface, intf5)
self.assertIs(stream1.device, tgen)
self.assertIs(stream1.testbed, Genie.testbed)
self.assertTypedEqual(stream1.tgen_handle, None)
self.assertTypedEqual(stream1.source, intf5)
self.assertTypedEqual(stream1.source_instance, None)
self.assertTypedEqual(stream1.source_count, 1)
self.assertTypedEqual(stream1.destination, intf6)
self.assertTypedEqual(stream1.destination_instance, None)
self.assertTypedEqual(stream1.destination_count, 1)
self.assertTypedEqual(stream1.destination_tgen_interfaces,
frozenset([stream1.destination]))
self.assertTypedEqual(stream1.layer2_protocol,
Stream.Layer2Protocol.ethernet_ii)
self.assertTypedEqual(stream1.source_mac_address, None)
self.assertTypedEqual(stream1.source_mac_address_count, 0)
self.assertTypedEqual(stream1.source_mac_address_step, None)
self.assertTypedEqual(stream1.destination_mac_address,
MAC('ffff.ffff.ffff'))
self.assertTypedEqual(stream1.destination_mac_address_count, 1)
self.assertTypedEqual(stream1.destination_mac_address_step, 1)
self.assertTypedEqual(stream1.eth_encap_val1, None)
self.assertTypedEqual(stream1.eth_encap_count1, 0)
self.assertTypedEqual(stream1.eth_encap_step1, None)
self.assertTypedEqual(stream1.eth_encap_range1, range(0))
self.assertTypedEqual(stream1.eth_encap_val2, None)
self.assertTypedEqual(stream1.eth_encap_count2, 0)
self.assertTypedEqual(stream1.eth_encap_step2, None)
self.assertTypedEqual(stream1.eth_encap_range2, range(0))
self.assertTypedEqual(stream1.mpls_labels, ())
self.assertTypedEqual(stream1.ip_version, None)
self.assertIs(stream1.layer3_protocol, None)
self.assertTypedEqual(stream1.source_ip, None)
self.assertTypedEqual(stream1.source_ip_count, 0)
self.assertTypedEqual(stream1.source_ip_step, None)
self.assertTypedEqual(stream1.destination_ip, None)
self.assertTypedEqual(stream1.destination_ip_count, 0)
self.assertTypedEqual(stream1.destination_ip_step, None)
self.assertIs(stream1.layer4_protocol, None)
self.assertTypedEqual(stream1.bandwidth, 10)
self.assertIs(stream1.bandwidth_units, Stream.BandwidthUnits.kbps)
self.assertTypedEqual(stream1.frame_length, 512)
self.assertTypedEqual(stream1.frame_length_mode,
Stream.FrameLengthMode.l2)
self.assertTypedEqual(stream1.name, None)
self.assertTypedEqual(stream1.obj_state, 'active')
self.assertTypedEqual(stream1.sub_stream_increments,
typedset(Stream.SubStreamIncrement, ()))
intf1.mac_address = 'baad.baad.beef'
intf5.mac_address = 'aaaa.bbbb.cccc'
intf5.eth_encap_val1 = 1001
stream1.eth_encap_range2 = range(2000, 3000, 2)
intf5.ipv4 = '1.2.3.1/24'
intf6.ipv4 = '1.2.3.2/24'
self.assertIs(stream1.source_tgen_interface, intf5)
self.assertIs(stream1.device, tgen)
self.assertIs(stream1.testbed, Genie.testbed)
self.assertTypedEqual(stream1.tgen_handle, None)
self.assertTypedEqual(stream1.source, intf5)
self.assertTypedEqual(stream1.source_instance, None)
self.assertTypedEqual(stream1.source_count, 1)
self.assertTypedEqual(stream1.destination, intf6)
self.assertTypedEqual(stream1.destination_instance, None)
self.assertTypedEqual(stream1.destination_count, 1)
self.assertTypedEqual(stream1.destination_tgen_interfaces,
frozenset([stream1.destination]))
self.assertTypedEqual(stream1.layer2_protocol,
Stream.Layer2Protocol.ethernet_ii)
self.assertTypedEqual(stream1.source_mac_address,
MAC('aaaa.bbbb.cccc'))
self.assertTypedEqual(stream1.source_mac_address_count, 1)
self.assertTypedEqual(stream1.source_mac_address_step, 1)
self.assertTypedEqual(stream1.destination_mac_address,
MAC('ffff.ffff.ffff'))
self.assertTypedEqual(stream1.destination_mac_address_count, 1)
self.assertTypedEqual(stream1.destination_mac_address_step, 1)
self.assertTypedEqual(stream1.eth_encap_val1, 1001)
self.assertTypedEqual(stream1.eth_encap_count1, 1)
self.assertTypedEqual(stream1.eth_encap_step1, 1)
self.assertTypedEqual(stream1.eth_encap_range1, range(1001, 1002))
self.assertTypedEqual(stream1.eth_encap_val2, 2000)
self.assertTypedEqual(stream1.eth_encap_count2, 500)
self.assertTypedEqual(stream1.eth_encap_step2, 2)
self.assertTypedEqual(stream1.eth_encap_range2, range(2000, 3000, 2))
self.assertTypedEqual(stream1.mpls_labels, ())
self.assertTypedEqual(stream1.ip_version, 4)
self.assertIs(stream1.layer3_protocol, Stream.Layer3Protocol.ipv4)
self.assertTypedEqual(stream1.source_ip, IPv4Address('1.2.3.1'))
self.assertTypedEqual(stream1.source_ip_count, 1)
self.assertTypedEqual(stream1.source_ip_step, 1)
self.assertTypedEqual(stream1.destination_ip, IPv4Address('1.2.3.2'))
self.assertTypedEqual(stream1.destination_ip_count, 1)
self.assertTypedEqual(stream1.destination_ip_step, 1)
self.assertIs(stream1.layer4_protocol, None)
self.assertTypedEqual(stream1.bandwidth, 10)
self.assertIs(stream1.bandwidth_units, Stream.BandwidthUnits.kbps)
self.assertTypedEqual(stream1.frame_length, 512)
self.assertTypedEqual(stream1.frame_length_mode,
Stream.FrameLengthMode.l2)
self.assertTypedEqual(stream1.name, None)
self.assertTypedEqual(stream1.obj_state, 'active')
self.assertTypedEqual(stream1.sub_stream_increments,
typedset(Stream.SubStreamIncrement, ()))
class test_isis(unittest.TestCase):
def setUp(self):
pass
testbed = Genie.testbed = Testbed()
self.dev1 = Device(testbed=testbed, name='PE1', os='iosxe')
self.dev2 = Device(testbed=testbed, name='PE2', os='iosxe')
self.intf1 = Interface(name='GigabitEthernet0/0/1', device=self.dev1)
self.intf2 = Interface(name='GigabitEthernet0/0/2', device=self.dev2)
self.link = Link(name='1_2_1', testbed=testbed)
self.link.connect_interface(interface=self.intf1)
self.link.connect_interface(interface=self.intf2)
self.assertSetEqual(set(self.link.find_interfaces()),
set([self.intf1, self.intf2]))
self.assertSetEqual(set(self.dev1.find_interfaces()),
set([self.intf1]))
self.assertSetEqual(set(self.dev2.find_interfaces()),
set([self.intf2]))
isis = Isis("core")
self.assertSetEqual(set(isis.devices), set([]))
self.assertSetEqual(set(isis.links), set([]))
self.link.add_feature(isis)
for intf_iter in self.link.interfaces:
intf_iter.add_feature(isis)
self.assertCountEqual(isis.devices, [self.dev1, self.dev2])
self.assertSetEqual(set(isis.links), set([self.link]))
self.assertSetEqual(set(isis.interfaces), set([self.intf1,
self.intf2]))
def test_IsisGlobal(self):
isis = Isis("core")
self.link.add_feature(isis)
# Global ISIS config
isis.nsr = True
# override default for both devices
isis.is_type = Isis.IsType.level_1
isis.device_attr['PE1'].is_type = Isis.IsType.level_1_2
isis.device_attr['PE2'].is_type = Isis.IsType.level_2
# use no default
isis.device_attr['PE1'].nsf = Isis.Nsf.ietf
isis.device_attr['PE2'].nsf = Isis.Nsf.cisco
# override default for one device
val = 1
isis.device_attr['PE1'].net_id = "00.0000.0000.000{}.00".format(val)
val += 1
isis.device_attr['PE2'].net_id = "00.0000.0000.000{}.00".format(val)
cfg1 = isis.build_config(apply=False)
self.assertCountEqual(cfg1.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(cfg1['PE1']), '\n'.join([
'router isis core',
' is-type level-1-2',
' nsf ietf',
' nsr',
' net 00.0000.0000.0001.00',
' exit',
'interface GigabitEthernet0/0/1',
' ip router isis core',
' exit',
]))
self.assertMultiLineEqual(
str(cfg1['PE2']), '\n'.join([
'router isis core',
' is-type level-2-only',
' nsf cisco',
' nsr',
' net 00.0000.0000.0002.00',
' exit',
'interface GigabitEthernet0/0/2',
' ip router isis core',
' exit',
]))
uncfg1 = isis.build_unconfig(apply=False)
self.assertCountEqual(uncfg1.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(uncfg1['PE1']), '\n'.join([
'no router isis core',
'interface GigabitEthernet0/0/1',
' no ip router isis core',
' exit',
]))
self.assertMultiLineEqual(
str(uncfg1['PE2']), '\n'.join([
'no router isis core',
'interface GigabitEthernet0/0/2',
' no ip router isis core',
' exit',
]))
partial_cfg1 = isis.build_config(apply=False,
attributes='device_attr__PE1__nsf')
self.assertCountEqual(partial_cfg1.keys(), ['PE1'])
self.assertMultiLineEqual(
str(partial_cfg1['PE1']), '\n'.join([
'router isis core',
' nsf ietf',
' exit',
]))
def test_IsisPerAddrFamily(self):
isis = Isis("core")
self.link.add_feature(isis)
val = 1
isis.device_attr['PE1'].net_id = "00.0000.0000.000{}.00".format(val)
val += 1
isis.device_attr['PE2'].net_id = "00.0000.0000.000{}.00".format(val)
# per address family ISIS
isis.device_attr['PE1'].address_family_attr[
'ipv4 unicast'].metric_style = "wide"
isis.device_attr['PE1'].address_family_attr[
'ipv4 unicast'].mpls_te_level = Isis.IsType.level_1
isis.device_attr['PE1'].address_family_attr[
'ipv4 unicast'].mpls_te_rtrid = self.intf1
isis.device_attr['PE2'].address_family_attr[
'ipv4 unicast'].metric_style = "narrow"
isis.device_attr['PE2'].address_family_attr[
'ipv4 unicast'].mpls_te_level = Isis.IsType.level_2
isis.device_attr['PE2'].address_family_attr[
'ipv4 unicast'].mpls_te_rtrid = self.intf2
cfg1 = isis.build_config(apply=False)
self.assertMultiLineEqual(
str(cfg1['PE1']), '\n'.join([
'router isis core',
' net 00.0000.0000.0001.00',
' metric-style wide',
' mpls traffic-eng level-1',
' mpls traffic-eng router-id GigabitEthernet0/0/1',
' exit',
'interface GigabitEthernet0/0/1',
' ip router isis core',
' exit',
]))
self.assertMultiLineEqual(
str(cfg1['PE2']), '\n'.join([
'router isis core',
' net 00.0000.0000.0002.00',
' metric-style narrow',
' mpls traffic-eng level-2',
' mpls traffic-eng router-id GigabitEthernet0/0/2',
' exit',
'interface GigabitEthernet0/0/2',
' ip router isis core',
' exit',
]))
isis.address_families = set(
[AddressFamily.ipv4_unicast, AddressFamily.ipv6_unicast])
isis.device_attr['PE1'].address_family_attr[
'ipv6 unicast'].metric_style = "wide"
isis.device_attr['PE2'].address_family_attr[
'ipv6 unicast'].metric_style = "narrow"
cfg2 = isis.build_config(apply=False)
self.assertMultiLineEqual(
str(cfg2['PE1']), '\n'.join([
'router isis core',
' net 00.0000.0000.0001.00',
' metric-style wide',
' mpls traffic-eng level-1',
' mpls traffic-eng router-id GigabitEthernet0/0/1',
' address-family ipv6 unicast',
' metric-style wide',
' exit',
' exit',
'interface GigabitEthernet0/0/1',
' ip router isis core',
' ipv6 router isis core',
' exit',
]))
self.assertMultiLineEqual(
str(cfg2['PE2']), '\n'.join([
'router isis core',
' net 00.0000.0000.0002.00',
' metric-style narrow',
' mpls traffic-eng level-2',
' mpls traffic-eng router-id GigabitEthernet0/0/2',
' address-family ipv6 unicast',
' metric-style narrow',
' exit',
' exit',
'interface GigabitEthernet0/0/2',
' ip router isis core',
' ipv6 router isis core',
' exit',
]))
uncfg2 = isis.build_unconfig(apply=False)
self.assertMultiLineEqual(
str(uncfg2['PE1']), '\n'.join([
'no router isis core',
'interface GigabitEthernet0/0/1',
' no ip router isis core',
' no ipv6 router isis core',
' exit',
]))
self.assertMultiLineEqual(
str(uncfg2['PE2']), '\n'.join([
'no router isis core',
'interface GigabitEthernet0/0/2',
' no ip router isis core',
' no ipv6 router isis core',
' exit',
]))
partial_uncfg2 = isis.build_unconfig(
apply=False, attributes='device_attr__*__address_family_attr__*')
self.assertMultiLineEqual(
str(partial_uncfg2['PE1']), '\n'.join([
'router isis core',
' no metric-style wide',
' no mpls traffic-eng level-1',
' no mpls traffic-eng router-id GigabitEthernet0/0/1',
' no address-family ipv4 unicast',
' no address-family ipv6 unicast',
' exit',
]))
self.assertMultiLineEqual(
str(partial_uncfg2['PE2']), '\n'.join([
'router isis core',
' no metric-style narrow',
' no mpls traffic-eng level-2',
' no mpls traffic-eng router-id GigabitEthernet0/0/2',
' no address-family ipv4 unicast',
' no address-family ipv6 unicast',
' exit',
]))
partial_cfg21 = isis.build_config(
apply=False,
attributes=
'device_attr__PE1__address_family_attr__ipv4 unicast__metric_style'
)
self.assertMultiLineEqual(
str(partial_cfg21['PE1']), '\n'.join([
'router isis core',
' metric-style wide',
' exit',
]))
partial_uncfg21 = isis.build_unconfig(
apply=False,
attributes=
'device_attr__PE1__address_family_attr__ipv4 unicast__metric_style'
)
self.assertMultiLineEqual(
str(partial_uncfg21['PE1']), '\n'.join([
'router isis core',
' no metric-style wide',
' exit',
]))
partial_cfg22 = isis.build_config(
apply=False,
attributes=
'device_attr__PE1__address_family_attr__ipv6 unicast__metric_style'
)
self.assertMultiLineEqual(
str(partial_cfg22['PE1']), '\n'.join([
'router isis core',
' address-family ipv6 unicast',
' metric-style wide',
' exit',
' exit',
]))
partial_uncfg22 = isis.build_unconfig(
apply=False,
attributes=
'device_attr__PE1__address_family_attr__ipv6 unicast__metric_style'
)
self.assertMultiLineEqual(
str(partial_uncfg22['PE1']), '\n'.join([
'router isis core',
' address-family ipv6 unicast',
' no metric-style wide',
' exit',
' exit',
]))
def test_IsisPerInterface(self):
isis = Isis("core")
self.link.add_feature(isis)
for intf_iter in self.link.interfaces:
intf_iter.add_feature(isis)
val = 1
isis.device_attr['PE1'].net_id = "00.0000.0000.000{}.00".format(val)
val += 1
isis.device_attr['PE2'].net_id = "00.0000.0000.000{}.00".format(val)
# per interface ISIS
isis.device_attr['PE1'].interface_attr[self.intf1].passive = True
isis.device_attr['PE1'].interface_attr[self.intf1].metric = 20
isis.device_attr['PE2'].interface_attr[
self.intf2].point_to_point = True
cfg1 = isis.build_config(apply=False)
self.assertMultiLineEqual(
str(cfg1['PE1']), '\n'.join([
'router isis core',
' net 00.0000.0000.0001.00',
' passive-interface GigabitEthernet0/0/1',
' exit',
'interface GigabitEthernet0/0/1',
' ip router isis core',
' isis metric 20',
' exit',
]))
self.assertMultiLineEqual(
str(cfg1['PE2']), '\n'.join([
'router isis core',
' net 00.0000.0000.0002.00',
' exit',
'interface GigabitEthernet0/0/2',
' ip router isis core',
' isis network point-to-point',
' exit',
]))
isis.address_families = set(
[AddressFamily.ipv4_unicast, AddressFamily.ipv6_unicast])
cfg2 = isis.build_config(apply=False)
self.assertMultiLineEqual(
str(cfg2['PE1']), '\n'.join([
'router isis core',
' net 00.0000.0000.0001.00',
' passive-interface GigabitEthernet0/0/1',
' exit',
'interface GigabitEthernet0/0/1',
' ip router isis core',
' ipv6 router isis core',
' isis metric 20',
' exit',
]))
self.assertMultiLineEqual(
str(cfg2['PE2']), '\n'.join([
'router isis core',
' net 00.0000.0000.0002.00',
' exit',
'interface GigabitEthernet0/0/2',
' ip router isis core',
' ipv6 router isis core',
' isis network point-to-point',
' exit',
]))
partial_cfg21 = isis.build_config(
apply=False,
attributes='device_attr__*__interface_attr__*__metric')
self.assertMultiLineEqual(
str(partial_cfg21['PE1']), '\n'.join([
'interface GigabitEthernet0/0/1',
' isis metric 20',
' exit',
]))
partial_cfg22 = isis.build_config(
apply=False,
attributes=
'device_attr__*__interface_attr__*__address_family_attr__ipv6 unicast'
)
self.assertMultiLineEqual(
str(partial_cfg22['PE1']), '\n'.join([
'interface GigabitEthernet0/0/1',
' ipv6 router isis core',
' exit',
]))
class test_vrf(TestCase):
def setUp(self):
tb = Genie.testbed = Testbed()
self.dev1 = Device(testbed=tb, name='PE1', os='iosxr')
self.dev2 = Device(testbed=tb, name='PE2', os='iosxr')
self.i1 = Interface(name='GigabitEthernet0/0/0/1', device=self.dev1)
self.i2 = Interface(name='GigabitEthernet0/0/0/2', device=self.dev2)
self.i3 = Interface(name='GigabitEthernet0/0/0/3', device=self.dev1)
self.i4 = Interface(name='GigabitEthernet0/0/0/4', device=self.dev2)
self.i5 = Interface(name='GigabitEthernet0/0/0/5', device=self.dev1)
self.i6 = Interface(name='GigabitEthernet0/0/0/6', device=self.dev2)
self.i7 = Interface(name='GigabitEthernet0/0/0/7', device=self.dev1)
self.i8 = Interface(name='GigabitEthernet0/0/0/8', device=self.dev2)
self.link = Link(name='1_2_1', testbed=tb)
self.link.connect_interface(interface=self.i1)
self.link.connect_interface(interface=self.i2)
self.link2 = Link(name='1_2_2', testbed=tb)
self.link2.connect_interface(interface=self.i3)
self.link2.connect_interface(interface=self.i4)
self.link3 = Link(name='1_2_3', testbed=tb)
self.link3.connect_interface(interface=self.i5)
self.link3.connect_interface(interface=self.i6)
self.link4 = Link(name='1_2_4', testbed=tb)
self.link4.connect_interface(interface=self.i7)
self.link4.connect_interface(interface=self.i8)
self.assertSetEqual(set(self.link.find_interfaces()),
set([self.i1, self.i2]))
self.assertSetEqual(set(self.dev1.find_interfaces()),
set([self.i1, self.i3, self.i5, self.i7]))
self.assertSetEqual(set(self.dev2.find_interfaces()),
set([self.i2, self.i4, self.i6, self.i8]))
def test_init(self):
with self.subTest('name'):
with self.assertRaises(TypeError):
vrf = Vrf()
with self.assertRaises(AssertionError):
vrf = Vrf(name=123)
vrf = Vrf(name='red')
vrf2 = Vrf(name='blue')
with self.subTest('fallback_vrf'):
# fallback_vrf's type is defined post Vrf class; Make sure it works.
self.assertIs(vrf.fallback_vrf, None)
with self.assertRaises(ValueError):
vrf.fallback_vrf = 123
self.assertIs(vrf.fallback_vrf, None)
with self.assertRaises(ValueError):
vrf.fallback_vrf = 'blue'
self.assertIs(vrf.fallback_vrf, None)
vrf.fallback_vrf = None
self.assertIs(vrf.fallback_vrf, None)
vrf.fallback_vrf = vrf2
self.assertIs(vrf.fallback_vrf, vrf2)
def test_1_top_level(self):
vrf = Vrf(name='vrf1')
self.dev1.add_feature(vrf)
self.dev2.add_feature(vrf)
vrf.device_attr['PE1'].scale_mode = 'big'
vrf.description = 'my description'
cfgs = vrf.build_config(apply=False)
self.assertMultiLineDictEqual(
cfgs, {
self.dev1.name:
'\n'.join([
'vrf vrf1',
' description my description',
' mode big',
' exit',
]),
self.dev2.name:
'\n'.join([
'vrf vrf1',
' description my description',
' exit',
]),
})
cfgs = vrf.build_config(apply=False,
attributes='device_attr__*__description')
self.assertMultiLineDictEqual(
cfgs, {
self.dev1.name:
'\n'.join([
'vrf vrf1',
' description my description',
' exit',
]),
self.dev2.name:
'\n'.join([
'vrf vrf1',
' description my description',
' exit',
]),
})
cfgs = vrf.build_config(apply=False,
attributes={
'device_attr': {
'*': {
'description': None,
},
},
})
self.assertMultiLineDictEqual(
cfgs, {
self.dev1.name:
'\n'.join([
'vrf vrf1',
' description my description',
' exit',
]),
self.dev2.name:
'\n'.join([
'vrf vrf1',
' description my description',
' exit',
]),
})
cfgs = vrf.build_config(apply=False,
devices=[self.dev2],
attributes={
'device_attr': {
'*': {
'description': None,
},
},
})
self.assertMultiLineDictEqual(
cfgs, {
self.dev2.name:
'\n'.join([
'vrf vrf1',
' description my description',
' exit',
]),
})
cfgs = vrf.build_config(apply=False,
devices=[self.dev2],
attributes={
'device_attr': {
self.dev1: {
'description': None,
},
},
})
self.assertMultiLineDictEqual(cfgs, {})
cfgs = vrf.build_config(apply=False,
devices=[self.dev2],
attributes={
'device_attr': {
self.dev2: {
'description': None,
},
},
})
self.assertMultiLineDictEqual(
cfgs, {
self.dev2.name:
'\n'.join([
'vrf vrf1',
' description my description',
' exit',
]),
})
cfgs = vrf.build_config(apply=False,
devices=[self.dev2],
attributes={
'device_attr': {
self.dev2.name: {
'description': None,
},
},
})
self.assertMultiLineDictEqual(
cfgs, {
self.dev2.name:
'\n'.join([
'vrf vrf1',
' description my description',
' exit',
]),
})
def test_4_per_af(self):
vrf = Vrf(name='vrf1')
self.dev1.add_feature(vrf)
vrf.address_families |= {AddressFamily.ipv6_unicast}
vrf.device_attr['PE1'].address_family_attr[
'ipv4 unicast'].export_route_targets = [
RouteTarget.ImportExport('100:200', stitching=True),
]
cfgs = vrf.build_config(apply=False)
self.assertMultiLineDictEqual(
cfgs, {
self.dev1.name:
'\n'.join([
'vrf vrf1',
' address-family ipv4 unicast',
' export route-target 100:200 stitching',
' exit',
' exit',
]),
})
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='iosxr')
dev2 = Device(testbed=testbed, name='PE2', os='iosxr')
intf1 = Interface(name='GigabitEthernet0/0/1', device=dev1)
intf2 = Interface(name='GigabitEthernet0/0/2', device=dev2)
intf3 = Interface(name='GigabitEthernet0/0/3', 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)
# Defining attributes section
vlan.device_attr[dev1]
vlan.device_attr[dev2]
vlan.device_attr[dev1].interface_attr[intf1]
vlan.device_attr[dev1].interface_attr[intf1].eth_encap_type1 = 'dot1q'
vlan.device_attr[dev1].interface_attr[intf1].eth_encap_val1 = 2
vlan.device_attr[dev1].interface_attr[
intf1].eth_encap_type2 = 'second-dot1q'
vlan.device_attr[dev1].interface_attr[intf1].eth_encap_val2 = 5
# Unconfig testing
# Set a mock
dev1.cli = Mock()
dev1.configure = Mock()
dev2.cli = Mock()
dev2.configure = Mock()
dev1.add_feature(vlan)
dev2.add_feature(vlan)
# Mock config
output = vlan.build_config(apply=True)
uncfg = vlan.build_unconfig(apply=False)
self.assertMultiLineEqual(
str(uncfg['PE1']), '\n'.join([
'interface GigabitEthernet0/0/1',
' no encapsulation dot1q 2 second-dot1q 5',
' 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 encapsulation dot1q 2 second-dot1q 5',
' exit',
]))
partial_vlan_interface_uncfg = vlan.build_unconfig(apply=False,
attributes={'device_attr':\
{'*':{'interface_attr':\
{'*':"eth_encap_type1"}}}})
self.assertCountEqual(partial_vlan_interface_uncfg.keys(),
['PE1', 'PE2'])
self.assertMultiLineEqual(
str(partial_vlan_interface_uncfg['PE1']), '\n'.join([
'interface GigabitEthernet0/0/1',
' no encapsulation dot1q 2 second-dot1q 5',
' exit',
]))
class test_rsvp(TestCase):
def setUp(self):
tb = Genie.testbed = Testbed()
self.dev1 = Device(testbed=tb, name='PE1', os='iosxe')
self.dev2 = Device(testbed=tb, name='PE2', os='iosxe')
self.i1 = Interface(name='GigabitEthernet0/0/1', device=self.dev1)
self.i2 = Interface(name='GigabitEthernet0/0/2', device=self.dev2)
self.i3 = Interface(name='GigabitEthernet0/0/3', device=self.dev1)
self.i4 = Interface(name='GigabitEthernet0/0/4', device=self.dev2)
self.i5 = Interface(name='GigabitEthernet0/0/5', device=self.dev1)
self.i6 = Interface(name='GigabitEthernet0/0/6', device=self.dev2)
self.i7 = Interface(name='GigabitEthernet0/0/7', device=self.dev1)
self.i8 = Interface(name='GigabitEthernet0/0/8', device=self.dev2)
self.link = Link(name='1_2_1')
self.link.connect_interface(interface=self.i1)
self.link.connect_interface(interface=self.i2)
self.link2 = Link(name='1_2_2')
self.link2.connect_interface(interface=self.i3)
self.link2.connect_interface(interface=self.i4)
self.link3 = Link(name='1_2_3')
self.link3.connect_interface(interface=self.i5)
self.link3.connect_interface(interface=self.i6)
self.link4 = Link(name='1_2_4')
self.link4.connect_interface(interface=self.i7)
self.link4.connect_interface(interface=self.i8)
self.assertCountEqual(self.link.find_interfaces(), [self.i1, self.i2])
self.assertCountEqual(self.dev1.find_interfaces(),
[self.i1, self.i3, self.i5, self.i7])
self.assertCountEqual(self.dev2.find_interfaces(),
[self.i2, self.i4, self.i6, self.i8])
def test_0_interface_only(self):
rsvp = Rsvp()
self.assertCountEqual(rsvp.devices, [])
self.assertCountEqual(rsvp.links, [])
self.assertCountEqual(rsvp.interfaces, [])
self.link.add_feature(rsvp)
self.link2.add_feature(rsvp)
self.link3.add_feature(rsvp)
self.link4.add_feature(rsvp)
self.assertCountEqual(rsvp.devices, [self.dev1, self.dev2])
self.assertCountEqual(rsvp.links,
[self.link, self.link2, self.link3, self.link4])
self.assertCountEqual(rsvp.interfaces, [
self.i1, self.i2, self.i3, self.i4, self.i5, self.i6, self.i7,
self.i8
])
rsvp.enable_default_bw = True
cfgs = rsvp.build_config(apply=False)
self.assertMultiLineEqual(
str(cfgs['PE1']), '\n'.join([
'interface GigabitEthernet0/0/1',
' ip rsvp bandwidth',
' exit',
'interface GigabitEthernet0/0/3',
' ip rsvp bandwidth',
' exit',
'interface GigabitEthernet0/0/5',
' ip rsvp bandwidth',
' exit',
'interface GigabitEthernet0/0/7',
' ip rsvp bandwidth',
' exit',
]))
self.assertMultiLineEqual(
str(cfgs['PE2']), '\n'.join([
'interface GigabitEthernet0/0/2',
' ip rsvp bandwidth',
' exit',
'interface GigabitEthernet0/0/4',
' ip rsvp bandwidth',
' exit',
'interface GigabitEthernet0/0/6',
' ip rsvp bandwidth',
' exit',
'interface GigabitEthernet0/0/8',
' ip rsvp bandwidth',
' exit',
]))
uncfgs = rsvp.build_unconfig(apply=False)
self.assertMultiLineEqual(
str(uncfgs['PE1']), '\n'.join([
'interface GigabitEthernet0/0/1',
' no ip rsvp bandwidth',
' exit',
'interface GigabitEthernet0/0/3',
' no ip rsvp bandwidth',
' exit',
'interface GigabitEthernet0/0/5',
' no ip rsvp bandwidth',
' exit',
'interface GigabitEthernet0/0/7',
' no ip rsvp bandwidth',
' exit',
]))
self.assertMultiLineEqual(
str(uncfgs['PE2']), '\n'.join([
'interface GigabitEthernet0/0/2',
' no ip rsvp bandwidth',
' exit',
'interface GigabitEthernet0/0/4',
' no ip rsvp bandwidth',
' exit',
'interface GigabitEthernet0/0/6',
' no ip rsvp bandwidth',
' exit',
'interface GigabitEthernet0/0/8',
' no ip rsvp bandwidth',
' exit',
]))
def test_basic_uncfg(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='nxos')
dev2 = Device(testbed=testbed, name='PE2', os='nxos')
intf1 = Interface(name='Ethernet0/0/1', device=dev1, layer=Layer.L3)
intf2 = Interface(name='Ethernet0/0/2', device=dev2, layer=Layer.L2)
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].access_map_attr[access_map_id]
vlan.device_attr[dev2].access_map_attr[access_map_id]
vlan.device_attr[dev1].interface_attr[intf1]
vlan.device_attr[dev2].interface_attr[intf2]
vlan.device_attr[dev2].interface_attr[intf2].switchport_mode = \
L2_type.TRUNK
vlan.device_attr[dev2].interface_attr[intf2].sw_trunk_allowed_vlan = \
'200-201'
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.egress_port_channel_load_balance_random = True
vlan.device_attr[dev1].access_map_action = 'drop'
vlan.datalink_flow_monitor = True
# Unconfig testing
# Set a mock
dev1.cli = Mock()
dev1.configure = Mock()
dev2.cli = Mock()
dev2.configure = Mock()
dev1.add_feature(vlan)
dev2.add_feature(vlan)
# Mock config
output = vlan.build_config(apply=True)
uncfg1 = vlan.build_unconfig(apply=False)
self.assertCountEqual(uncfg1.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(uncfg1['PE1']),
'\n'.join([
'no vlan 100',
'no vlan access-map ed',
'no vlan configuration 3',
]))
self.assertMultiLineEqual(
str(uncfg1['PE2']),
'\n'.join([
'no vlan 100',
'no vlan access-map ed',
'no vlan configuration 3',
'interface Ethernet0/0/2',
' no switchport mode trunk',
' no switchport trunk allowed vlan 200-201',
' 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['PE2']),
'\n'.join([
'interface Ethernet0/0/2',
' 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['PE2']),
'\n'.join([
'interface Ethernet0/0/2',
' no switchport trunk allowed vlan 200-201',
' exit',
]))
def test_init(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='iosxe')
intf1 = Interface(device=dev1, name='GigabitEthernet0/0/1')
intf2 = Interface(device=dev1, name='GigabitEthernet0/0/2')
dev2 = Device(testbed=testbed, name='PE2', os='iosxe')
intf3 = Interface(device=dev2, name='GigabitEthernet0/0/3')
intf4 = Interface(device=dev2, name='GigabitEthernet0/0/4')
link1 = Link(testbed=testbed, name='link1', interfaces=(intf1, intf3))
link2 = Link(testbed=testbed, name='link2', interfaces=(intf2, intf4))
with self.assertRaises(TypeError):
xc1 = Xconnect()
xc1 = Xconnect(name='xc1')
self.assertIs(xc1.xconnect_type, Xconnect.Type.p2p)
self.assertEqual(xc1.name, 'xc1')
self.assertCountEqual(xc1.devices, [])
self.assertCountEqual(xc1.interfaces, [])
self.assertCountEqual(xc1.segments, [])
self.assertCountEqual(xc1.link.interfaces, [])
dev1.add_feature(xc1)
self.assertCountEqual(xc1.devices, [dev1])
self.assertCountEqual(xc1.interfaces, [])
self.assertCountEqual(xc1.segments, [])
self.assertCountEqual(xc1.link.interfaces, [])
cfgs = xc1.build_config(apply=False)
self.assertCountEqual(cfgs.keys(), [dev1.name])
self.assertMultiLineEqual(
str(cfgs[dev1.name]), '\n'.join([
'l2vpn xconnect context xc1',
' exit',
]))
efp = Interface(device=dev1,
name='GigabitEthernet0/0/1.20',
service_instance=20)
efp.encapsulation = 'dot1q'
efp.rewrite_ingress = 'pop 1 symmetric'
xc1.add_segment(efp)
self.assertCountEqual(xc1.interfaces, [efp])
self.assertCountEqual(xc1.devices, [dev1])
self.assertCountEqual(xc1.segments, [efp])
self.assertCountEqual(xc1.device_attr[dev1].interfaces, [efp])
self.assertCountEqual(xc1.device_attr[dev2].interfaces, [])
self.assertCountEqual(xc1.device_attr[dev1].segments, [efp])
self.assertCountEqual(xc1.device_attr[dev2].segments, [])
cfgs = xc1.build_config(apply=False)
self.assertCountEqual(cfgs.keys(), [dev1.name])
self.assertMultiLineEqual(
str(cfgs[dev1.name]), '\n'.join([
'l2vpn xconnect context xc1',
' member GigabitEthernet0/0/1 service-instance 20',
' exit',
]))
def test_basic_cfg(self):
testbed = Genie.testbed = Testbed()
dev1 = Device(testbed=testbed, name='PE1', os='nxos')
dev2 = Device(testbed=testbed, name='PE2', os='nxos')
intf1 = Interface(name='Ethernet0/0/1', device=dev1, layer=Layer.L3)
intf2 = Interface(name='Ethernet0/0/2', device=dev2, layer=Layer.L2)
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]
vlan.device_attr[dev2].interface_attr[intf2]
vlan.device_attr[dev2].interface_attr[intf2].switchport_mode = \
L2_type.TRUNK
vlan.device_attr[dev2].interface_attr[intf2].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.egress_load_balance = True
vlan.device_attr[dev1].access_map_action = 'drop'
vlan.datalink_flow_monitor = True
intf1.ipv4 = '201.0.12.1'
intf1.ipv4.netmask = '255.255.255.0'
intf1.speed = 1000
intf1.mtu = 500
intf1.ipv6 = '2001::12:1'
cfg1 = vlan.build_config(apply=False)
cfg2 = intf1.build_config(apply=False)
self.assertCountEqual(cfg1.keys(), ['PE1', 'PE2'])
self.assertMultiLineEqual(
str(cfg1['PE1']),
'\n'.join([
'vlan 100',
' media enet',
' no shutdown',
' exit',
'vlan access-map ed',
' action drop',
' exit',
'vlan configuration 3',
' egress port-channel load-balance random',
' exit',
'interface Ethernet0/0/1',
' mtu 500',
' ip address 201.0.12.1 255.255.255.0',
' ipv6 address 2001::12:1/128',
' speed 1000',
' exit',
]))
self.assertMultiLineEqual(
str(cfg1['PE2']),
'\n'.join([
'vlan 100',
' media enet',
' no shutdown',
' exit',
'vlan access-map ed',
' exit',
'vlan configuration 3',
' egress port-channel load-balance random',
' exit',
'interface Ethernet0/0/2',
' switchport mode trunk',
' switchport trunk allowed vlan 200-201',
' exit',
]))
self.assertMultiLineEqual(
str(cfg2),
'\n'.join([
'interface Ethernet0/0/1',
' mtu 500',
' ip address 201.0.12.1 255.255.255.0',
' ipv6 address 2001::12:1/128',
' speed 1000',
' exit',
]))
