Extract language and package type from pURLs on SBOM decode (#777)

* add language detection from pURLs Signed-off-by: Alex Goodman <alex.goodman@anchore.com> * add package type detection from pURLs Signed-off-by: Alex Goodman <alex.goodman@anchore.com> * add cargo and npm pURL support Signed-off-by: Alex Goodman <alex.goodman@anchore.com> * fix npm tests and linting Signed-off-by: Alex Goodman <alex.goodman@anchore.com>
2026-11-02 07:26:34 +01:00 · 2022-01-27 09:35:16 -05:00
parent 9f7104d4f1
commit d7a23e4bb2
10 changed files with 287 additions and 21 deletions
@@ -0,0 +1,66 @@
+package pkg
+
+import (
+	"github.com/scylladb/go-set/strset"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestLanguageFromPURL(t *testing.T) {
+
+	tests := []struct {
+		purl string
+		want Language
+	}{
+
+		{
+			purl: "pkg:npm/util@2.32",
+			want: JavaScript,
+		},
+		{
+			purl: "pkg:pypi/util-linux@2.32.1-27.el8",
+			want: Python,
+		},
+		{
+			purl: "pkg:gem/ruby-advisory-db-check@0.12.4",
+			want: Ruby,
+		},
+		{
+			purl: "pkg:golang/github.com/gorilla/context@234fd47e07d1004f0aed9c",
+			want: Go,
+		},
+		{
+			purl: "pkg:cargo/clap@2.33.0",
+			want: Rust,
+		},
+		{
+			purl: "pkg:composer/laravel/laravel@5.5.0",
+			want: PHP,
+		},
+		{
+			purl: "pkg:maven/org.apache.xmlgraphics/batik-anim@1.9.1?type=zip&classifier=dist",
+			want: Java,
+		},
+	}
+
+	var languages []string
+	var expectedLanguages = strset.New()
+	for _, ty := range AllLanguages {
+		expectedLanguages.Add(string(ty))
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.purl, func(t *testing.T) {
+			actual := LanguageFromPURL(tt.purl)
+
+			if actual != "" {
+				languages = append(languages, string(actual))
+			}
+
+			assert.Equalf(t, tt.want, actual, "LanguageFromPURL(%v)", tt.purl)
+		})
+	}
+
+	assert.ElementsMatch(t, expectedLanguages.List(), languages, "missing one or more languages to test against (maybe a package type was added?)")
+
+}