def test_model_cleaning_position(self):
report = ReportFactory()
num_of_findings = 10
findings = []
for finding_id in range(num_of_findings):
findings.append(
ReportFindingLinkFactory(report=report, severity=self.critical_severity)
)
# New position values
first_pos = 1
second_pos = 2
# Bump first finding from ``1`` to new value
findings[0].position = second_pos
findings[0].save()
cleaned_findings = []
for f in findings:
f.refresh_from_db()
cleaned_findings.append(f)
# Assert first finding is now in second position
self.assertEqual(cleaned_findings[0].position, second_pos)
# Assert second finding has moved into first positon
self.assertEqual(cleaned_findings[1].position, first_pos)
# Test triggering ``clean()`` method when parent ``Report`` is deleted
report.delete()
def setUpTestData(cls):
cls.project = ProjectFactory()
cls.report = ReportFactory(project=cls.project)
cls.user = UserFactory(password=PASSWORD)
cls.uri = reverse("reporting:report_create_no_project")
cls.project_uri = reverse(
"reporting:report_create", kwargs={"pk": cls.project.pk}
)
def test_position_set_to_zero(self):
report = ReportFactory()
finding = ReportFindingLinkFactory(report=report, severity=self.critical_severity)
finding.position = -10
finding.save()
finding.refresh_from_db()
# Assert the other ``position`` values updated properly
self.assertEqual(finding.position, 1)
def setUpTestData(cls):
cls.user = UserFactory(password=PASSWORD)
cls.num_of_reports = 10
cls.reports = []
for report_id in range(cls.num_of_reports):
title = f"Report {report_id}"
cls.reports.append(ReportFactory(title=title))
cls.uri = reverse("reporting:reports")
def setUpTestData(cls):
cls.report = ReportFactory()
cls.ReportFindingLink = ReportFindingLinkFactory._meta.model
cls.user = UserFactory(password=PASSWORD)
cls.num_of_findings = 10
cls.findings = []
for finding_id in range(cls.num_of_findings):
title = f"Finding {finding_id}"
cls.findings.append(ReportFindingLinkFactory(title=title, report=cls.report))
cls.uri = reverse("reporting:report_clone", kwargs={"pk": cls.report.pk})
def setUpTestData(cls):
cls.group_1 = GroupFactory(name="Group 1")
cls.group_2 = GroupFactory(name="Group 2")
cls.user = UserFactory(password=PASSWORD, groups=(cls.group_1, ))
cls.project = ProjectFactory()
cls.report = ReportFactory(project=cls.project)
cls.num_of_findings = 3
ReportFindingLinkFactory.create_batch(cls.num_of_findings,
report=cls.report,
assigned_to=cls.user)
cls.uri = reverse("home:dashboard")
def test_position_set_higher_than_count(self):
report = ReportFactory()
num_of_findings = 10
findings = []
for finding_id in range(num_of_findings):
findings.append(
ReportFindingLinkFactory(report=report,
severity=self.critical_severity))
findings[0].position = 100
findings[0].save()
findings[0].refresh_from_db()
# Assert the other ``position`` values updated properly
self.assertEqual(findings[0].position, num_of_findings)
def test_model_cleaning_severity_and_positon_changes(self):
report = ReportFactory()
num_of_findings = 5
findings = []
for finding_id in range(num_of_findings):
findings.append(
ReportFindingLinkFactory(report=report,
severity=self.critical_severity), )
for finding_id in range(num_of_findings):
findings.append(
ReportFindingLinkFactory(report=report,
severity=self.high_severity), )
# Bounce findings around to shuffle positions several times
findings[8].severity = self.critical_severity
findings[8].position = 2
findings[8].save()
findings[5].severity = self.critical_severity
findings[5].position = 1
findings[5].save()
findings[3].severity = self.high_severity
findings[3].position = 2
findings[3].save()
cleaned_findings = []
for f in findings:
f.refresh_from_db()
cleaned_findings.append(f)
# Assert ``severity`` and ``position`` changes committed correctly
self.assertEqual(cleaned_findings[8].severity, self.critical_severity)
self.assertEqual(cleaned_findings[8].position, 3)
self.assertEqual(cleaned_findings[5].severity, self.critical_severity)
self.assertEqual(cleaned_findings[5].position, 1)
self.assertEqual(cleaned_findings[3].severity, self.high_severity)
self.assertEqual(cleaned_findings[3].position, 2)
# Assert the ``position`` values updated properly for "Critical"
self.assertEqual(cleaned_findings[5].position, 1)
self.assertEqual(cleaned_findings[0].position, 2)
self.assertEqual(cleaned_findings[8].position, 3)
self.assertEqual(cleaned_findings[1].position, 4)
self.assertEqual(cleaned_findings[2].position, 5)
self.assertEqual(cleaned_findings[4].position, 6)
# Assert the ``position`` values updated properly for "High"
self.assertEqual(cleaned_findings[5].position, 1)
self.assertEqual(cleaned_findings[3].position, 2)
self.assertEqual(cleaned_findings[7].position, 3)
self.assertEqual(cleaned_findings[9].position, 4)
def setUpTestData(cls):
cls.report = ReportFactory()
cls.user = UserFactory(password=PASSWORD)
# These must exist for the view to function
cls.high_severity = SeverityFactory(severity="High", weight=1)
cls.med_severity = SeverityFactory(severity="Medium", weight=2)
cls.low_severity = SeverityFactory(severity="Low", weight=3)
cls.info_severity = SeverityFactory(severity="Informational", weight=4)
cls.finding_type = FindingTypeFactory(finding_type="Network")
cls.uri = reverse("reporting:assign_blank_finding", kwargs={"pk": cls.report.pk})
cls.redirect_uri = reverse(
"reporting:report_detail", kwargs={"pk": cls.report.pk}
)
def setUpTestData(cls):
cls.user = UserFactory(password=PASSWORD)
cls.Project = ProjectFactory._meta.model
cls.ProjectAssignment = ProjectAssignmentFactory._meta.model
cls.ReportFindingLink = ReportFindingLinkFactory._meta.model
cls.current_project = ProjectFactory(
start_date=date.today() - timedelta(days=14),
end_date=date.today(),
complete=True
)
cls.future_project = ProjectFactory(
start_date=date.today() + timedelta(days=14),
end_date=date.today() + timedelta(days=28),
complete=False
)
ProjectAssignmentFactory(
project=cls.current_project,
operator=cls.user,
start_date=date.today(),
end_date=date.today() + timedelta(days=14),
)
ProjectAssignmentFactory(
project=cls.future_project,
operator=cls.user,
start_date=date.today() + timedelta(days=14),
end_date=date.today() + timedelta(days=28),
)
cls.report = ReportFactory(project=cls.current_project)
ReportFindingLinkFactory.create_batch(3, report=cls.report, assigned_to=cls.user)
cls.user_tasks = (
cls.ReportFindingLink.objects.select_related("report", "report__project")
.filter(
Q(assigned_to=cls.user) & Q(report__complete=False) & Q(complete=False)
)
.order_by("report__project__end_date")[:10]
)
cls.user_projects = cls.ProjectAssignment.objects.select_related(
"project", "project__client", "role"
).filter(Q(operator=cls.user))
cls.active_projects = cls.ProjectAssignment.objects.select_related(
"project", "project__client", "role"
).filter(Q(operator=cls.user) & Q(project__complete=False))
cls.uri = reverse("home:dashboard")
def setUpTestData(cls):
cls.report = ReportFactory(
docx_template=ReportDocxTemplateFactory(),
pptx_template=ReportPptxTemplateFactory(),
)
cls.high_severity = SeverityFactory(severity="High", weight=1)
cls.critical_severity = SeverityFactory(severity="Critical", weight=0)
cls.user = UserFactory(password=PASSWORD)
cls.new_user = UserFactory(password=PASSWORD)
cls.num_of_findings = 10
cls.findings = []
for finding_id in range(cls.num_of_findings):
title = f"Finding {finding_id}"
cls.findings.append(ReportFindingLinkFactory(title=title, report=cls.report))
cls.uri = reverse("reporting:local_edit", kwargs={"pk": cls.findings[0].pk})
def test_crud_report(self):
# Create
report = ReportFactory(title="New Report")
# Read
self.assertEqual(report.title, "New Report")
self.assertEqual(report.pk, report.id)
self.assertEqual(len(self.Report.objects.all()), 1)
self.assertEqual(self.Report.objects.first(), report)
# Update
report.title = "Updated Report"
report.save()
self.assertEqual(report.title, "Updated Report")
# Delete
report.delete()
assert not self.Report.objects.all().exists()
def test_position_change_on_delete(self):
report = ReportFactory()
num_of_findings = 5
findings = []
for finding_id in range(num_of_findings):
findings.append(
ReportFindingLinkFactory(report=report,
severity=self.critical_severity), )
for finding_id in range(num_of_findings):
findings.append(
ReportFindingLinkFactory(report=report,
severity=self.high_severity), )
# Delete several findings to create gaps in the severity groups
# Need to use atomic because ``TestCase`` and a ``post_delete`` Signal
with transaction.atomic():
findings[3].delete()
findings[5].delete()
findings[8].delete()
cleaned_findings = []
for f in findings:
try:
f.refresh_from_db()
cleaned_findings.append(f)
except self.ReportFindingLink.DoesNotExist:
pass
# Assert the ``position`` values updated properly for "Critical"
self.assertEqual(cleaned_findings[0].position, 1)
self.assertEqual(cleaned_findings[1].position, 2)
self.assertEqual(cleaned_findings[2].position, 3)
self.assertEqual(cleaned_findings[3].position, 4)
# Assert the ``position`` values updated properly for "High"
self.assertEqual(cleaned_findings[4].position, 1)
self.assertEqual(cleaned_findings[5].position, 2)
self.assertEqual(cleaned_findings[6].position, 3)
def test_model_cleaning_severity_change(self):
report = ReportFactory()
num_of_findings = 10
findings = []
for finding_id in range(num_of_findings):
findings.append(
ReportFindingLinkFactory(report=report,
severity=self.critical_severity))
# Bump the first half of the findings to the new severity in reverse order
for f in reversed(range(5)):
findings[f].severity = self.high_severity
findings[f].save()
cleaned_findings = []
for f in findings:
f.refresh_from_db()
cleaned_findings.append(f)
# Assert severity was properly updated
self.assertEqual(cleaned_findings[0].severity, self.high_severity)
# Assert the positions were set properly
self.assertEqual(cleaned_findings[5].position, 1)
self.assertEqual(cleaned_findings[9].position, 5)
self.assertEqual(cleaned_findings[0].position, 1)
self.assertEqual(cleaned_findings[4].position, 5)
def setUpTestData(cls):
cls.project = ProjectFactory()
cls.report = ReportFactory(project=cls.project)
cls.report_dict = cls.report.__dict__
def setUpTestData(cls):
cls.report = ReportFactory()
cls.user = UserFactory(password=PASSWORD)
cls.uri = reverse("reporting:report_delete", kwargs={"pk": cls.report.pk})
def setUpTestData(cls):
cls.report = ReportFactory()
cls.docx_template = cls.report.docx_template
cls.pptx_template = cls.report.pptx_template
def setUpTestData(cls):
cls.ReportFindingLink = ReportFindingLinkFactory._meta.model
cls.report = ReportFactory()
for x in range(3):
ReportFindingLinkFactory(report=cls.report)
def test_prop_count_findings(self):
project = ProjectFactory()
report = ReportFactory(project=project)
for x in range(3):
ReportFindingLinkFactory(report=report)
self.assertEqual(project.count_findings(), 3)
def test_get_absolute_url(self):
report = ReportFactory()
try:
report.get_absolute_url()
except:
self.fail("Report.get_absolute_url() raised an exception")
def setUpTestData(cls):
cls.report = ReportFactory()
cls.finding = FindingFactory()
cls.user = UserFactory(password=PASSWORD)
cls.uri = reverse("reporting:ajax_assign_finding", kwargs={"pk": cls.finding.pk})