package inject import ( "testing" "github.com/drone/drone/Godeps/_workspace/src/github.com/franela/goblin" ) func Test_Inject(t *testing.T) { g := goblin.Goblin(t) g.Describe("Inject params", func() { g.It("Should replace vars with $$", func() { s := "echo $$FOO $BAR" m := map[string]string{} m["FOO"] = "BAZ" g.Assert("echo BAZ $BAR").Equal(Inject(s, m)) }) g.It("Should not replace vars with single $", func() { s := "echo $FOO $BAR" m := map[string]string{} m["FOO"] = "BAZ" g.Assert(s).Equal(Inject(s, m)) }) g.It("Should not replace vars in nil map", func() { s := "echo $$FOO $BAR" g.Assert(s).Equal(Inject(s, nil)) }) }) } func Test_InjectSafe(t *testing.T) { g := goblin.Goblin(t) g.Describe("Safely Inject params", func() { m := map[string]string{} m["TOKEN"] = "FOO" m["SECRET"] = "BAR" c, _ := parse(InjectSafe(yml, m)) g.It("Should replace vars in notify section", func() { g.Assert(c.Deploy["digital_ocean"].Config["token"]).Equal("FOO") g.Assert(c.Deploy["digital_ocean"].Config["secret"]).Equal("BAR") }) g.It("Should not replace vars in script section", func() { g.Assert(c.Build.Config["commands"].([]interface{})[0]).Equal("echo $$TOKEN") g.Assert(c.Build.Config["commands"].([]interface{})[1]).Equal("echo $$SECRET") }) }) } var yml = ` build: image: foo commands: - echo $$TOKEN - echo $$SECRET deploy: digital_ocean: token: $$TOKEN secret: $$SECRET `