drone-yaml/yaml/compiler/transform/labels_test.go
2019-01-22 15:44:17 -08:00

49 lines
1.0 KiB
Go

package transform
import (
"testing"
"github.com/drone/drone-runtime/engine"
"github.com/google/go-cmp/cmp"
)
func TestWithLabels(t *testing.T) {
volume := &engine.Volume{
Metadata: engine.Metadata{
Labels: map[string]string{},
},
}
step := &engine.Step{
Metadata: engine.Metadata{
Labels: map[string]string{},
},
Envs: map[string]string{},
}
spec := &engine.Spec{
Metadata: engine.Metadata{
Labels: map[string]string{},
},
Steps: []*engine.Step{step},
Docker: &engine.DockerConfig{
Volumes: []*engine.Volume{volume},
},
}
labels := map[string]string{
"io.drone.build.number": "1",
"io.drone.build.event": "push",
}
WithLables(labels)(spec)
if diff := cmp.Diff(labels, spec.Metadata.Labels); diff != "" {
t.Errorf("Unexpected spec labels")
t.Log(diff)
}
if diff := cmp.Diff(labels, step.Metadata.Labels); diff != "" {
t.Errorf("Unexpected step labels")
t.Log(diff)
}
if diff := cmp.Diff(labels, volume.Metadata.Labels); diff != "" {
t.Errorf("Unexpected volume labels")
t.Log(diff)
}
}