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})