def test_method_count_lines(self):
scope = ProjectScopeFactory(scope="1.1.1.1\n1.2.3.4\nhostname.local")
try:
lines = scope.count_lines()
self.assertEqual(3, lines)
except Exception:
self.fail(
"ProjectScope model `count_lines` method failed unexpectedly!")
def test_method_count_lines_str(self):
scope = ProjectScopeFactory(scope="1.1.1.1\n1.2.3.4\nhostname.local")
try:
# Test with multiple lines
lines = scope.count_lines_str()
self.assertEqual("3 Lines", lines)
# Test with single line
scope.scope = "1.1.1.1"
scope.save()
lines = scope.count_lines_str()
self.assertEqual("1 Line", lines)
except Exception:
self.fail(
"ProjectScope model `count_lines_str` method failed unexpectedly!"
)
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_crud_finding(self):
# Create
scope = ProjectScopeFactory(name="CDE")
# Read
self.assertEqual(scope.name, "CDE")
self.assertEqual(scope.pk, scope.id)
self.assertQuerysetEqual(
self.ProjectScope.objects.all(),
[f"<ProjectScope: {scope.project}: CDE>"],
)
# Update
scope.name = "Mainframes"
scope.save()
self.assertQuerysetEqual(
self.ProjectScope.objects.all(),
[f"<ProjectScope: {scope.project}: Mainframes>"],
)
# Delete
scope.delete()
assert not self.ProjectScope.objects.all().exists()
def setUpTestData(cls):
cls.user = UserFactory(password=PASSWORD)
cls.scope = ProjectScopeFactory(name="TestScope")
cls.uri = reverse("rolodex:ajax_export_project_scope", kwargs={"pk": cls.scope.pk})
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.scope = ProjectScopeFactory()
cls.scope_dict = cls.scope.__dict__