def test_system_form_empty(self):
"""test minimum form requirements / INVALID"""
# get object
form = SystemForm(data={})
# compare
self.assertFalse(form.is_valid())
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 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 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_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_osarch_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['osarch'].label, 'Osarch')
def test_system_dnsname_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['dnsname'].label, 'Dnsname')
def test_system_domain_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['domain'].label, 'Domain')
def test_system_contact_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['contact'].label, 'Contact')
def test_system_host_system_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['host_system'].label, 'Host system')
def test_system_case_form_label(self):
"""test form label"""
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['case'].label, 'Cases')
def test_system_systemtype_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEquals(form.fields['systemtype'].label, 'Systemtype')
def test_system_tag_form_label(self):
"""test form label"""
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['tag'].label, 'Tags')
def test_system_systemstatus_form_label(self):
"""test form label"""
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['systemstatus'].label, 'Systemstatus (*)')
def test_system_is_vm_form_label(self):
"""test form label"""
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['system_is_vm'].label, 'System is a VM')
def test_system_analysisstatus_form_label(self):
"""test form label"""
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['analysisstatus'].label, 'Analysisstatus')
def test_system_reason_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEquals(form.fields['reason'].label, 'Reason')
def test_system_location_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['location'].label, 'Location')
def test_system_deprecated_time_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['system_deprecated_time'].label,
'System is deprecated since (YYYY-MM-DD HH:MM:SS)')
def test_system_export_spreadsheet_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['system_export_spreadsheet'].label,
'Export system to spreadsheet')
def test_system_export_markdown_form_label(self):
"""test form label"""
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['system_export_markdown'].label,
'Export system to markdown')
def test_system_os_form_label(self):
"""test form label"""
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['os'].label, 'Operating system')
self.assertEqual(form.fields['os'].empty_label, 'Select OS (optional)')
def test_system_lastbooted_time_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['system_lastbooted_time'].label,
'Last booted (YYYY-MM-DD HH:MM:SS)')
def test_system_install_time_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['system_install_time'].label,
'System install time')
def test_system_deprecated_time_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['system_deprecated_time'].label,
'System deprecated time')
def test_system_serviceprovider_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['serviceprovider'].label,
'Serviceprovider')
def test_system_install_time_form_label(self):
""" test form label """
# get object
form = SystemForm()
# compare
self.assertEqual(form.fields['system_install_time'].label,
'Installation time (YYYY-MM-DD HH:MM:SS)')
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})
