67 lines
1.9 KiB
Go
67 lines
1.9 KiB
Go
|
package featurens_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
|
||
|
|
"github.com/coreos/clair/database"
|
||
|
|
"github.com/coreos/clair/ext/featurens"
|
||
|
|
"github.com/coreos/clair/pkg/tarutil"
|
||
|
|
|
||
|
|
_ "github.com/coreos/clair/ext/featurens/alpinerelease"
|
||
|
|
_ "github.com/coreos/clair/ext/featurens/aptsources"
|
||
|
|
_ "github.com/coreos/clair/ext/featurens/lsbrelease"
|
||
|
|
_ "github.com/coreos/clair/ext/featurens/osrelease"
|
||
|
|
_ "github.com/coreos/clair/ext/featurens/redhatrelease"
|
||
|
|
)
|
||
|
|
|
||
|
|
type MultipleNamespaceTestData struct {
|
||
|
|
Files tarutil.FilesMap
|
||
|
|
ExpectedNamespaces []database.Namespace
|
||
|
|
}
|
||
|
|
|
||
|
|
func assertnsNameEqual(t *testing.T, nslist_expected, nslist []database.Namespace) {
|
||
|
|
assert.Equal(t, len(nslist_expected), len(nslist))
|
||
|
|
expected := map[string]struct{}{}
|
||
|
|
input := map[string]struct{}{}
|
||
|
|
// compare the two sets
|
||
|
|
for i := range nslist_expected {
|
||
|
|
expected[nslist_expected[i].Name] = struct{}{}
|
||
|
|
input[nslist[i].Name] = struct{}{}
|
||
|
|
}
|
||
|
|
assert.Equal(t, expected, input)
|
||
|
|
}
|
||
|
|
|
||
|
|
func testMultipleNamespace(t *testing.T, testData []MultipleNamespaceTestData) {
|
||
|
|
for _, td := range testData {
|
||
|
|
nslist, err := featurens.Detect(td.Files)
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assertnsNameEqual(t, td.ExpectedNamespaces, nslist)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMultipleNamespaceDetector(t *testing.T) {
|
||
|
|
testData := []MultipleNamespaceTestData{
|
||
|
|
{
|
||
|
|
ExpectedNamespaces: []database.Namespace{
|
||
|
|
database.Namespace{Name: "debian:8", VersionFormat: "dpkg"},
|
||
|
|
database.Namespace{Name: "alpine:v3.3", VersionFormat: "dpkg"},
|
||
|
|
},
|
||
|
|
Files: tarutil.FilesMap{
|
||
|
|
"etc/os-release": []byte(`
|
||
|
|
PRETTY_NAME="Debian GNU/Linux 8 (jessie)"
|
||
|
|
NAME="Debian GNU/Linux"
|
||
|
|
VERSION_ID="8"
|
||
|
|
VERSION="8 (jessie)"
|
||
|
|
ID=debian
|
||
|
|
HOME_URL="http://www.debian.org/"
|
||
|
|
SUPPORT_URL="http://www.debian.org/support/"
|
||
|
|
BUG_REPORT_URL="https://bugs.debian.org/"`),
|
||
|
|
"etc/alpine-release": []byte(`3.3.4`),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
testMultipleNamespace(t, testData)
|
||
|
|
}
|