feat: refactor cvss property name

golang · Jun 23, 2021 · 4d4b7f4 · 4d4b7f4
1 parent 8503ad3
commit 4d4b7f4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/report/report.go b/report/report.go
@@ -31,9 +31,9 @@ type CVEMeta struct {
 	CVSSMeta    *CVSS   `yaml:",omitempty"`
 }
 type CVSS struct {
-	Version   string  `yaml:",omitempty"`
-	BaseScore float32 `yaml:",omitempty"`
-	Vector    string  `yaml:",omitempty"`
+	Version string  `yaml:",omitempty"`
+	Score   float32 `yaml:",omitempty"`
+	Vector  string  `yaml:",omitempty"`
 }
 
 type Report struct {

diff --git a/report/utils.go b/report/utils.go
@@ -8,9 +8,9 @@ func CvssScoreToSeverity(cvss *CVSS) string {
 	}
 	switch cvss.Version {
 	case "v2":
-		return cvssV2SeverityByScore(cvss.BaseScore)
+		return cvssV2SeverityByScore(cvss.Score)
 	case "v3":
-		return cvssV3SeverityByScore(cvss.BaseScore)
+		return cvssV3SeverityByScore(cvss.Score)
 	default:
 		return ""
 	}

diff --git a/report/utils_test.go b/report/utils_test.go
@@ -23,7 +23,7 @@ func TestReverseString1(t *testing.T) {
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			if got := CvssScoreToSeverity(&CVSS{Version: tt.version, BaseScore: tt.baseScore}); got != tt.want {
+			if got := CvssScoreToSeverity(&CVSS{Version: tt.version, Score: tt.baseScore}); got != tt.want {
 				t.Errorf("CvssScoreToSeverity() = %v, want %v", got, tt.want)
 			}
 		})