def test_no_changes(self):
old = BatmanParser(iulinet)
new = BatmanParser(iulinet)
result = diff(old, new)
self.assertIsInstance(result, dict)
self.assertIsNone(result['added'])
self.assertIsNone(result['removed'])
def test_json_dict(self):
p = BatmanParser(iulinet)
data = p.json(dict=True)
self.assertIsInstance(data, dict)
self.assertEqual(data['type'], 'NetworkGraph')
self.assertEqual(data['protocol'], 'batman-adv')
self.assertEqual(data['version'], '2014.3.0')
self.assertEqual(data['metric'], 'TQ')
self.assertIsInstance(data['nodes'], list)
self.assertIsInstance(data['links'], list)
self.assertEqual(len(data['nodes']), 5)
self.assertEqual(len(data['links']), 4)
self.assertIsInstance(data['links'][0]['cost'], float)
# ensure additional node properties are present
found = False
for node in data['nodes']:
if node['id'] == '90:f6:52:f2:8c:2c':
self.assertIsInstance(node['local_addresses'], list)
self.assertIsInstance(node['properties']['clients'], list)
found = True
break
self.assertTrue(found)
found = False
# ensure local_addresses not present if empty
for node in data['nodes']:
if node['id'] == 'a0:f3:c1:96:94:06':
self.assertFalse('local_addresses' in node)
found = True
break
self.assertTrue(found)
def test_json_dict(self):
p = BatmanParser(iulinet)
data = p.json(dict=True)
self.assertIsInstance(data, dict)
self.assertEqual(data['type'], 'NetworkGraph')
self.assertEqual(data['protocol'], 'batman-adv')
self.assertEqual(data['version'], '2014.3.0')
self.assertEqual(data['metric'], 'TQ')
self.assertIsInstance(data['nodes'], list)
self.assertIsInstance(data['links'], list)
self.assertEqual(len(data['nodes']), 5)
self.assertEqual(len(data['links']), 4)
self.assertIsInstance(data['links'][0]['cost'], float)
# ensure additional node properties are present
found = False
for node in data['nodes']:
if node['id'] == '90:f6:52:f2:8c:2c':
self.assertIsInstance(node['local_addresses'], list)
self.assertIsInstance(node['properties']['clients'], list)
found = True
break
self.assertTrue(found)
found = False
# ensure local_addresses not present if empty
for node in data['nodes']:
if node['id'] == 'a0:f3:c1:96:94:06':
self.assertFalse('local_addresses' in node)
found = True
break
self.assertTrue(found)
def test_json_dict(self):
p = BatmanParser(iulinet)
data = p.json(dict=True)
self.assertIsInstance(data, dict)
self.assertEqual(data['type'], 'NetworkGraph')
self.assertEqual(data['protocol'], 'batman-adv')
self.assertEqual(data['version'], '2014.3.0')
self.assertEqual(data['metric'], 'TQ')
self.assertIsInstance(data['nodes'], list)
self.assertIsInstance(data['links'], list)
self.assertEqual(len(data['nodes']), 5)
self.assertEqual(len(data['links']), 4)
def test_json_string(self):
p = BatmanParser(iulinet)
data = p.json()
self.assertIsInstance(data, six.string_types)
self.assertIn('NetworkGraph', data)
self.assertIn('protocol', data)
self.assertIn('version', data)
self.assertIn('metric', data)
self.assertIn('batman-adv', data)
self.assertIn('2014.3.0', data)
self.assertIn('TQ', data)
self.assertIn('links', data)
self.assertIn('nodes', data)
def test_json_dict(self):
p = BatmanParser(iulinet)
data = p.json(dict=True)
self.assertIsInstance(data, dict)
self.assertEqual(data['type'], 'NetworkGraph')
self.assertEqual(data['protocol'], 'batman-adv')
self.assertEqual(data['version'], '2015.0')
self.assertEqual(data['metric'], 'TQ')
self.assertIsInstance(data['nodes'], list)
self.assertIsInstance(data['links'], list)
self.assertEqual(len(data['nodes']), 5)
self.assertEqual(len(data['links']), 4)
self.assertIsInstance(data['links'][0]['cost'], float)
def test_json_string(self):
p = BatmanParser(iulinet)
data = p.json()
self.assertIsInstance(data, six.string_types)
self.assertIn('NetworkGraph', data)
self.assertIn('protocol', data)
self.assertIn('version', data)
self.assertIn('metric', data)
self.assertIn('batman-adv', data)
self.assertIn('2014.3.0', data)
self.assertIn('TQ', data)
self.assertIn('links', data)
self.assertIn('nodes', data)
def test_added_removed_1_node(self):
old = BatmanParser(iulinet)
new = BatmanParser(iulinet2)
result = diff(old, new)
self.assertIsInstance(result, dict)
self.assertTrue(type(result['added']['links']) is list)
self.assertTrue(type(result['removed']['links']) is list)
# ensure there are no differences
self.assertEqual(len(result['added']['links']), 1)
self.assertEqual(len(result['removed']['links']), 1)
self._test_expected_links(graph=result['added'],
expected_links=[('a0:f3:c1:96:94:10',
'90:f6:52:f2:8c:2c')])
self._test_expected_links(graph=result['removed'],
expected_links=[('a0:f3:c1:96:94:06',
'90:f6:52:f2:8c:2c')])
def test_parse(self):
p = BatmanParser(iulinet)
self.assertIsInstance(p.graph, networkx.Graph)
self.assertEqual(len(p.graph.nodes()), 5)
self.assertEqual(len(p.graph.edges()), 4)
properties = list(p.graph.edges(data=True))[0][2]
self.assertIsInstance(properties['weight'], float)
def test_changed_links(self):
old = BatmanParser(iulinet)
new = BatmanParser(iulinet2)
result = diff(old, new)
self.assertEqual(result['changed']['nodes'], [])
self.assertEqual(len(result['changed']['links']), 2)
link = result['changed']['links'][0]
self.assertEqual(link['source'], '10:fe:ed:37:3a:39')
self.assertEqual(link['target'], 'a0:f3:c1:ac:6c:44')
self.assertEqual(link['cost'], 1.0)
self.assertEqual(link['cost_text'], '')
self.assertEqual(link['properties'], {})
link = result['changed']['links'][1]
self.assertEqual(link['source'], '90:f6:52:f2:8c:2c')
self.assertEqual(link['target'], '10:fe:ed:37:3a:39')
self.assertEqual(link['cost'], 1.0)
self.assertEqual(link['cost_text'], '')
self.assertEqual(link['properties'], {})
def test_parse(self):
p = BatmanParser(iulinet)
self.assertIsInstance(p.graph, networkx.Graph)
properties = list(p.graph.edges(data=True))[0][2]
self.assertIsInstance(properties['weight'], float)
# test additional properties in nodes of networkx graph
properties = list(p.graph.nodes(data=True))[0][1]
self.assertIsInstance(properties['local_addresses'], list)
self.assertIsInstance(properties['clients'], list)
def test_parse_exception(self):
with self.assertRaises(ParserError):
BatmanParser('WRONG')
def test_get_primary_address_ValueError(self):
p = BatmanParser(iulinet)
with self.assertRaises(ValueError):
p._get_primary_address('wrong', [['aa:bb:cc:dd:ee:ff']])
def test_duplicated(self):
nodup = BatmanParser(iulinet)
dup = BatmanParser(duplicated)
# nodup and dup have the same amount of nodes and edges
self.assertEqual(len(nodup.graph.edges()), len(dup.graph.edges()))
self.assertEqual(len(nodup.graph.nodes()), len(dup.graph.nodes()))
def test_get_primary_address_ValueError(self):
p = BatmanParser(iulinet)
r = p._get_primary_address('bb:aa:cc:dd:ee:ff',
[['aa:bb:cc:dd:ee:ff']])
self.assertEqual(r, 'bb:aa:cc:dd:ee:ff')
def test_get_primary_address_ValueError(self):
p = BatmanParser(iulinet)
r = p._get_primary_address('bb:aa:cc:dd:ee:ff', [['aa:bb:cc:dd:ee:ff']])
self.assertEqual(r, 'bb:aa:cc:dd:ee:ff')
def test_parse_exception(self):
with self.assertRaises(ParserError):
BatmanParser('{ "test": "test" }')
def test_parse_exception2(self):
with self.assertRaises(ParserError):
BatmanParser('{ "topology": [{ "a": "a" }] }')
