56 lines
1.5 KiB
Go
56 lines
1.5 KiB
Go
|
package updater
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/coreos/clair/database"
|
||
|
|
"github.com/coreos/clair/utils/types"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDoVulnerabilityNamespacing(t *testing.T) {
|
||
|
|
fv1 := database.FeatureVersion{
|
||
|
|
Feature: database.Feature{
|
||
|
|
Namespace: database.Namespace{Name: "Namespace1"},
|
||
|
|
Name: "Feature1",
|
||
|
|
},
|
||
|
|
Version: types.NewVersionUnsafe("0.1"),
|
||
|
|
}
|
||
|
|
|
||
|
|
fv2 := database.FeatureVersion{
|
||
|
|
Feature: database.Feature{
|
||
|
|
Namespace: database.Namespace{Name: "Namespace2"},
|
||
|
|
Name: "Feature1",
|
||
|
|
},
|
||
|
|
Version: types.NewVersionUnsafe("0.2"),
|
||
|
|
}
|
||
|
|
|
||
|
|
fv3 := database.FeatureVersion{
|
||
|
|
Feature: database.Feature{
|
||
|
|
Namespace: database.Namespace{Name: "Namespace2"},
|
||
|
|
Name: "Feature2",
|
||
|
|
},
|
||
|
|
Version: types.NewVersionUnsafe("0.3"),
|
||
|
|
}
|
||
|
|
|
||
|
|
vulnerability := database.Vulnerability{
|
||
|
|
Name: "DoVulnerabilityNamespacing",
|
||
|
|
FixedIn: []database.FeatureVersion{fv1, fv2, fv3},
|
||
|
|
}
|
||
|
|
|
||
|
|
vulnerabilities := DoVulnerabilityNamespacing(vulnerability)
|
||
|
|
for _, vulnerability := range vulnerabilities {
|
||
|
|
switch vulnerability.Namespace.Name {
|
||
|
|
case fv1.Feature.Namespace.Name:
|
||
|
|
assert.Len(t, vulnerability.FixedIn, 1)
|
||
|
|
assert.Contains(t, vulnerability.FixedIn, fv1)
|
||
|
|
case fv2.Feature.Namespace.Name:
|
||
|
|
assert.Len(t, vulnerability.FixedIn, 2)
|
||
|
|
assert.Contains(t, vulnerability.FixedIn, fv2)
|
||
|
|
assert.Contains(t, vulnerability.FixedIn, fv3)
|
||
|
|
default:
|
||
|
|
t.Errorf("Should not have a Vulnerability with '%s' as its Namespace.", vulnerability.Namespace.Name)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|