class Test_HostsModel(Common):
"""Unit tests for HostsModel"""
model = None
def setup(self):
self.model = HostsModel()
self.db.q("TRUNCATE TABLE `hosts`")
self.db.q("TRUNCATE TABLE `ips`")
def test_get_id_by_name(self):
self.db.q("INSERT INTO `hosts` (id, project_id, ip_id, name, descr) VALUES (2,1,1,'test.com', '')")
assert self.model.get_id_by_name(1, 'test.com') == 2
def test_exists(self):
self.db.q("INSERT INTO `hosts` (id, project_id, ip_id, name, descr) VALUES (2,1,1,'test.com', '')")
assert self.model.exists(1, 'test.com')
assert not self.model.exists(1, 'test1.com')
def test_add(self):
assert not self.model.exists(1, 'test.com')
assert self.model.add(1, 1, 'test.com', '') == 1
assert self.model.exists(1, 'test.com')
def test_list(self):
self.db.q("INSERT INTO `ips` (id, project_id, ip, descr) VALUES (1, 1, '111.111.111.111', '')")
self.db.q("INSERT INTO `hosts` (id, project_id, ip_id, name, descr) VALUES (1, 1, 1, 'test1.com', 'desc1')")
self.db.q("INSERT INTO `hosts` (id, project_id, ip_id, name, descr) VALUES (2, 1, 1, 'test2.com', 'desc2')")
self.db.q("INSERT INTO `hosts` (id, project_id, ip_id, name, descr) VALUES (3, 1, 1, 'test3.com', 'desc3')")
test_data = [
{'id': 1, 'name': 'test1.com', 'ip': '111.111.111.111', 'descr': 'desc1'},
{'id': 2, 'name': 'test2.com', 'ip': '111.111.111.111', 'descr': 'desc2'},
{'id': 3, 'name': 'test3.com', 'ip': '111.111.111.111', 'descr': 'desc3'},
]
assert test_data == self.model.list(1, '111.111.111.111')
def test_list_of_names(self):
self.db.q("INSERT INTO `ips` (id, project_id, ip, descr) VALUES (1, 1, '111.111.111.111', '')")
self.db.q("INSERT INTO `hosts` (id, project_id, ip_id, name, descr) VALUES (1, 1, 1, 'test1.com', 'desc1')")
self.db.q("INSERT INTO `hosts` (id, project_id, ip_id, name, descr) VALUES (2, 1, 1, 'test2.com', 'desc2')")
self.db.q("INSERT INTO `hosts` (id, project_id, ip_id, name, descr) VALUES (3, 1, 1, 'test3.com', 'desc3')")
test_data = ['test1.com', 'test2.com', 'test3.com']
assert test_data == self.model.list_of_names(1)
def test_delete(self):
self.db.q("INSERT INTO `hosts` (id, project_id, ip_id, name, descr) VALUES (2,1,1,'test.com', '')")
assert self.model.exists(1, 'test.com')
self.model.delete(1, 'test.com')
assert not self.model.exists(1, 'test.com')
def _insert_hosts(self, hosts):
""" Insert found hosts in db """
pid = Registry().get('pData')['id']
H = HostsModel()
I = IpsModel()
added = 0
for host in hosts:
ip_id = I.get_id_or_add(pid, host['ip'])
if H.add(pid, ip_id, host['name'], founder='pre'):
added += 1
return added
def _insert_hosts(self, hosts):
""" Insert found hosts in db """
pid = Registry().get('pData')['id']
H = HostsModel()
I = IpsModel()
added = 0
for host in hosts:
ip_id = I.get_id_or_add(pid, host['ip'])
if H.add(pid, ip_id, host['name'], founder='pre'):
added += 1
return added
def _insert_hosts(self, hosts):
""" Add found hosts in db """
pid = Registry().get('pData')['id']
Hosts = HostsModel()
Ips = IpsModel()
added = 0
for host in hosts:
ip_id = Ips.get_id_or_add(pid, host['ip'])
if Hosts.add(pid, ip_id, host['name'], founder='dnsbrute'):
added += 1
return added
def _insert_hosts(self, hosts):
""" Add found hosts in db """
pid = Registry().get('pData')['id']
Hosts = HostsModel()
Ips = IpsModel()
added = 0
for host in hosts:
ip_id = Ips.get_id_or_add(pid, host['ip'])
if Hosts.add(pid, ip_id, host['name'], founder='dnsbrute'):
added += 1
return added
class Hosts(WSModule):
""" Class of Hosts module """
model = None
log_path = '/dev/null'
options = {}
options_sets = {
"list": {
"ip": WSOption("ip", "IP for host list", "", False, ['--ip']),
},
"delete": {
"host": WSOption("host", "Host for delete", "", True, ['--host']),
},
"add": {
"host":
WSOption("host", "Host for add", "", True, ['--host']),
"ip":
WSOption("ip", "Custom IP for this host", "", False, ['--ip']),
"descr":
WSOption("descr", "Description of host", "", False, ['--descr'])
}
}
def __init__(self, kernel):
WSModule.__init__(self, kernel)
self.model = HostsModel()
def validate_main(self):
""" Check users params """
if not validate_host(self.options['host'].value):
raise WSException("'{0}' is not valid host name!".format(
self.options['host'].value))
if 'ip' in self.options and self.options['ip'].value:
if not validate_ip(self.options['ip'].value):
raise WSException("IP '{0}' is not valid ip-address!".format(
self.options['ip'].value))
def run(self, action):
""" Method of run the module """
WSModule.run(self, action)
self.done = True
def add_action(self):
""" Action add of module """
self.validate_main()
pData = Registry().get('pData')
name = self.options['host'].value
descr = self.options['descr'].value
if self.options['ip'].value:
ip = self.options['ip'].value
else:
try:
ip = socket.gethostbyname(name)
print " IP for host '{0}' is '{1}'".format(name, ip)
except socket.gaierror:
raise WSException(
"Can`t lookup hostname '{0}'. Check it or set ip-address in --ip param!"
.format(name))
IPs = IpsModel()
ip_exists = IPs.exists(pData['id'], ip)
ip_id = IPs.get_id_or_add(pData['id'], ip)
if not ip_exists:
print " IP '{0}' was automatically added to project '{1}'".format(
ip, pData['name'])
if self.model.exists(pData['id'], name):
raise WSException(
"Host '{0}' already exists in project '{1}'!".format(
name, pData['name']))
self.model.add(pData['id'], ip_id, name, descr)
print " Host '{0}' successfully added to project '{1}' with IP '{2}' ! ".format(
name, pData['name'], ip)
def list_action(self):
""" Action list of module """
if self.options['ip'].value:
print "{0:=^51}".format("")
print "|{0: ^49}|".format("Hosts for IP '{0}'".format(
self.options['ip'].value))
print "{0:=^51}".format("")
print "| {0: ^23}| {1: ^23}|".format('Title', 'Description')
print "{0:=^51}".format("")
for host in self.model.list(Registry().get('pData')['id'],
self.options['ip'].value):
print "| {0: <23}| {1: <23}|".format(host['name'],
host['descr'])
print "{0:=^51}".format("")
else:
print "{0:=^76}".format("")
print "|{0: ^74}|".format("All host for project '{0}'".format(
Registry().get('pData')['name']))
print "{0:=^76}".format("")
print "| {0: ^23}| {1: ^23}| {2: ^23}|".format(
'Title', 'Description', 'IP')
print "{0:=^76}".format("")
for host in self.model.list_without_ip(
Registry().get('pData')['id']):
print "| {0: <23}| {1: <23}| {2: <23}|".format(
host['name'], host['descr'], host['ip'])
print "{0:=^76}".format("")
def delete_action(self):
""" Delete action of module """
self.validate_main()
name = self.options['host'].value
if not self.model.exists(Registry().get('pData')['id'], name):
raise WSException(
"Host '{0}' not exists in this project!".format(name))
answer = raw_input(
"You really want to delete host '{0}' [y/n]? ".format(name))
if answer.lower() == 'y':
self.model.delete(Registry().get('pData')['id'], name)
print "Host '{0}' successfully deleted.".format(name)
else:
print "Host '{0}' not deleted.".format(name)
class Hosts(WSModule):
""" Class of Hosts module """
model = None
log_path = '/dev/null'
options = {}
options_sets = {
"list": {
"ip": WSOption("ip", "IP for host list", "", False, ['--ip']),
},
"delete": {
"host": WSOption("host", "Host for delete", "", True, ['--host']),
},
"add": {
"host": WSOption("host", "Host for add", "", True, ['--host']),
"ip": WSOption("ip", "Custom IP for this host", "", False, ['--ip']),
"descr": WSOption("descr", "Description of host", "", False, ['--descr'])
}
}
def __init__(self, kernel):
WSModule.__init__(self, kernel)
self.model = HostsModel()
def validate_main(self):
""" Check users params """
if not validate_host(self.options['host'].value):
raise WSException("'{0}' is not valid host name!".format(self.options['host'].value))
if 'ip' in self.options and self.options['ip'].value:
if not validate_ip(self.options['ip'].value):
raise WSException("IP '{0}' is not valid ip-address!".format(self.options['ip'].value))
def run(self, action):
""" Method of run the module """
WSModule.run(self, action)
self.done = True
def add_action(self):
""" Action add of module """
self.validate_main()
pData = Registry().get('pData')
name = self.options['host'].value
descr = self.options['descr'].value
if self.options['ip'].value:
ip = self.options['ip'].value
else:
try:
ip = socket.gethostbyname(name)
print " IP for host '{0}' is '{1}'".format(name, ip)
except socket.gaierror:
raise WSException("Can`t lookup hostname '{0}'. Check it or set ip-address in --ip param!".format(name))
IPs = IpsModel()
ip_exists = IPs.exists(pData['id'], ip)
ip_id = IPs.get_id_or_add(pData['id'], ip)
if not ip_exists:
print " IP '{0}' was automatically added to project '{1}'".format(ip, pData['name'])
if self.model.exists(pData['id'], name):
raise WSException("Host '{0}' already exists in project '{1}'!".format(name, pData['name']))
self.model.add(pData['id'], ip_id, name, descr)
print " Host '{0}' successfully added to project '{1}' with IP '{2}' ! ".format(
name, pData['name'], ip
)
def list_action(self):
""" Action list of module """
if self.options['ip'].value:
print "{0:=^51}".format("")
print "|{0: ^49}|".format("Hosts for IP '{0}'".format(self.options['ip'].value))
print "{0:=^51}".format("")
print "| {0: ^23}| {1: ^23}|".format('Title', 'Description')
print "{0:=^51}".format("")
for host in self.model.list(Registry().get('pData')['id'], self.options['ip'].value):
print "| {0: <23}| {1: <23}|".format(host['name'], host['descr'])
print "{0:=^51}".format("")
else:
print "{0:=^76}".format("")
print "|{0: ^74}|".format("All host for project '{0}'".format(Registry().get('pData')['name']))
print "{0:=^76}".format("")
print "| {0: ^23}| {1: ^23}| {2: ^23}|".format('Title', 'Description', 'IP')
print "{0:=^76}".format("")
for host in self.model.list_without_ip(Registry().get('pData')['id']):
print "| {0: <23}| {1: <23}| {2: <23}|".format(host['name'], host['descr'], host['ip'])
print "{0:=^76}".format("")
def delete_action(self):
""" Delete action of module """
self.validate_main()
name = self.options['host'].value
if not self.model.exists(Registry().get('pData')['id'], name):
raise WSException("Host '{0}' not exists in this project!".format(name))
answer = raw_input("You really want to delete host '{0}' [y/n]? ".format(name))
if answer.lower() == 'y':
self.model.delete(Registry().get('pData')['id'], name)
print "Host '{0}' successfully deleted.".format(name)
else:
print "Host '{0}' not deleted.".format(name)