fix tests

armosec · Sep 14, 2023 · 7d6495d · 7d6495d
1 parent f5b39ab
commit 7d6495d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/attackchains/attackchainhandler_mocks.go b/attackchains/attackchainhandler_mocks.go
@@ -94,6 +94,9 @@ func CommonContainerScanSummaryResultMock(hasRelevancyData bool, relevantLabel c
 		Vulnerabilities: vuls,
 		ContainerScanID: uuid.New().String(),
 		SeverityStats:   cscanlib.SeverityStats{Severity: "Critical"},
+		SeveritiesStats: []cscanlib.SeverityStats{
+			{Severity: "Critical", TotalCount: 1, RelevantCount: 1},
+		},
 	}
 }
 

diff --git a/attackchains/attackchainutils_test.go b/attackchains/attackchainutils_test.go
@@ -58,7 +58,7 @@ func TestIsVulnarableRelevantToAttackChange(t *testing.T) {
 				ImageID:          "ss",
 				HasRelevancyData: false,
 				RelevantLabel:    "no",
-				SeveritiesStats:  []cscanlib.SeverityStats{{Severity: "Critical", RelevantCount: 1}},
+				SeveritiesStats:  []cscanlib.SeverityStats{{Severity: "Critical", RelevantCount: 0, TotalCount: 1}},
 			},
 			expected: true,
 			wantErr:  false,