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

Promene u dizajnu dokumenata, controla, residual graph, residual tabele...

Browse Source
This commit is contained in:
Amir
2025-06-13 17:45:22 +02:00
parent fa734e5be9
commit aff66589e9
14 changed files with 711 additions and 2132 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
backend/core/tests/test_utils.py
View File

@@ -128,4 +128,45 @@ class UtilsTests(TestCase):
graph_data = generate_risk_graph(risks_with_controls)
self.assertIsInstance(graph_data, str)
self.assertTrue(len(graph_data) > 1000)
self.assertTrue(len(graph_data) > 1000)
def test_generate_residual_risk_graph_base64(self):
risks_with_controls = [
{
'risk': {'id': 1, 'name': 'Risk 1'},
'residual_impact': 3,
'residual_likelihood': 4,
},
{
'risk': {'id': 2, 'name': 'Risk 2'},
'residual_impact': 2,
'residual_likelihood': 2,
}
]
graph_data = generate_residual_risk_graph(risks_with_controls)
self.assertIsInstance(graph_data, str)
self.assertTrue(len(graph_data) > 1000)
def test_get_safeguard_summary_table_basic(self):
from backend.core.tables import get_safeguard_summary_table
risks_with_controls = [
{
'risk': {'id': 1, 'name': 'Risk 1'},
'controls': [
{'control': 101, 'control__name': 'Control A'},
{'control': 102, 'control__name': 'Control B'},
]
},
{
'risk': {'id': 2, 'name': 'Risk 2'},
'controls': [
{'control': 101, 'control__name': 'Control A'},
]
}
]
summary = get_safeguard_summary_table(risks_with_controls)
self.assertEqual(summary, [
{'id': 101, 'name': 'Control A', 'count': 2},
{'id': 102, 'name': 'Control B', 'count': 1},
])