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