def test_system_form_empty(self):
"""test minimum form requirements / INVALID"""
# get object
form = SystemForm(data={})
# compare
self.assertFalse(form.is_valid())
def systems_add(request):
if request.method == 'POST':
form = SystemForm(request.POST)
if form.is_valid():
system = form.save(commit=False)
system.system_created_by_user_id = request.user
system.system_modified_by_user_id = request.user
system.system_modify_time = timezone.now()
system.save()
form.save_m2m()
# get user string
request_user = str(request.user)
# extract lines from ip list
lines = request.POST.get('iplist').splitlines()
# call function to save ips
ips_save(request, system, lines)
# call logger
system.logger(str(request.user), ' SYSTEM_ADD_EXECUTED')
messages.success(request, 'System added')
return redirect('/systems/' + str(system.system_id))
else:
# show empty form with default values for convenience and speed reasons
form = SystemForm(initial={
'systemstatus': 2,
'analysisstatus': 1,
})
# call logger
debug_logger(str(request.user), " SYSTEM_ADD_ENTERED")
return render(request, 'dfirtrack_main/system/systems_add.html',
{'form': form})
def test_system_name_form_filled(self):
"""test minimum form requirements / INVALID"""
# get object
form = SystemForm(data={
'system_name': 'system_1',
})
# compare
self.assertFalse(form.is_valid())
def systems_edit(request, pk):
system = get_object_or_404(System, pk=pk)
if request.method == 'POST':
form = SystemForm(request.POST, instance=system)
if form.is_valid():
system = form.save(commit=False)
system.system_modified_by_user_id = request.user
system.system_modify_time = timezone.now()
system.save()
form.save_m2m()
system.ip.clear()
# get user string
request_user = str(request.user)
# extract lines from ip list
lines = request.POST.get('iplist').splitlines()
# call function to save ips
ips_save(request, system, lines)
# call logger
system.logger(str(request.user), ' SYSTEM_EDIT_EXECUTED')
messages.success(request, 'System edited')
return redirect('/systems/' + str(system.system_id))
else:
system = System.objects.get(system_id=pk)
""" get all existing ip addresses """
# get objects
ips = system.ip.all()
# count objects
iplen = len(ips)
# set counter
i = 0
# set default string if there is no object at all
ipstring = ''
for ip in ips:
# add ip to string
ipstring = ipstring + str(ip.ip_ip)
# increment counter
i += 1
# add newline but not for last occurence
if i < iplen:
ipstring = ipstring + '\n'
# show form for system with all ip addresses
form = SystemForm(
instance=system,
initial={
'iplist': ipstring,
},
)
# call logger
system.logger(str(request.user), " SYSTEM_EDIT_ENTERED")
return render(request, 'dfirtrack_main/system/systems_edit.html',
{'form': form})
def test_system_name_proper_chars(self):
"""test for max length"""
# get object
systemstatus_id = Systemstatus.objects.get(
systemstatus_name='systemstatus_1').systemstatus_id
# get object
form = SystemForm(data={
'system_name': 'n' * 50,
'systemstatus': systemstatus_id,
})
# compare
self.assertTrue(form.is_valid())
def test_system_systemstatus_form_filled(self):
"""test minimum form requirements / VALID"""
# get object
systemstatus_id = Systemstatus.objects.get(
systemstatus_name='systemstatus_1').systemstatus_id
# get object
form = SystemForm(data={
'system_name': 'system_1',
'systemstatus': systemstatus_id,
})
# compare
self.assertTrue(form.is_valid())
def test_system_deprecated_time_formatcheck(self):
"""test input format"""
# get object
systemstatus_id = Systemstatus.objects.get(
systemstatus_name='systemstatus_1').systemstatus_id
# get object
form = SystemForm(
data={
'system_name': 'system_1',
'systemstatus': systemstatus_id,
'system_deprecated_time': 'wrong format',
})
# compare
self.assertFalse(form.is_valid())
def test_system_is_vm_form_filled(self):
"""test additional form content"""
# get object
systemstatus_id = Systemstatus.objects.get(
systemstatus_name='systemstatus_1').systemstatus_id
# get object
form = SystemForm(
data={
'system_name': 'system_1',
'systemstatus': systemstatus_id,
'system_is_vm': True,
})
# compare
self.assertTrue(form.is_valid())
def test_system_case_form_filled(self):
"""test additional form content"""
# get object
systemstatus_id = Systemstatus.objects.get(
systemstatus_name='systemstatus_1').systemstatus_id
# get object
case_1_id = Case.objects.get(case_name='case_1').case_id
case_2_id = Case.objects.get(case_name='case_2').case_id
# get object
form = SystemForm(
data={
'system_name': 'system_1',
'systemstatus': systemstatus_id,
'case': [case_1_id, case_2_id],
})
# compare
self.assertTrue(form.is_valid())
def test_system_tag_form_filled(self):
"""test additional form content"""
# get object
systemstatus_id = Systemstatus.objects.get(
systemstatus_name='systemstatus_1').systemstatus_id
# get object
tag_1_id = Tag.objects.get(tag_name='tag_1').tag_id
tag_2_id = Tag.objects.get(tag_name='tag_2').tag_id
# get object
form = SystemForm(
data={
'system_name': 'system_1',
'systemstatus': systemstatus_id,
'tag': [tag_1_id, tag_2_id],
})
# compare
self.assertTrue(form.is_valid())
def test_system_recommendation_form_filled(self):
"""test additional form content"""
# get object
systemstatus_id = Systemstatus.objects.get(
systemstatus_name='systemstatus_1').systemstatus_id
# get object
recommendation_id = Recommendation.objects.get(
recommendation_name='recommendation_1').recommendation_id
# get object
form = SystemForm(
data={
'system_name': 'system_1',
'systemstatus': systemstatus_id,
'recommendation': recommendation_id,
})
# compare
self.assertTrue(form.is_valid())
