132 lines
2.7 KiB
Go
132 lines
2.7 KiB
Go
package transform
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/drone/drone/yaml"
|
|
|
|
"github.com/franela/goblin"
|
|
)
|
|
|
|
func Test_pull(t *testing.T) {
|
|
g := goblin.Goblin(t)
|
|
g.Describe("pull image", func() {
|
|
|
|
g.It("should be enabled for plugins", func() {
|
|
c := newConfig(&yaml.Container{})
|
|
|
|
ImagePull(c, true)
|
|
g.Assert(c.Pipeline[0].Pull).IsTrue()
|
|
})
|
|
|
|
g.It("should be disabled for plugins", func() {
|
|
c := newConfig(&yaml.Container{})
|
|
|
|
ImagePull(c, false)
|
|
g.Assert(c.Pipeline[0].Pull).IsFalse()
|
|
})
|
|
|
|
g.It("should not apply to commands", func() {
|
|
c := newConfig(&yaml.Container{
|
|
Commands: []string{
|
|
"go build",
|
|
"go test",
|
|
},
|
|
})
|
|
|
|
ImagePull(c, true)
|
|
g.Assert(c.Pipeline[0].Pull).IsFalse()
|
|
})
|
|
|
|
g.It("should not apply to services", func() {
|
|
c := newConfigService(&yaml.Container{
|
|
Image: "mysql",
|
|
})
|
|
|
|
ImagePull(c, true)
|
|
g.Assert(c.Services[0].Pull).IsFalse()
|
|
})
|
|
})
|
|
}
|
|
|
|
func Test_escalate(t *testing.T) {
|
|
|
|
g := goblin.Goblin(t)
|
|
g.Describe("privileged transform", func() {
|
|
|
|
g.It("should handle matches", func() {
|
|
c := newConfig(&yaml.Container{
|
|
Image: "plugins/docker",
|
|
})
|
|
|
|
ImageEscalate(c, []string{"plugins/docker"})
|
|
g.Assert(c.Pipeline[0].Privileged).IsTrue()
|
|
})
|
|
|
|
g.It("should handle glob matches", func() {
|
|
c := newConfig(&yaml.Container{
|
|
Image: "plugins/docker:latest",
|
|
})
|
|
|
|
ImageEscalate(c, []string{"plugins/docker:*"})
|
|
g.Assert(c.Pipeline[0].Privileged).IsTrue()
|
|
})
|
|
|
|
g.It("should handle non matches", func() {
|
|
c := newConfig(&yaml.Container{
|
|
Image: "plugins/git:latest",
|
|
})
|
|
|
|
ImageEscalate(c, []string{"plugins/docker:*"})
|
|
g.Assert(c.Pipeline[0].Privileged).IsFalse()
|
|
})
|
|
|
|
g.It("should handle non glob matches", func() {
|
|
c := newConfig(&yaml.Container{
|
|
Image: "plugins/docker:latest",
|
|
})
|
|
|
|
ImageEscalate(c, []string{"plugins/docker"})
|
|
g.Assert(c.Pipeline[0].Privileged).IsFalse()
|
|
})
|
|
|
|
g.It("should not escalate plugin with commands", func() {
|
|
c := newConfig(&yaml.Container{
|
|
Image: "docker",
|
|
Commands: []string{"echo foo"},
|
|
})
|
|
|
|
err := ImageEscalate(c, []string{"docker"})
|
|
g.Assert(c.Pipeline[0].Privileged).IsFalse()
|
|
g.Assert(err.Error()).Equal("Custom commands disabled for the docker plugin")
|
|
})
|
|
})
|
|
}
|
|
|
|
func Test_normalize(t *testing.T) {
|
|
|
|
g := goblin.Goblin(t)
|
|
g.Describe("normalizing", func() {
|
|
|
|
g.Describe("images", func() {
|
|
|
|
g.It("should append tag if empty", func() {
|
|
c := newConfig(&yaml.Container{
|
|
Image: "golang",
|
|
})
|
|
|
|
ImageTag(c)
|
|
g.Assert(c.Pipeline[0].Image).Equal("golang:latest")
|
|
})
|
|
|
|
g.It("should not override existing tag", func() {
|
|
c := newConfig(&yaml.Container{
|
|
Image: "golang:1.5",
|
|
})
|
|
|
|
ImageTag(c)
|
|
g.Assert(c.Pipeline[0].Image).Equal("golang:1.5")
|
|
})
|
|
})
|
|
})
|
|
}
|