mirror of https://github.com/docker/buildx.git
120 lines
2.6 KiB
Go
120 lines
2.6 KiB
Go
package buildflags
|
|
|
|
import (
|
|
"cmp"
|
|
"slices"
|
|
"testing"
|
|
|
|
"github.com/moby/buildkit/exporter/containerimage/exptypes"
|
|
ocispecs "github.com/opencontainers/image-spec/specs-go/v1"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestParseAnnotations(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
in []string
|
|
want map[exptypes.AnnotationKey]string
|
|
wantErr string
|
|
}{
|
|
{
|
|
name: "basic",
|
|
in: []string{"a=b"},
|
|
want: map[exptypes.AnnotationKey]string{
|
|
{Key: "a"}: "b",
|
|
},
|
|
},
|
|
{
|
|
name: "reverse-DNS key",
|
|
in: []string{"com.example=a"},
|
|
want: map[exptypes.AnnotationKey]string{
|
|
{Key: "com.example"}: "a",
|
|
},
|
|
},
|
|
{
|
|
name: "specify type",
|
|
in: []string{"manifest:com.example=a"},
|
|
want: map[exptypes.AnnotationKey]string{
|
|
{Type: "manifest", Key: "com.example"}: "a",
|
|
},
|
|
},
|
|
{
|
|
name: "specify bad type",
|
|
in: []string{"bad:com.example=a"},
|
|
wantErr: "unknown annotation type",
|
|
},
|
|
{
|
|
name: "specify type and platform",
|
|
in: []string{"manifest[plat/form]:com.example=a"},
|
|
want: map[exptypes.AnnotationKey]string{
|
|
{
|
|
Type: "manifest",
|
|
Platform: &ocispecs.Platform{
|
|
OS: "plat",
|
|
Architecture: "form",
|
|
},
|
|
Key: "com.example",
|
|
}: "a",
|
|
},
|
|
},
|
|
{
|
|
name: "specify multiple types",
|
|
in: []string{"index,manifest:com.example=a"},
|
|
want: map[exptypes.AnnotationKey]string{
|
|
{Type: "index", Key: "com.example"}: "a",
|
|
{Type: "manifest", Key: "com.example"}: "a",
|
|
},
|
|
},
|
|
{
|
|
name: "specify multiple types and platform",
|
|
in: []string{"index,manifest[plat/form]:com.example=a"},
|
|
want: map[exptypes.AnnotationKey]string{
|
|
{Type: "index", Key: "com.example"}: "a",
|
|
{
|
|
Type: "manifest",
|
|
Platform: &ocispecs.Platform{
|
|
OS: "plat",
|
|
Architecture: "form",
|
|
},
|
|
Key: "com.example",
|
|
}: "a",
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
got, err := ParseAnnotations(test.in)
|
|
if test.wantErr != "" {
|
|
require.ErrorContains(t, err, test.wantErr)
|
|
} else {
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
// Can't compare maps with pointer in their keys, need to extract and sort the map entries
|
|
wantKVs := entries(test.want)
|
|
gotKVs := entries(got)
|
|
|
|
assert.Equal(t, wantKVs, gotKVs)
|
|
})
|
|
}
|
|
}
|
|
|
|
type kv struct {
|
|
Key exptypes.AnnotationKey
|
|
Val string
|
|
}
|
|
|
|
func entries(in map[exptypes.AnnotationKey]string) []kv {
|
|
var out []kv
|
|
for k, v := range in {
|
|
out = append(out, kv{k, v})
|
|
}
|
|
|
|
sortFunc := func(a, b kv) int { return cmp.Compare(a.Key.String(), b.Key.String()) }
|
|
slices.SortFunc(out, sortFunc)
|
|
|
|
return out
|
|
}
|