def test_creating_new_services(self):
service = Service()
service.name = "test"
service.description = "Test Service"
service.port = 8080
service.protocol = Service.HTTP
service.save()
all_services_in_database = Service.objects.all()
self.assertEquals(len(all_services_in_database), 1)
only_service_in_database = all_services_in_database[0]
self.assertEquals(only_service_in_database, service)
self.assertEquals(only_service_in_database.name, "test")
self.assertEquals(only_service_in_database.description, "Test Service")
self.assertEquals(only_service_in_database.port, 8080)
def servicemodule_test(request, servmdulid=None):
c = getAuthValues(request, {})
if c["auth_name"] != "auth_team_white":
return HttpResponseRedirect("/")
c.update(csrf(request))
c['servmdul_obj'] = ServiceModule.objects.get(servmdulid=servmdulid)
serv_obj = Service(servicemodule=c['servmdul_obj'])
if request.method != "POST":
# Serve blank form with no results
c["service_configs"] = buildServiceConfigForm(serv_obj,
TestServiceForm())
return render_to_response('AdminConfig/servicemodule_test.html', c)
c["service_configs"] = buildServiceConfigForm(serv_obj, TestServiceForm(),
request.POST)
form_dict = request.POST.copy().dict()
form_dict.pop('csrfmiddlewaretoken')
# Clean network address
if form_dict['networkaddr'][-1] == ".":
form_dict['networkaddr'] = form_dict['networkaddr'][:-1]
if form_dict['networkaddr'][0] == ".":
form_dict['networkaddr'] = form_dict['networkaddr'][1:]
# Clean machine address value
if form_dict['networkloc'][0] == ".":
form_dict['networkloc'] = form_dict['networkloc'][1:]
if form_dict['networkloc'][-1] == ".":
form_dict['networkloc'] = form_dict['networkloc'][:-1]
# Prepare the service object for use in the module
if form_dict.pop('connectip') == u'0':
serv_obj.connectip = False
else:
serv_obj.connectip = True
serv_obj.name = c['servmdul_obj'].modulename
serv_obj.networkloc = str(form_dict.pop('networkloc'))
serv_obj.defaultport = int(form_dict.pop('defaultport'))
serv_obj.points = 100
c['score_obj'] = run_plugin_test(serv_obj, form_dict)
return render_to_response('AdminConfig/servicemodule_test.html', c)
def servicemodule_test(request, servmdulid = None):
c = getAuthValues(request, {})
if c["auth_name"] != "auth_team_white":
return HttpResponseRedirect("/")
c.update(csrf(request))
c['servmdul_obj'] = ServiceModule.objects.get(servmdulid = servmdulid)
serv_obj = Service(servicemodule = c['servmdul_obj'])
if request.method != "POST":
# Serve blank form with no results
c["service_configs"] = buildServiceConfigForm(serv_obj, TestServiceForm())
return render_to_response('AdminConfig/servicemodule_test.html', c)
c["service_configs"] = buildServiceConfigForm(serv_obj, TestServiceForm(), request.POST)
form_dict = request.POST.copy().dict()
form_dict.pop('csrfmiddlewaretoken')
# Clean network address
if form_dict['networkaddr'][-1] == ".":
form_dict['networkaddr'] = form_dict['networkaddr'][:-1]
if form_dict['networkaddr'][0] == ".":
form_dict['networkaddr'] = form_dict['networkaddr'][1:]
# Clean machine address value
if form_dict['networkloc'][0] == ".":
form_dict['networkloc'] = form_dict['networkloc'][1:]
if form_dict['networkloc'][-1] == ".":
form_dict['networkloc'] = form_dict['networkloc'][:-1]
# Prepare the service object for use in the module
if form_dict.pop('connectip') == u'0':
serv_obj.connectip = False
else:
serv_obj.connectip = True
serv_obj.name = c['servmdul_obj'].modulename
serv_obj.networkloc = str(form_dict.pop('networkloc'))
serv_obj.defaultport = int(form_dict.pop('defaultport'))
serv_obj.points = 100
c['score_obj'] = run_plugin_test(serv_obj, form_dict)
return render_to_response('AdminConfig/servicemodule_test.html', c)
