Reworked implementation with test coverage

pkg/plugin/deploy/bash.go

@@ -5,9 +5,14 @@ import (
 )
 
 type Bash struct {
-	Command string `yaml:"command,omitempty"`
+	Script  []string `yaml:"script,omitempty"`
+	Command string   `yaml:"command,omitempty"`
 }
 
 func (g *Bash) Write(f *buildfile.Buildfile) {
-	f.WriteCmd(g.Command)
+	g.Script = append(g.Script, g.Command)
+
+	for _, cmd := range g.Script {
+		f.WriteCmd(cmd)
+	}
 }

pkg/plugin/deploy/bash_test.go

@@ -0,0 +1,94 @@
+package deploy
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/drone/drone/pkg/build/buildfile"
+
+	"launchpad.net/goyaml"
+)
+
+// emulate Build struct
+type buildWithBash struct {
+	Deploy *Deploy `yaml:"deploy,omitempty"`
+}
+
+var sampleYmlWithBash = `
+deploy:
+  bash:
+    command: 'echo bash_deployed'
+`
+
+var sampleYmlWithScript = `
+deploy:
+  bash:
+    script:
+      - ./bin/deploy.sh
+      - ./bin/check.sh
+`
+
+var sampleYmlWithBashAndScript = `
+deploy:
+  bash:
+    command: ./bin/some_cmd.sh
+    script:
+      - ./bin/deploy.sh
+      - ./bin/check.sh
+`
+
+func setUpWithBash(input string) (string, error) {
+	var buildStruct buildWithBash
+	err := goyaml.Unmarshal([]byte(input), &buildStruct)
+	if err != nil {
+		return "", err
+	}
+	bf := buildfile.New()
+	buildStruct.Deploy.Write(bf)
+	return bf.String(), err
+}
+
+func TestBashDeployment(t *testing.T) {
+	bscr, err := setUpWithBash(sampleYmlWithBash)
+	if err != nil {
+		t.Fatalf("Can't unmarshal deploy script: %s", err)
+	}
+
+	if !strings.Contains(bscr, "echo bash_deployed") {
+		t.Error("Expect script to contains bash command")
+	}
+}
+
+func TestBashDeploymentWithScript(t *testing.T) {
+	bscr, err := setUpWithBash(sampleYmlWithScript)
+	if err != nil {
+		t.Fatalf("Can't unmarshal deploy script: %s", err)
+	}
+
+	if !strings.Contains(bscr, "./bin/deploy.sh") {
+		t.Error("Expect script to contains bash script")
+	}
+
+	if !strings.Contains(bscr, "./bin/check.sh") {
+		t.Error("Expect script to contains bash script")
+	}
+}
+
+func TestBashDeploymentWithBashAndScript(t *testing.T) {
+	bscr, err := setUpWithBash(sampleYmlWithBashAndScript)
+	if err != nil {
+		t.Fatalf("Can't unmarshal deploy script: %s", err)
+	}
+
+	if !strings.Contains(bscr, "./bin/deploy.sh") {
+		t.Error("Expect script to contains bash script")
+	}
+
+	if !strings.Contains(bscr, "./bin/check.sh") {
+		t.Error("Expect script to contains bash script")
+	}
+
+	if !strings.Contains(bscr, "./bin/some_cmd.sh") {
+		t.Error("Expect script to contains bash script")
+	}
+}