chore: use trivy code (#475)

Signed-off-by: guoguangwu <guoguangwu@magic-shield.com>
project-copacetic · Jan 20, 2024 · 45e40f5 · 45e40f5
1 parent 86dded1
commit 45e40f5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/report/trivy.go b/pkg/report/trivy.go
@@ -37,7 +37,7 @@ func (t *TrivyParser) Parse(file string) (*unversioned.UpdateManifest, error) {
 	result := trivyTypes.Result{}
 	for i := range report.Results {
 		r := &report.Results[i]
-		if r.Class == "os-pkgs" {
+		if r.Class == trivyTypes.ClassOSPkg {
 			if result.Class != "" {
 				return nil, errors.New("unexpected multiple results for os-pkgs")
 			}

diff --git a/pkg/report/trivy_test.go b/pkg/report/trivy_test.go
@@ -37,7 +37,7 @@ func TestParseTrivyReport(t *testing.T) {
 				Results: []trivyTypes.Result{
 					{
 						Target: "alpine:3.14.0 (alpine 3.14.0)",
-						Class:  "os-pkgs",
+						Class:  trivyTypes.ClassOSPkg,
 						Type:   "alpine",
 						Vulnerabilities: []trivyTypes.DetectedVulnerability{
 							{