senaduka/old-riskletpy
1
0
Fork 0
Code Issues 5 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

#8 Promenjen unos i snimanje kontrola

Browse Source
This commit is contained in:
Amir
2025-02-13 18:29:54 +01:00
parent f41997fd59
commit 8d21587376
9 changed files with 203 additions and 127 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
backend/core/management/commands/tests/test_export_controls.py
View File

@@ -2,25 +2,12 @@ import os
import csv
from django.core.management import call_command
from django.test import TestCase
from backend.core.models import Risk, Control
from backend.core.models import Control
class ExportControlsCommandTest(TestCase):
def setUp(self):
self.risk1 = Risk.objects.create(risk_id=1, risk_name="Test Risk 1")
self.risk2 = Risk.objects.create(risk_id=2, risk_name="Test Risk 2")
self.control1 = Control.objects.create(
risk=self.risk1,
safeguard="Test Safeguard 1",
description="Description 1",
weight=5
)
self.control2 = Control.objects.create(
risk=self.risk2,
safeguard="Test Safeguard 2",
description="Description 2",
weight=10
)
Control.objects.create(name="Test Safeguard 1")
Control.objects.create(name="Test Safeguard 2")
self.csv_file_path = 'test_export_controls.csv'
@@ -28,7 +15,7 @@ class ExportControlsCommandTest(TestCase):
if os.path.exists(self.csv_file_path):
os.remove(self.csv_file_path)
def test_export_controls_success(self):
def test_export_controls(self):
call_command('export_controls', self.csv_file_path)
self.assertTrue(os.path.exists(self.csv_file_path))
@@ -37,27 +24,7 @@ class ExportControlsCommandTest(TestCase):
reader = csv.DictReader(csv_file)
rows = list(reader)
self.assertEqual(len(rows), 2)
self.assertEqual(len(rows), 2)
self.assertEqual(rows[0]["Risk #"], str(self.risk1.risk_id))
self.assertEqual(rows[0]["Risk Description"], self.risk1.risk_name)
self.assertEqual(rows[0]["CIS v8.1 Safeguards (Sub-Controls)"], self.control1.safeguard)
self.assertEqual(rows[0]["Weight (0-10)"], str(self.control1.weight))
self.assertEqual(rows[1]["Risk #"], str(self.risk2.risk_id))
self.assertEqual(rows[1]["Risk Description"], self.risk2.risk_name)
self.assertEqual(rows[1]["CIS v8.1 Safeguards (Sub-Controls)"], self.control2.safeguard)
self.assertEqual(rows[1]["Weight (0-10)"], str(self.control2.weight))
def test_export_controls_empty(self):
Control.objects.all().delete()
call_command('export_controls', self.csv_file_path)
self.assertTrue(os.path.exists(self.csv_file_path))
with open(self.csv_file_path, mode='r', encoding='utf-8') as csv_file:
reader = csv.DictReader(csv_file)
rows = list(reader)
self.assertEqual(len(rows), 0)
self.assertEqual(rows[0]["CIS v8.1 Safeguards (Sub-Controls)"], "Test Safeguard 1")
self.assertEqual(rows[1]["CIS v8.1 Safeguards (Sub-Controls)"], "Test Safeguard 2")

78
backend/core/management/commands/tests/test_import_controls.py
View File

@@ -1,56 +1,40 @@
import os
import csv
import tempfile
from django.core.management import call_command
from django.test import TestCase
from backend.core.models import Risk, Control
from backend.core.models import Control
class ImportControlsCommandTest(TestCase):
def setUp(self):
"""Create a temporary CSV file with test control data"""
self.temp_csv = tempfile.NamedTemporaryFile(mode="w", delete=False, newline='', encoding='utf-8')
fieldnames = [
"Risk #", "Risk Description", "CIS v8.1 Safeguards (Sub-Controls)", "Weight (0-10)"
]
writer = csv.DictWriter(self.temp_csv, fieldnames=fieldnames)
writer.writeheader()
writer.writerow({
"Risk #": "1",
"Risk Description": "Test Risk 1",
"CIS v8.1 Safeguards (Sub-Controls)": "Test Safeguard 1",
"Weight (0-10)": "5"
})
writer.writerow({
"Risk #": "2",
"Risk Description": "Test Risk 2",
"CIS v8.1 Safeguards (Sub-Controls)": "Test Safeguard 2",
"Weight (0-10)": "10"
})
self.temp_csv.close()
def test_import_controls_command(self):
"""Test importing controls from a CSV file"""
self.assertEqual(Control.objects.count(), 0)
self.assertEqual(Risk.objects.count(), 0)
Risk.objects.create(risk_id=1, risk_name="Test Risk 1")
Risk.objects.create(risk_id=2, risk_name="Test Risk 2")
call_command('import_controls', self.temp_csv.name)
self.assertEqual(Control.objects.count(), 2)
control1 = Control.objects.get(risk__risk_id=1)
self.assertEqual(control1.safeguard, "Test Safeguard 1")
self.assertEqual(control1.weight, 5)
control2 = Control.objects.get(risk__risk_id=2)
self.assertEqual(control2.safeguard, "Test Safeguard 2")
self.assertEqual(control2.weight, 10)
self.csv_file_path = 'test_import_controls.csv'
with open(self.csv_file_path, mode='w', encoding='utf-8') as csv_file:
writer = csv.DictWriter(csv_file, fieldnames=["CIS v8.1 Safeguards (Sub-Controls)"])
writer.writeheader()
writer.writerow({"CIS v8.1 Safeguards (Sub-Controls)": "Test Safeguard 1"})
writer.writerow({"CIS v8.1 Safeguards (Sub-Controls)": "Test Safeguard 2"})
def tearDown(self):
"""Remove temporary CSV file after test"""
os.remove(self.temp_csv.name)
if os.path.exists(self.csv_file_path):
os.remove(self.csv_file_path)
def test_import_controls(self):
self.assertEqual(Control.objects.count(), 0)
call_command('import_controls', self.csv_file_path)
self.assertEqual(Control.objects.count(), 2)
safeguards = Control.objects.values_list('name', flat=True)
self.assertIn("Test Safeguard 1", safeguards)
self.assertIn("Test Safeguard 2", safeguards)
def test_import_controls_update(self):
Control.objects.create(name="Test Safeguard 1")
call_command('import_controls', self.csv_file_path)
self.assertEqual(Control.objects.count(), 2)
safeguards = Control.objects.values_list('name', flat=True)
self.assertIn("Test Safeguard 1", safeguards)
self.assertIn("Test Safeguard 2", safeguards)