diff --git a/pkg/advisory/import.go b/pkg/advisory/import.go index 1fbeb99c..ecce41ae 100644 --- a/pkg/advisory/import.go +++ b/pkg/advisory/import.go @@ -15,13 +15,8 @@ import ( ) // ImporAdvisoriesYAML import and yaml Advisories data and present as a config index struct -func ImporAdvisoriesYAML(inputData string) (tempDir string, documents *configs.Index[v2.Document], err error) { - inputAdv, err := os.ReadFile(inputData) - if err != nil { - return "", nil, fmt.Errorf("unable to create output file: %v", err) - } - - yamlDocs := bytes.Split(inputAdv, []byte("\n---\n")) +func ImporAdvisoriesYAML(inputData []byte) (tempDir string, documents *configs.Index[v2.Document], err error) { + yamlDocs := bytes.Split(inputData, []byte("\n---\n")) // Unmarshal YAML documents var docs []v2.Document for _, doc := range yamlDocs { diff --git a/pkg/advisory/import_test.go b/pkg/advisory/import_test.go index b055f327..e25289c3 100644 --- a/pkg/advisory/import_test.go +++ b/pkg/advisory/import_test.go @@ -2,6 +2,7 @@ package advisory import ( "context" + "os" "testing" "github.com/stretchr/testify/require" @@ -29,7 +30,10 @@ func Test_ImportAdvisoriesYAML(t *testing.T) { advisoryDocs, err := v2.NewIndex(context.Background(), advisoryFsys) require.NoError(t, err) - _, importedDocuments, err := ImporAdvisoriesYAML(tt.pathToInputData) + b, err := os.ReadFile(tt.pathToInputData) + require.NoError(t, err) + + _, importedDocuments, err := ImporAdvisoriesYAML(b) require.NoError(t, err) require.Equal(t, advisoryDocs.Select().Len(), importedDocuments.Select().Len()) })