test(pnpm/v5): refactor test cases

Author: iseki0

module/pnpm/v5/data_test.go

@@ -1,26 +1,12 @@
 package v5
 
-import _ "embed"
-
-//go:embed testdata/1.yaml
-var testData1 string
-
-//go:embed testdata/2.yaml
-var testData2 string
-
-//go:embed testdata/3.yaml
-var testData3 string
-
-//go:embed testdata/4.yaml
-var testData4 string
+import (
+	"embed"
+	_ "embed"
+)
 
 //go:embed testdata/5.yaml
 var testData5 string
 
-var testDataList = []string{
-	testData1,
-	testData2,
-	testData3,
-	testData4,
-	testData5,
-}
+//go:embed testdata/*
+var testFiles embed.FS

module/pnpm/v5/v5_test.go

@@ -4,19 +4,31 @@ import (
 	"encoding/json"
 	"github.com/murphysecurity/murphysec/utils/must"
 	"github.com/stretchr/testify/assert"
+	"io"
 	"testing"
 )
 
-func TestParseLockfile(t *testing.T) {
-	for i, s := range testDataList {
-		lockfile, e := ParseLockfile([]byte(s))
-		assert.NoError(t, e, i)
-		assert.NotNil(t, lockfile, i)
+func TestLockfile(t *testing.T) {
+	files, _ := testFiles.ReadDir("testdata")
+	assert.NotEmpty(t, files)
+	for _, s := range files {
+		t.Run(s.Name(), func(t *testing.T) {
+			f, e := testFiles.Open("testdata/" + s.Name())
+			assert.NoError(t, e)
+			defer func() { assert.NoError(t, f.Close()) }()
+			data, e := io.ReadAll(f)
+			assert.NoError(t, e)
+			lockfile, e := ParseLockfile(data)
+			assert.NoError(t, e)
+			assert.NotNil(t, lockfile)
+
+		})
 	}
 }
 
 func TestBuildDepTree(t *testing.T) {
-	l, _ := ParseLockfile([]byte(testDataList[4]))
+
+	l, _ := ParseLockfile([]byte(testData5))
 	tree := BuildDepTree(l, nil, "")
 	assert.NotNil(t, tree)
 	t.Log(string(must.A(json.MarshalIndent(tree, "| ", " "))))