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.project = ProjectFactory()
cls.project_dict = cls.project.__dict__
cls.assignment_1 = ProjectAssignmentFactory(project=cls.project)
cls.assignment_2 = ProjectAssignmentFactory(project=cls.project)
cls.to_be_deleted = ProjectAssignmentFactory(project=cls.project)
cls.new_assignee = UserFactory()
def setUpTestData(cls):
cls.unavailable_status = ServerStatusFactory(
server_status="Unavailable")
cls.server = StaticServerFactory()
cls.unavailable_server = StaticServerFactory(
server_status=cls.unavailable_status)
cls.project = ProjectFactory()
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_crud_finding(self):
# Create
project = ProjectFactory(codename="S3kr3t Codename")
# Read
self.assertEqual(project.codename, "S3kr3t Codename")
self.assertEqual(project.pk, project.id)
self.assertQuerysetEqual(
self.Project.objects.all(),
[
f"<Project: {project.start_date} {project.client} {project.project_type} (S3kr3t Codename)>"
],
)
# Update
project.codename = "New Name"
project.save()
self.assertQuerysetEqual(
self.Project.objects.all(),
[
f"<Project: {project.start_date} {project.client} {project.project_type} (New Name)>"
],
)
# Delete
project.delete()
assert not self.Project.objects.all().exists()
def setUpTestData(cls):
cls.unavailable_status = DomainStatusFactory(domain_status="Unavailable")
cls.domain = DomainFactory(expiration=date.today() + timedelta(days=360))
cls.unavailable_domain = DomainFactory(
domain_status=cls.unavailable_status,
expiration=date.today() + timedelta(days=360),
)
cls.expired_domain = DomainFactory(expiration=date.today() - timedelta(days=30))
cls.project = ProjectFactory()
def setUpTestData(cls):
cls.project = ProjectFactory()
cls.project_dict = cls.project.__dict__
cls.objective_1 = ProjectObjectiveFactory(
project=cls.project, deadline=cls.project.end_date
)
cls.objective_2 = ProjectObjectiveFactory(
project=cls.project, deadline=cls.project.end_date
)
cls.to_be_deleted = ProjectObjectiveFactory(project=cls.project)
def setUpTestData(cls):
cls.project = ProjectFactory()
cls.domain = HistoryFactory(project=cls.project)
cls.server = ServerHistoryFactory(project=cls.project)
cls.vps = TransientServerFactory(project=cls.project)
cls.link = DomainServerConnectionFactory(
project=cls.project,
domain=cls.domain,
static_server=cls.server,
transient_server=None,
)
cls.link_dict = cls.link.__dict__
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 setUpTestData(cls):
cls.ProjectObjective = ProjectObjectiveFactory._meta.model
cls.project = ProjectFactory()
for x in range(3):
ProjectObjectiveFactory(project=cls.project)
cls.server = StaticServerFactory()
cls.aux_address_1 = AuxServerAddressFactory(
static_server=cls.server, ip_address="1.1.1.1", primary=True
)
cls.aux_address_2 = AuxServerAddressFactory(
static_server=cls.server, ip_address="1.1.1.2", primary=False
)
cls.scope = ProjectScopeFactory(
project=cls.project,
scope="1.1.1.1\r\n1.1.1.2\r\n1.1.1.3\r\n1.1.1.4\r\n1.1.1.5",
)
def test_checkout_adjustment_signal(self):
yesterday = date.today() - timedelta(days=1)
project = ProjectFactory(
start_date=date.today() - timedelta(days=14),
end_date=date.today() + timedelta(days=14),
)
domain_checkout = HistoryFactory(start_date=project.start_date,
end_date=project.end_date,
project=project)
exp_domain_checkout = HistoryFactory(start_date=project.start_date,
end_date=yesterday,
project=project)
server_checkout = ServerHistoryFactory(start_date=project.start_date,
end_date=project.end_date,
project=project)
exp_server_checkout = ServerHistoryFactory(
start_date=project.start_date, end_date=yesterday, project=project)
new_start = project.start_date - timedelta(days=7)
new_end = project.end_date + timedelta(days=7)
project.start_date = new_start
project.end_date = new_end
project.save()
domain_checkout.refresh_from_db()
exp_domain_checkout.refresh_from_db()
server_checkout.refresh_from_db()
exp_server_checkout.refresh_from_db()
self.assertEqual(domain_checkout.start_date, new_start)
self.assertEqual(server_checkout.start_date, new_start)
self.assertEqual(domain_checkout.end_date, new_end)
self.assertEqual(server_checkout.end_date, new_end)
self.assertEqual(exp_domain_checkout.end_date, yesterday)
self.assertEqual(exp_server_checkout.end_date, yesterday)
def test_valid_data(self):
project = ProjectFactory()
oplog = OplogFactory.build(project=project)
form = self.form_data(**oplog.__dict__)
self.assertTrue(form.is_valid())
def setUpTestData(cls):
cls.project = ProjectFactory(complete=False)
cls.user = UserFactory(password=PASSWORD)
cls.uri = reverse("oplog:oplog_create_no_project")
cls.project_uri = reverse("oplog:oplog_create",
kwargs={"pk": cls.project.pk})
def setUpTestData(cls):
cls.project = ProjectFactory(complete=False)
cls.user = UserFactory(password=PASSWORD)
cls.uri = reverse("rolodex:ajax_toggle_project", kwargs={"pk": cls.project.pk})
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):
project = ProjectFactory()
try:
project.get_absolute_url()
except:
self.fail("Project.get_absolute_url() raised an exception")
def setUpTestData(cls):
cls.project = ProjectFactory()
cls.project_dict = cls.project.__dict__
cls.target_1 = ProjectTargetFactory(project=cls.project)
cls.target_2 = ProjectTargetFactory(project=cls.project)
cls.to_be_deleted = ProjectTargetFactory(project=cls.project)
def setUpTestData(cls):
cls.project = ProjectFactory()
cls.project_dict = cls.project.__dict__
cls.scope_1 = ProjectScopeFactory(project=cls.project)
cls.scope_2 = ProjectScopeFactory(project=cls.project)
cls.to_be_deleted = ProjectScopeFactory(project=cls.project)
def setUpTestData(cls):
cls.project = ProjectFactory()
cls.report = ReportFactory(project=cls.project)
cls.report_dict = cls.report.__dict__
def setUpTestData(cls):
cls.project = ProjectFactory()
cls.project_dict = cls.project.__dict__
def setUpTestData(cls):
cls.project = ProjectFactory()
cls.server = TransientServerFactory(project=cls.project)
cls.server_dict = cls.server.__dict__
