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(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
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',
]))
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',
]))
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',
]))
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',
]),
})
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_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',
]),
})
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',
]))
