1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
#!/usr/bin/python3
import json
import shutil
import logging
import unittest
import ipaddress
import subprocess
from staslib import iputil, log, trid
IP = shutil.which('ip')
class Test(unittest.TestCase):
'''iputil.py unit tests'''
def setUp(self):
log.init(syslog=False)
self.logger = logging.getLogger()
self.logger.setLevel(logging.INFO)
# Retrieve the list of Interfaces and all the associated IP addresses
# using standard bash utility (ip address). We'll use this to make sure
# iputil.get_interface() returns the same data as "ip address".
try:
cmd = [IP, '-j', 'address', 'show']
p = subprocess.run(cmd, stdout=subprocess.PIPE, check=True)
self.ifaces = json.loads(p.stdout.decode().strip())
except subprocess.CalledProcessError:
self.ifaces = []
def test_get_interface(self):
'''Check that get_interface() returns the right info'''
for iface in self.ifaces:
for addr_entry in iface['addr_info']:
addr = ipaddress.ip_address(addr_entry['local'])
# Link local addresses may appear on more than one interface and therefore cannot be used.
if not addr.is_link_local:
self.assertEqual(iface['ifname'], iputil.get_interface(str(addr)))
self.assertEqual('', iputil.get_interface('255.255.255.255'))
def test_remove_invalid_addresses(self):
good_tcp = trid.TID({'transport': 'tcp', 'traddr': '1.1.1.1', 'subsysnqn': '', 'trsvcid': '8009'})
bad_tcp = trid.TID({'transport': 'tcp', 'traddr': '555.555.555.555', 'subsysnqn': '', 'trsvcid': '8009'})
any_fc = trid.TID({'transport': 'fc', 'traddr': 'blah', 'subsysnqn': ''})
bad_trtype = trid.TID({'transport': 'whatever', 'traddr': 'blah', 'subsysnqn': ''})
l1 = [
good_tcp,
bad_tcp,
any_fc,
bad_trtype,
]
l2 = iputil.remove_invalid_addresses(l1)
self.assertNotEqual(l1, l2)
self.assertIn(good_tcp, l2)
self.assertIn(any_fc, l2) # We currently don't check for invalid FC (all FCs are allowed)
self.assertNotIn(bad_tcp, l2)
self.assertNotIn(bad_trtype, l2)
if __name__ == "__main__":
unittest.main()
|
