Viewing file:  test_network.py (3.93 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
import testtools
import json
from probert.network import Network, NetworkInfo
from probert.tests.fakes import FAKE_PROBE_ALL_JSON
class ProbertTestNetwork(testtools.TestCase):
def setUp(self):
super(ProbertTestNetwork, self).setUp()
self.results = json.load(open(FAKE_PROBE_ALL_JSON))
self.network = Network(results=self.results)
def test_network_init(self):
self.assertNotEqual(None, self.network)
def test_network_get_interfaces(self):
ifaces = self.results['network'].keys()
self.assertEqual(sorted(ifaces), sorted(self.network.get_interfaces()))
def test_network_get_interfaces_no_nics(self):
ifaces = []
n = Network()
self.assertEqual(ifaces, n.get_interfaces())
def test_network_get_ips(self):
for iface in self.network.get_interfaces():
ip = self.results['network'][iface]['ip']
self.assertEqual(ip, self.network.get_ips(iface))
def test_network_get_ips_no_ips(self):
n = Network()
self.assertEqual([], n.get_ips('noiface'))
def test_network_get_hwaddr(self):
for iface in self.network.get_interfaces():
hwaddr = \
self.results['network'][iface]['hardware']['attrs']['address']
self.assertEqual(hwaddr, self.network.get_hwaddr(iface))
def test_network_get_iface_type(self):
# TODO: mock out open/read of sysfs
# and use _get_iface_type()
self.assertEqual('eth', self.network.get_iface_type('eth0'))
# needs mocking of pyudev.Context()
# and return mock data
#def test_network_probe(self):
class ProbertTestNetworkInfo(testtools.TestCase):
''' properties:
.name = eth7
.type = eth
.vendor = Innotec
.model = SuperSonicEtherRocket
.driver = sser
.devpath = /devices
.hwaddr = aa:bb:cc:dd:ee:ff
.addr = 10.2.7.2
.netmask = 255.255.255.0
.broadcast = 10.2.7.255
.addr6 =
.is_virtual =
.raw = {raw dictionary}
.ip = {ip dict}
.bond = {bond dict}
.bridge = {bridge_dict}
'''
def setUp(self):
super(ProbertTestNetworkInfo, self).setUp()
self.results = json.load(open(FAKE_PROBE_ALL_JSON))
def test_networkinfo_init(self):
probe_data = {
'em1': {
'bond': {
'is_slave': False,
'is_master': False,
'slaves': [],
'mode': None,
},
"bridge": {
"interfaces": [],
"is_bridge": False,
"is_port": False,
"options": {}
},
'hardware': {
'attrs': {
'address': '00:11:22:33:44:55',
}
},
'type': 'eth',
'ip' : {},
}
}
ni = NetworkInfo(probe_data)
self.assertNotEqual(ni, None)
def test_networkinfo_attributes(self):
eth0 = {'eth0': self.results.get('network').get('eth0')}
ni = NetworkInfo(probe_data=eth0)
props = {
'name': 'eth0',
'type': 'eth',
'vendor': 'ASIX Electronics Corp.',
'model': 'AX88179',
'driver': 'ax88179_178a',
'devpath': '/devices/pci0000:00/0000:00:14.0/usb3/3-2/3-2.1/3-2.1.1/3-2.1.1:1.0/net/eth0',
'hwaddr': '00:0a:cd:26:45:33',
'addr': '192.168.11.58',
'netmask': '255.255.255.0',
'broadcast': '192.168.11.255',
'is_virtual': False,
'raw': eth0.get('eth0'),
'bond': eth0.get('eth0').get('bond'),
'bridge': eth0.get('eth0').get('bridge'),
'ip': eth0.get('eth0').get('ip'),
}
for (prop, value) in props.items():
self.assertEqual(getattr(ni, prop), value)
|