test: make models tests their own package (#423)

This reduces the diff with #183 a bit, and is good practice
google · Jun 27, 2023 · d378b92 · d378b92
1 parent d572130
commit d378b92
Showing 1 changed file with 12 additions and 11 deletions.
diff --git a/pkg/models/vulnerability_test.go b/pkg/models/vulnerability_test.go
@@ -1,19 +1,20 @@
-package models
+package models_test
 
 import (
 	"encoding/json"
 	"testing"
 
+	"github.com/google/osv-scanner/pkg/models"
 	"gopkg.in/yaml.v3"
 )
 
 func TestAffected_MarshalJSONWithPackage(t *testing.T) {
 	t.Parallel()
-	v := Vulnerability{
+	v := models.Vulnerability{
 		ID: "TEST-0000",
-		Affected: []Affected{
+		Affected: []models.Affected{
 			{
-				Package:  Package{Ecosystem: EcosystemPyPI, Name: "requests"},
+				Package:  models.Package{Ecosystem: models.EcosystemPyPI, Name: "requests"},
 				Versions: []string{"1.0.0"},
 			},
 		},
@@ -30,9 +31,9 @@ func TestAffected_MarshalJSONWithPackage(t *testing.T) {
 
 func TestAffected_MarshalJSONWithoutPackage(t *testing.T) {
 	t.Parallel()
-	v := Vulnerability{
+	v := models.Vulnerability{
 		ID: "TEST-0000",
-		Affected: []Affected{
+		Affected: []models.Affected{
 			{
 				Versions: []string{"1.0.0"},
 			},
@@ -50,11 +51,11 @@ func TestAffected_MarshalJSONWithoutPackage(t *testing.T) {
 
 func TestAffected_MarshalYAMLWithPackage(t *testing.T) {
 	t.Parallel()
-	v := Vulnerability{
+	v := models.Vulnerability{
 		ID: "TEST-0000",
-		Affected: []Affected{
+		Affected: []models.Affected{
 			{
-				Package:  Package{Ecosystem: EcosystemPyPI, Name: "requests"},
+				Package:  models.Package{Ecosystem: models.EcosystemPyPI, Name: "requests"},
 				Versions: []string{"1.0.0"},
 			},
 		},
@@ -79,9 +80,9 @@ affected:
 
 func TestAffected_MarshalYAMLWithoutPackage(t *testing.T) {
 	t.Parallel()
-	v := Vulnerability{
+	v := models.Vulnerability{
 		ID: "TEST-0000",
-		Affected: []Affected{
+		Affected: []models.Affected{
 			{
 				Versions: []string{"1.0.0"},
 			},