Return error for 500 errors from server (AST-000)

Author: cx-daniel-greenspan

go.mod

@@ -28,7 +28,6 @@ require (
 	golang.org/x/text v0.22.0
 	google.golang.org/grpc v1.67.3
 	google.golang.org/protobuf v1.36.3
-	gopkg.in/yaml.v2 v2.4.0
 	gopkg.in/yaml.v3 v3.0.1
 	gotest.tools v2.2.0+incompatible
 )
@@ -286,6 +285,7 @@ require (
 	gopkg.in/inf.v0 v0.9.1 // indirect
 	gopkg.in/ini.v1 v1.67.0 // indirect
 	gopkg.in/warnings.v0 v0.1.2 // indirect
+	gopkg.in/yaml.v2 v2.4.0 // indirect
 	helm.sh/helm/v3 v3.17.0 // indirect
 	k8s.io/api v0.32.0 // indirect
 	k8s.io/apiextensions-apiserver v0.32.0 // indirect

internal/wrappers/mock/risk-management-mock.go

@@ -8,8 +8,8 @@ type RiskManagementMockWrapper struct{}
 
 func (r *RiskManagementMockWrapper) GetTopVulnerabilitiesByProjectID(projectID string) (*wrappers.ASPMResult, *wrappers.WebError, error) {
 	mockResults := []wrappers.RiskManagementResult{
-		{ID: "1", Name: "Vuln1", Severity: "High"},
-		{ID: "2", Name: "Vuln2", Severity: "Medium"},
+		{ID: "1", Name: "Vuln1", Severity: "High", Traits: map[string]string{wrappers.ExplPathKey: wrappers.ExplPathValue}},
+		{ID: "2", Name: "Vuln2", Severity: "Medium", Traits: map[string]string{wrappers.SuspMalwareKey: wrappers.SuspMalwareValue}},
 	}
 
 	mockASPMResult := &wrappers.ASPMResult{

internal/wrappers/risk-management.go

@@ -2,6 +2,18 @@ package wrappers
 
 import "time"
 
+const (
+	SuspMalwareKey = "suspMalware"
+	ExplPathKey    = "explPath"
+	PubExposedKey  = "pubExposed"
+	RuntimeKey     = "runtime"
+
+	SuspMalwareValue = "Suspected Malware"
+	ExplPathValue    = "Exploitable Path"
+	PubExposedValue  = "Public Exposed"
+	RuntimeValue     = "Runtime"
+)
+
 type RiskManagementWrapper interface {
 	GetTopVulnerabilitiesByProjectID(projectID string) (*ASPMResult, *WebError, error)
 }
@@ -27,6 +39,7 @@ type RiskManagementResult struct {
 	Severity           string             `json:"severity"`
 	RiskScore          float64            `json:"riskScore"`
 	EnrichmentSources  map[string]string  `json:"enrichmentSources"`
+	Traits             map[string]string  `json:"traits"`
 	CreatedAt          time.Time          `json:"createdAt"`
 	ApplicationsScores []ApplicationScore `json:"applicationsScores"`
 }