def test_check_dhcp_with_vlans(self, make_listeners, send_discover):
config_sample = {
'eth0': (100, 101),
'eth1': (100, 102)
}
api.check_dhcp_with_vlans(config_sample, timeout=1)
make_listeners.assert_called_once_with(('eth1', 'eth0'))
self.assertEqual(send_discover.call_count, 6)
def test_check_dhcp_with_vlans(self, make_listeners, send_discover):
config_sample = {
'eth0': (100, 101),
'eth1': (100, 102)
}
api.check_dhcp_with_vlans(config_sample, timeout=1)
make_listeners.assert_called_once_with(('eth1', 'eth0'))
self.assertEqual(send_discover.call_count, 2)
def test_check_dhcp_with_vlans_repeat_2(self, make_listeners,
send_discover, sleep_mock):
config_sample = {
'eth0': (),
}
api.check_dhcp_with_vlans(config_sample, timeout=1, repeat=3)
self.assertEqual(sleep_mock.call_count, 3)
make_listeners.assert_called_once_with(('eth0',))
self.assertEqual(send_discover.call_count, 3)
def test_check_dhcp_with_vlans(self, check_dhcp):
config_sample = {
'eth0': (100, 101),
'eth1': (100, 102)
}
api.check_dhcp_with_vlans(config_sample)
check_dhcp.assert_called_once_with(
['eth1', 'eth1.100', 'eth1.102', 'eth0', 'eth0.100', 'eth0.101'],
repeat=2, timeout=5)
def test_check_dhcp_with_vlans_repeat_2(self, make_listeners,
send_discover, sleep_mock):
config_sample = {
'eth0': (),
}
api.check_dhcp_with_vlans(config_sample, timeout=1, repeat=3)
self.assertEqual(sleep_mock.call_count, 3)
make_listeners.assert_called_once_with(('eth0', ))
self.assertEqual(send_discover.call_count, 3)
def take_action(self, parsed_args):
res = api.check_dhcp_with_vlans(json.loads(parsed_args.config),
timeout=parsed_args.timeout,
repeat=parsed_args.repeat)
first = res.next()
columns = first.keys()
return columns, [first.values()] + [item.values() for item in res]
def take_action(self, parsed_args):
res = api.check_dhcp_with_vlans(json.loads(parsed_args.config),
timeout=parsed_args.timeout,
repeat=parsed_args.repeat)
first = res.next()
columns = first.keys()
return columns, [first.values()] + [item.values() for item in res]
def test_with_vlans(self):
config = {
'eth0': (100, 101),
'eth1': (103, 105),
'eth2': range(106, 120)
}
result = api.check_dhcp_with_vlans(config)
self.assertEqual(len(list(result)), 3)
def take_action(self, parsed_args):
res = list(api.check_dhcp_with_vlans(json.loads(parsed_args.config),
timeout=parsed_args.timeout,
repeat=parsed_args.repeat,
w_vlans=parsed_args.with_vlans))
if not res:
res = [{}]
return (utils.DHCP_OFFER_COLUMNS,
[utils.get_item_properties(item, utils.DHCP_OFFER_COLUMNS)
for item in res])
def test_with_vlans(self):
config = {'eth0': (100, 101), 'eth1': (103, 105),
'eth2': range(106, 120)}
result = api.check_dhcp_with_vlans(config)
self.assertEqual(len(list(result)), 3)
def test_check_dhcp_wo_vlans(self, make_listeners, send_discover, *_):
api.check_dhcp_with_vlans(self.config_sample, timeout=1, repeat=1, w_vlans=False)
make_listeners.assert_called_once_with(("eth1", "eth0"))
self.assertEqual(send_discover.call_count, 2)