def test_create_dhcp_host_twice(self):
"""Create DHCP host twice; raise AlreadyExists."""
dhcp_host = "testcreatehosttwice"
host = SpokeDHCPHost(self.dhcp_server, self.dhcp_group)
host.create(dhcp_host)
self.assertRaises(error.AlreadyExists, host.create, dhcp_host)
host.delete(dhcp_host)
def test_create_dhcp_host_twice(self):
"""Create DHCP host twice; raise AlreadyExists."""
dhcp_host = 'testcreatehosttwice'
host = SpokeDHCPHost(self.dhcp_server, self.dhcp_group)
host.create(dhcp_host)
self.assertRaises(error.AlreadyExists, host.create, dhcp_host)
host.delete(dhcp_host)
def test_delete_dhcp_group_before_hosts(self):
"""Delete DHCP group while hosts present, raise SaveTheBabies."""
dhcp_host = "testdeletebeforehost"
host = SpokeDHCPHost(self.dhcp_server, self.dhcp_group)
host.create(dhcp_host)
group = SpokeDHCPGroup(self.dhcp_server)
self.assertRaises(error.SaveTheBabies, group.delete, self.dhcp_group)
host.delete(dhcp_host)
def test_delete_dhcp_group_before_hosts(self):
"""Delete DHCP group while hosts present, raise SaveTheBabies."""
dhcp_host = 'testdeletebeforehost'
host = SpokeDHCPHost(self.dhcp_server, self.dhcp_group)
host.create(dhcp_host)
group = SpokeDHCPGroup(self.dhcp_server)
self.assertRaises(error.SaveTheBabies, group.delete, self.dhcp_group)
host.delete(dhcp_host)
def tearDown(self):
host = SpokeDHCPHost(self.dhcp_server, self.dhcp_group)
host.delete(self.dhcp_host)
group = SpokeDHCPGroup(self.dhcp_server)
group.delete(self.dhcp_group)
dhcp = SpokeDHCPService()
dhcp.delete(self.dhcp_server)
server = SpokeDHCPServer()
server.delete(self.dhcp_server)
def tearDown(self):
host = SpokeDHCPHost(self.dhcp_server, self.dhcp_group)
host.delete(self.dhcp_host)
group = SpokeDHCPGroup(self.dhcp_server)
group.delete(self.dhcp_group)
dhcp = SpokeDHCPService()
dhcp.delete(self.dhcp_server)
server = SpokeDHCPServer()
server.delete(self.dhcp_server)
def test_create_dhcp_host(self):
"""Create DHCP host; return results object."""
dhcp_host = "testcreatehost"
host = SpokeDHCPHost(self.dhcp_server, self.dhcp_group)
result = host.create(dhcp_host)["data"]
service_name = self.dhcp_server + self.dhcp_conf_suffix
group_base_dn = "cn=%s,cn=%s,%s" % (self.dhcp_group, service_name, self.base_dn)
dn = "cn=%s,%s" % (dhcp_host, group_base_dn)
dn_info = {"objectClass": ["top", self.dhcp_host_class, self.dhcp_options_class], "cn": [dhcp_host]}
expected_result = [(dn, dn_info)]
self.assertEqual(result, expected_result)
host.delete(dhcp_host)
def test_create_dhcp_host(self):
"""Create DHCP host; return results object."""
dhcp_host = 'testcreatehost'
host = SpokeDHCPHost(self.dhcp_server, self.dhcp_group)
result = host.create(dhcp_host)['data']
service_name = self.dhcp_server + self.dhcp_conf_suffix
group_base_dn = 'cn=%s,cn=%s,%s' % (self.dhcp_group, service_name,
self.base_dn)
dn = 'cn=%s,%s' % (dhcp_host, group_base_dn)
dn_info = {'objectClass': ['top', self.dhcp_host_class,
self.dhcp_options_class],
'cn': [dhcp_host]}
expected_result = [(dn, dn_info)]
self.assertEqual(result, expected_result)
host.delete(dhcp_host)
def reservation_delete(dhcp_server, dhcp_group, host_name):
try:
from spoke.lib.dhcp import SpokeDHCPHost
host = SpokeDHCPHost(dhcp_server, dhcp_group)
result = host.delete(host_name)
except error.SpokeError as e:
result = common.handle_error(e)
return result
def reservation_delete(dhcp_server, dhcp_group, host_name):
try:
from spoke.lib.dhcp import SpokeDHCPHost
host = SpokeDHCPHost(dhcp_server, dhcp_group)
result = host.delete(host_name)
except error.SpokeError as e:
result = common.handle_error(e)
return result
def test_create_dhcp_host(self):
"""Create DHCP host; return results object."""
dhcp_host = 'testcreatehost'
host = SpokeDHCPHost(self.dhcp_server, self.dhcp_group)
result = host.create(dhcp_host)['data']
service_name = self.dhcp_server + self.dhcp_conf_suffix
group_base_dn = 'cn=%s,cn=%s,%s' % (self.dhcp_group, service_name,
self.base_dn)
dn = 'cn=%s,%s' % (dhcp_host, group_base_dn)
dn_info = {
'objectClass':
['top', self.dhcp_host_class, self.dhcp_options_class],
'cn': [dhcp_host]
}
expected_result = [(dn, dn_info)]
self.assertEqual(result, expected_result)
host.delete(dhcp_host)
def reservation_delete(host_name, dhcp_server=None, dhcp_group=None):
try:
conf = _spoke_config(_salt_config('config'))
if not dhcp_server:
dhcp_server = conf.get('DHCP', 'dhcp_def_server')
if not dhcp_group:
dhcp_group = conf.get('DHCP', 'dhcp_def_group')
from spoke.lib.dhcp import SpokeDHCPHost
host = SpokeDHCPHost(dhcp_server, dhcp_group)
result = host.delete(host_name)
except error.SpokeError as e:
result = common.handle_error(e)
return result
mc.info('Adding DHCP attributes for host %s' % hostname)
attr = SpokeDHCPAttr(server, group, hostname)
attr.create("dhcpHWAddress", "ethernet %s" % mac)
attr.create("dhcpStatements", "fixed-address %s" % ip)
attr.create("dhcpOption", "host-name \"%s\"" % hostname)
mc.data = host.get(hostname)
except error.SpokeError, e:
mc.fail(e.msg, e.exit_code)
elif request['action'] == 'search':
try:
hostname = request['data']['hostname']
except KeyError:
hostname = None
try:
mc.data = host.get(hostname)
attrs = mc.data['data'][0][1]
mc.mac = attrs['dhcpHWAddress'][0].split()[1]
mc.ip = attrs['dhcpStatements'][0].split()[1]
except error.SpokeError, e:
mc.fail(e.msg, e.exit_code)
elif request['action'] == 'delete':
try:
mc.data = host.delete(hostname)
except error.SpokeError, e:
mc.fail(e.msg, e.exit_code)
else:
msg = "Unknown action: " + request['action']
mc.fail(msg, 2)
log.info('Result via Mcollective: %s' % mc.data)
sys.exit(0)
mc.info('Adding DHCP attributes for host %s' % hostname)
attr = SpokeDHCPAttr(server, group, hostname)
attr.create("dhcpHWAddress", "ethernet %s" % mac )
attr.create("dhcpStatements", "fixed-address %s" %ip )
attr.create("dhcpOption", "host-name \"%s\"" %hostname )
mc.data = host.get(hostname)
except error.SpokeError, e:
mc.fail(e.msg, e.exit_code)
elif request['action'] == 'search':
try:
hostname = request['data']['hostname']
except KeyError:
hostname = None
try:
mc.data = host.get(hostname)
attrs = mc.data['data'][0][1]
mc.mac = attrs['dhcpHWAddress'][0].split()[1]
mc.ip = attrs['dhcpStatements'][0].split()[1]
except error.SpokeError, e:
mc.fail(e.msg, e.exit_code)
elif request['action'] == 'delete':
try:
mc.data = host.delete(hostname)
except error.SpokeError, e:
mc.fail(e.msg, e.exit_code)
else:
msg = "Unknown action: " + request['action']
mc.fail(msg, 2)
log.info('Result via Mcollective: %s' % mc.data)
sys.exit(0)
def test_delete_dhcp_host(self):
"""Delete DHCP host, return True."""
dhcp_host = "testdeletehost"
host = SpokeDHCPHost(self.dhcp_server, self.dhcp_group)
host.create(dhcp_host)
self.assertTrue(host.delete(dhcp_host))
def test_delete_dhcp_host(self):
"""Delete DHCP host, return True."""
dhcp_host = 'testdeletehost'
host = SpokeDHCPHost(self.dhcp_server, self.dhcp_group)
host.create(dhcp_host)
self.assertTrue(host.delete(dhcp_host))
