def connect(d1, d2, dmd, layers=None, update_catalog=True):
''' Connect two devices by l2 link '''
mac1 = random_mac()
mac2 = random_mac()
add_interface(d1, macaddress=mac1, clientmacs=[mac2], vlans=layers)
add_interface(d2, macaddress=mac2, clientmacs=[mac1], vlans=layers)
if update_catalog:
catapi = CatalogAPI(dmd.zport)
catapi.add_node(d1)
catapi.add_node(d2)
def test_get_vlan_connections_with_unaware_node(self):
a = get_device('a', self.dmd)
mac_a = random_mac()
mac_b = random_mac()
b = get_device('b', self.dmd)
# make a look like a switch
add_interface(a, macaddress=mac_a, clientmacs=[mac_b], vlans=['vlan1'])
# make b look like a server
add_interface(b, macaddress=mac_b, clientmacs=[], vlans=[])
catapi = CatalogAPI(self.dmd.zport)
catapi.add_node(a)
catapi.add_node(b)
res = get_connections(a, depth=3, layers=['vlan1'])
links = str(res['links'])
self.assertIn("{'color': ('layer2', u'vlan1'), 'directed': False",
links)
self.assertIn("{'color': ('layer2', u'vlan1'), 'directed': True",
links)
self.assertIn("{'color': ('layer2',), 'directed': False", links)
class TestCatalogAPI(BaseTestCase):
def afterSetUp(self):
super(TestCatalogAPI, self).afterSetUp()
self.cat = CatalogAPI(self.app.zport, name="test_l2")
self.cat.clear()
self.connection = fake_connection('test_id')
def test_catalog_is_empty(self):
self.assertEqual(len(self.cat.search()), 0)
def test_connection_is_added_to_catalog(self):
self.cat.add_connection(self.connection)
brains = self.cat.search()
self.assertEqual(len(brains), 3)
self.assertTrue('test_id' in [x.entity_id for x in brains])
self.assertEqual(brains[0].layers, ('layer1', 'layer2', 'layer1'))
def test_remove_connection(self):
self.cat.add_connection(self.connection)
self.cat.remove_connection(self.connection)
brains = self.cat.search()
self.assertEqual(len(brains), 0)
def test_validate_connection(self):
self.assertEqual(self.cat.validate_connection(self.connection),
self.connection)
def test_search(self):
self.cat.add_connection(self.connection)
self.assertEqual(len(self.cat.search(entity_id='test_id')), 1)
self.assertEqual(len(self.cat.search(entity_id='incorrect_id')), 0)
def test_add_remove_node(self):
cp = fake_connections_provider(self.dmd)
self.assertEqual(len(self.cat.search()), 0)
self.cat.add_node(cp)
self.assertEqual(len(self.cat.search(entity_id='connection_id')), 1)
self.cat.remove_node(cp)
self.assertEqual(len(self.cat.search(entity_id='connection_id')), 0)
def test_get_directly_connected(self):
self.cat.add_node(fake_connections_provider(self.dmd))
self.assertTrue(
list(self.cat.get_directly_connected('connection_id')) ==
['connected_to1', 'connected_to2'])
def test_get_reverse_connected(self):
self.cat.add_node(fake_connections_provider(self.dmd))
self.assertTrue(
list(self.cat.get_reverse_connected('connected_to1')) ==
['connection_id'])
def test_get_existing_layers(self):
self.cat.add_node(fake_connections_provider(self.dmd))
self.assertEqual(len(self.cat.get_existing_layers()), 2)
def test_clear(self):
self.cat.add_node(fake_connections_provider(self.dmd, 'con_id1'))
self.cat.add_node(fake_connections_provider(self.dmd, 'con_id2'))
self.cat.add_node(fake_connections_provider(self.dmd, 'con_id3'))
self.cat.add_node(fake_connections_provider(self.dmd, 'con_id4'))
self.cat.add_node(fake_connections_provider(self.dmd, 'con_id5'))
self.assertEqual(len(self.cat.search()), 15) # 5*3
self.cat.clear()
self.assertEqual(len(self.cat.search()), 0)
class TestCheckWorkingPath(BaseTestCase):
def afterSetUp(self):
super(TestCheckWorkingPath, self).afterSetUp()
self.dmd.Devices.createOrganizer('/Network/Router/Cisco')
self.cat = CatalogAPI(self.dmd.zport, name='test_l2')
self.cat.clear()
zcml.load_config('testing.zcml', Products.ZenTestCase)
zcml.load_config('configure.zcml', ZenPacks.zenoss.Layer2)
def topology(self, topology):
devices = create_topology(topology, self.dmd, False)
for device in devices:
self.cat.add_node(device)
def test_check_nearest_down(self):
self.topology('''
a b
b c
''')
self.cat.get_status = lambda x: x != router('b')
self.assertFalse(self.cat.check_working_path(router('a'), router('c')))
def test_check_next_nearest_down(self):
self.topology('''
a b
b c
c d
''')
self.cat.get_status = lambda x: x != router('c')
self.assertFalse(self.cat.check_working_path(router('a'), router('d')))
def test_check_one_way_down(self):
self.topology('''
a b
a c
c d
''')
self.cat.get_status = lambda x: x != router('c')
self.assertFalse(self.cat.check_working_path(router('a'), router('d')))
def test_check_two_ways_down(self):
self.topology('''
a b
a c
b d
c d
''')
self.cat.get_status = lambda x: x in (router('a'), router('d'))
self.assertFalse(self.cat.check_working_path(router('a'), router('d')))
def test_get_bsf_connected(self):
self.topology('''
a b
b c
b d
c d
''')
self.assertItemsEqual(
self.cat.get_bfs_connected(router('a'),
self.cat.get_directly_connected, 6),
[router('c'), router('d')])