|
| 1 | +package parser |
| 2 | + |
| 3 | +import ( |
| 4 | + "os" |
| 5 | + "io/ioutil" |
| 6 | + "testing" |
| 7 | + "github.com/stretchr/testify/assert" |
| 8 | + "github.com/google/osv-scanner/pkg/lockfile" |
| 9 | + cdx "github.com/CycloneDX/cyclonedx-go" |
| 10 | +) |
| 11 | + |
| 12 | +func TestParseCyclonedxSBOM(t *testing.T) { |
| 13 | + // Create a sample SBOM JSON file |
| 14 | + tempFile, _ := ioutil.TempFile("", "sbom_*.json") |
| 15 | + defer os.Remove(tempFile.Name()) |
| 16 | + sbomContent := `{ |
| 17 | + "Components": [ |
| 18 | + { |
| 19 | + "group": "testGroup", |
| 20 | + "name": "testName", |
| 21 | + "version": "1.0", |
| 22 | + "bom-ref": "pkg:pypi" |
| 23 | + }, |
| 24 | + { |
| 25 | + "group": "testGroup2", |
| 26 | + "name": "testName2", |
| 27 | + "version": "2.0", |
| 28 | + "bom-ref": "pkg:npm" |
| 29 | + } |
| 30 | + ] |
| 31 | + }` |
| 32 | + err := ioutil.WriteFile(tempFile.Name(), []byte(sbomContent), 0644) |
| 33 | + assert.Nil(t, err) |
| 34 | + |
| 35 | + packages, err := parseCyclonedxSBOM(tempFile.Name()) |
| 36 | + |
| 37 | + assert.Nil(t, err) |
| 38 | + assert.Len(t, packages, 2) |
| 39 | + assert.Equal(t, "testGroup:testName", packages[0].Name) |
| 40 | + assert.Equal(t, "testGroup2:testName2", packages[1].Name) |
| 41 | +} |
| 42 | + |
| 43 | +func TestConvertSbomComponent2LPD(t *testing.T) { |
| 44 | + component := cdx.Component{ |
| 45 | + Group: "testGroup", |
| 46 | + Name: "testName", |
| 47 | + Version: "1.0", |
| 48 | + BOMRef: "pkg:pypi", |
| 49 | + } |
| 50 | + |
| 51 | + pd, err := convertSbomComponent2LPD(&component) |
| 52 | + |
| 53 | + assert.Nil(t, err) |
| 54 | + assert.Equal(t, "testGroup:testName", pd.Name) |
| 55 | + assert.Equal(t, "1.0", pd.Version) |
| 56 | + assert.Equal(t, lockfile.PipEcosystem, pd.Ecosystem) |
| 57 | +} |
| 58 | + |
| 59 | +func TestConvertBomRefAsEcosystem(t *testing.T) { |
| 60 | + ecosystem, err := convertBomRefAsEcosystem("pkg:pypi") |
| 61 | + |
| 62 | + assert.Nil(t, err) |
| 63 | + assert.Equal(t, lockfile.PipEcosystem, ecosystem) |
| 64 | + |
| 65 | + ecosystem, err = convertBomRefAsEcosystem("pkg:npm") |
| 66 | + |
| 67 | + assert.Nil(t, err) |
| 68 | + assert.Equal(t, lockfile.NpmEcosystem, ecosystem) |
| 69 | + |
| 70 | + ecosystem, err = convertBomRefAsEcosystem("pkg:unknown") |
| 71 | + |
| 72 | + assert.NotNil(t, err) |
| 73 | + assert.Equal(t, lockfile.NpmEcosystem, ecosystem) // As per your code, it defaults to lockfile.NpmEcosystem |
| 74 | +} |
0 commit comments