def test_entry_form_empty(self):
""" test minimum form requirements / INVALID """
# get object
form = EntryForm(data = {})
# compare
self.assertFalse(form.is_valid())
def test_entry_time_form_filled(self):
""" test minimum form requirements / INVALID """
# get object
form = EntryForm(data = {
'entry_time': '2001-02-03 12:34:56',
})
# compare
self.assertFalse(form.is_valid())
def test_system_form_filled(self):
""" test minimum form requirements / VALID """
# get foreign key object id
system_id = System.objects.get(system_name='system_1').system_id
# get object
form = EntryForm(data = {
'entry_time': '2009-08-07 12:34:56',
'system': system_id,
})
# compare
self.assertTrue(form.is_valid())
def test_entry_sha1_form_filled(self):
""" test additional form content """
# get foreign key object id
system_id = System.objects.get(system_name='system_1').system_id
# get object
form = EntryForm(data = {
'entry_time': '2009-08-07 12:34:56',
'system': system_id,
'entry_sha1': 'da39a3ee5e6b4b0d3255bfef95601890afd80709',
})
# compare
self.assertTrue(form.is_valid())
def test_entry_note_form_filled(self):
""" test additional form content """
# get foreign key object id
system_id = System.objects.get(system_name='system_1').system_id
# get object
form = EntryForm(data = {
'entry_time': '2009-08-07 12:34:56',
'system': system_id,
'entry_note': 'lorem ipsum',
})
# compare
self.assertTrue(form.is_valid())
def test_entry_type_too_many_chars(self):
""" test for max length """
# define datetime string
entry_time_string = '2009-08-07 12:34:56'
# get foreign key object id
system_id = System.objects.get(system_name='system_1').system_id
# get object
form = EntryForm(data = {
'entry_time': entry_time_string,
'system': system_id,
'entry_type': 'ttttttttttttttttttttttttttttttt',
})
# compare
self.assertFalse(form.is_valid())
def test_case_form_label(self):
""" test form label """
# get object
form = EntryForm()
# compare
self.assertEquals(form.fields['case'].label, 'Case')
def test_entry_content_form_label(self):
""" test form label """
# get object
form = EntryForm()
# compare
self.assertEqual(form.fields['entry_content'].label, 'Entry content')
def test_tag_form_label(self):
"""test form label"""
# get object
form = EntryForm()
# compare
self.assertEqual(form.fields['tag'].label, 'Tags')
def test_entry_note_form_label(self):
""" test form label """
# get object
form = EntryForm()
# compare
self.assertEquals(form.fields['entry_note'].label, 'Entry note')
def test_entry_system_form_label(self):
""" test form label """
# get object
form = EntryForm()
# compare
self.assertEquals(form.fields['entry_system'].label, 'Entry system (for report)')
def test_entry_sha1_proper_chars(self):
"""test for max length"""
# define datetime string
entry_time_string = '2009-08-07 12:34:56'
# get foreign key object id
system_id = System.objects.get(system_name='system_1').system_id
# get object
form = EntryForm(
data={
'entry_time': entry_time_string,
'system': system_id,
'entry_sha1': 's' * 40,
})
# compare
self.assertTrue(form.is_valid())
def test_system_form_label(self):
""" test form label """
# get object
form = EntryForm()
# compare
self.assertEquals(form.fields['system'].label, 'System (*)')
def test_entry_utc_form_label(self):
""" test form label """
# get object
form = EntryForm()
# compare
self.assertEquals(form.fields['entry_utc'].label, 'Entry time (for report) (HH:MM:SS)')
def test_entry_time_form_label(self):
""" test form label """
# get object
form = EntryForm()
# compare
self.assertEquals(form.fields['entry_time'].label, 'Entry time (for sorting) (YYYY-MM-DD HH:MM:SS) (*)')
def test_entry_date_form_label(self):
""" test form label """
# get object
form = EntryForm()
# compare
self.assertEqual(form.fields['entry_date'].label,
'Entry date (YYYY-MM-DD)')
def test_case_form_label(self):
"""test form label"""
# get object
form = EntryForm()
# compare
self.assertEqual(form.fields['case'].label, 'Case')
self.assertEqual(form.fields['case'].empty_label,
'Select case (optional)')
def entrys_add(request):
if request.method == 'POST':
form = EntryForm(request.POST)
if form.is_valid():
entry = form.save(commit=False)
entry.entry_created_by_user_id = request.user
entry.entry_modified_by_user_id = request.user
entry.save()
entry.logger(str(request.user), " ENTRY_ADD_EXECUTED")
messages.success(request, 'Entry added')
return redirect('/systems/' + str(entry.system.system_id))
else:
if request.method == 'GET' and 'system' in request.GET:
system = request.GET['system']
form = EntryForm(initial={
'system': system,
})
else:
form = EntryForm()
debug_logger(str(request.user), " ENTRY_ADD_ENTERED")
return render(request, 'dfirtrack_main/entry/entrys_add.html',
{'form': form})
def entrys_edit(request, pk):
entry = get_object_or_404(Entry, pk=pk)
if request.method == 'POST':
form = EntryForm(request.POST, instance=entry)
if form.is_valid():
entry = form.save(commit=False)
entry.entry_modified_by_user_id = request.user
entry.save()
entry.logger(str(request.user), " ENTRY_EDIT_EXECUTED")
messages.success(request, 'Entry edited')
return redirect('/systems/' + str(entry.system.system_id))
else:
form = EntryForm(instance=entry)
entry.logger(str(request.user), " ENTRY_EDIT_ENTERED")
return render(request, 'dfirtrack_main/entry/entrys_edit.html',
{'form': form})
