updated the architecture

tests/test_scoring.py

@@ -0,0 +1,63 @@
+from app.schemas.scan import Issue
+from app.services.scoring import calculate_layer_statuses, calculate_score
+
+
+def test_perfect_score_no_issues():
+    assert calculate_score([]) == 100
+
+
+def test_critical_deduction():
+    issues = [Issue(issue="Test", severity="Critical", layer="Transport Layer", fix="Fix")]
+    assert calculate_score(issues) == 85
+
+
+def test_warning_deduction():
+    issues = [Issue(issue="Test", severity="Warning", layer="Transport Layer", fix="Fix")]
+    assert calculate_score(issues) == 95
+
+
+def test_info_deduction():
+    issues = [Issue(issue="Test", severity="Info", layer="Transport Layer", fix="Fix")]
+    assert calculate_score(issues) == 98
+
+
+def test_score_cannot_go_below_zero():
+    issues = [Issue(issue=f"Test {i}", severity="Critical", layer="Transport Layer", fix="Fix") for i in range(10)]
+    assert calculate_score(issues) == 0
+
+
+def test_all_layers_present():
+    statuses = calculate_layer_statuses([])
+    assert "Transport Layer" in statuses
+    assert "SSL/TLS Layer" in statuses
+    assert "Server Config Layer" in statuses
+    assert "Cookie Security" in statuses
+    assert "Exposure Layer" in statuses
+
+
+def test_layer_status_green_when_no_issues():
+    statuses = calculate_layer_statuses([])
+    for layer in statuses.values():
+        assert layer.status == "green"
+        assert layer.issues == 0
+
+
+def test_layer_status_yellow_for_few_issues():
+    issues = [
+        Issue(issue="Test 1", severity="Warning", layer="SSL/TLS Layer", fix="Fix"),
+        Issue(issue="Test 2", severity="Warning", layer="SSL/TLS Layer", fix="Fix"),
+    ]
+    statuses = calculate_layer_statuses(issues)
+    assert statuses["SSL/TLS Layer"].status == "yellow"
+    assert statuses["SSL/TLS Layer"].issues == 2
+
+
+def test_layer_status_red_for_many_issues():
+    issues = [
+        Issue(issue="Test 1", severity="Warning", layer="Cookie Security", fix="Fix"),
+        Issue(issue="Test 2", severity="Warning", layer="Cookie Security", fix="Fix"),
+        Issue(issue="Test 3", severity="Critical", layer="Cookie Security", fix="Fix"),
+    ]
+    statuses = calculate_layer_statuses(issues)
+    assert statuses["Cookie Security"].status == "red"
+    assert statuses["Cookie Security"].issues == 3