145 lines
4.3 KiB
Go
145 lines
4.3 KiB
Go
// SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
|
|
|
|
package spdxlib
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/spdx/tools-golang/spdx/common"
|
|
"github.com/spdx/tools-golang/spdx/v2_1"
|
|
"github.com/spdx/tools-golang/spdx/v2_2"
|
|
)
|
|
|
|
// ===== 2.1 tests =====
|
|
|
|
func Test2_1FilterForDependencies(t *testing.T) {
|
|
// set up document and some packages and relationships
|
|
doc := &v2_1.Document{
|
|
SPDXVersion: "SPDX-2.1",
|
|
DataLicense: "CC0-1.0",
|
|
SPDXIdentifier: common.ElementID("DOCUMENT"),
|
|
CreationInfo: &v2_1.CreationInfo{},
|
|
Packages: []*v2_1.Package{
|
|
{PackageName: "pkg1", PackageSPDXIdentifier: "p1"},
|
|
{PackageName: "pkg2", PackageSPDXIdentifier: "p2"},
|
|
{PackageName: "pkg3", PackageSPDXIdentifier: "p3"},
|
|
{PackageName: "pkg4", PackageSPDXIdentifier: "p4"},
|
|
{PackageName: "pkg5", PackageSPDXIdentifier: "p5"},
|
|
},
|
|
Relationships: []*v2_1.Relationship{
|
|
{
|
|
RefA: common.MakeDocElementID("", "DOCUMENT"),
|
|
RefB: common.MakeDocElementID("", "p1"),
|
|
Relationship: "DESCRIBES",
|
|
},
|
|
{
|
|
RefA: common.MakeDocElementID("", "DOCUMENT"),
|
|
RefB: common.MakeDocElementID("", "p5"),
|
|
Relationship: "DESCRIBES",
|
|
},
|
|
{
|
|
RefA: common.MakeDocElementID("", "p4"),
|
|
RefB: common.MakeDocElementID("", "DOCUMENT"),
|
|
Relationship: "DESCRIBED_BY",
|
|
},
|
|
{
|
|
RefA: common.MakeDocElementID("", "p1"),
|
|
RefB: common.MakeDocElementID("", "p2"),
|
|
Relationship: "DEPENDS_ON",
|
|
},
|
|
{
|
|
RefA: common.MakeDocElementID("", "p3"),
|
|
RefB: common.MakeDocElementID("", "p4"),
|
|
Relationship: "DEPENDENCY_OF",
|
|
},
|
|
},
|
|
}
|
|
|
|
eIDs, err := FilterRelationships2_1(doc, func(relationship *v2_1.Relationship) *common.ElementID {
|
|
p1EID := common.MakeDocElementID("", "p1")
|
|
if relationship.Relationship == "DEPENDS_ON" && relationship.RefA == p1EID {
|
|
return &relationship.RefB.ElementRefID
|
|
} else if relationship.Relationship == "DEPENDENCY_OF" && relationship.RefB == p1EID {
|
|
return &relationship.RefA.ElementRefID
|
|
}
|
|
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("expected non-nil err, got: %s", err.Error())
|
|
}
|
|
|
|
if len(eIDs) != 1 {
|
|
t.Fatalf("expected 1 ElementID, got: %v", eIDs)
|
|
}
|
|
|
|
if eIDs[0] != common.MakeDocElementID("", "p2").ElementRefID {
|
|
t.Fatalf("received unexpected relationship: %v", eIDs[0])
|
|
}
|
|
}
|
|
|
|
// ===== 2.2 tests =====
|
|
|
|
func Test2_2FindsDependsOnRelationships(t *testing.T) {
|
|
// set up document and some packages and relationships
|
|
doc := &v2_2.Document{
|
|
SPDXVersion: "SPDX-2.2",
|
|
DataLicense: "CC0-1.0",
|
|
SPDXIdentifier: common.ElementID("DOCUMENT"),
|
|
CreationInfo: &v2_2.CreationInfo{},
|
|
Packages: []*v2_2.Package{
|
|
{PackageName: "pkg1", PackageSPDXIdentifier: "p1"},
|
|
{PackageName: "pkg2", PackageSPDXIdentifier: "p2"},
|
|
{PackageName: "pkg3", PackageSPDXIdentifier: "p3"},
|
|
{PackageName: "pkg4", PackageSPDXIdentifier: "p4"},
|
|
{PackageName: "pkg5", PackageSPDXIdentifier: "p5"},
|
|
},
|
|
Relationships: []*v2_2.Relationship{
|
|
{
|
|
RefA: common.MakeDocElementID("", "DOCUMENT"),
|
|
RefB: common.MakeDocElementID("", "p1"),
|
|
Relationship: "DESCRIBES",
|
|
},
|
|
{
|
|
RefA: common.MakeDocElementID("", "DOCUMENT"),
|
|
RefB: common.MakeDocElementID("", "p5"),
|
|
Relationship: "DESCRIBES",
|
|
},
|
|
// inverse relationship -- should also get detected
|
|
{
|
|
RefA: common.MakeDocElementID("", "p4"),
|
|
RefB: common.MakeDocElementID("", "DOCUMENT"),
|
|
Relationship: "DESCRIBED_BY",
|
|
},
|
|
// different relationship
|
|
{
|
|
RefA: common.MakeDocElementID("", "p1"),
|
|
RefB: common.MakeDocElementID("", "p2"),
|
|
Relationship: "DEPENDS_ON",
|
|
},
|
|
},
|
|
}
|
|
|
|
eIDs, err := FilterRelationships2_2(doc, func(relationship *v2_2.Relationship) *common.ElementID {
|
|
p1EID := common.MakeDocElementID("", "p1")
|
|
if relationship.Relationship == "DEPENDS_ON" && relationship.RefA == p1EID {
|
|
return &relationship.RefB.ElementRefID
|
|
} else if relationship.Relationship == "DEPENDENCY_OF" && relationship.RefB == p1EID {
|
|
return &relationship.RefA.ElementRefID
|
|
}
|
|
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("expected non-nil err, got: %s", err.Error())
|
|
}
|
|
|
|
if len(eIDs) != 1 {
|
|
t.Fatalf("expected 1 ElementID, got: %v", eIDs)
|
|
}
|
|
|
|
if eIDs[0] != common.MakeDocElementID("", "p2").ElementRefID {
|
|
t.Fatalf("received unexpected relationship: %v", eIDs[0])
|
|
}
|
|
}
|