49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
|
package buildfile
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestWrite(t *testing.T) {
|
||
|
|
|
||
|
|
var f = New()
|
||
|
|
var got, want = f.String(), base
|
||
|
|
if got != want {
|
||
|
|
t.Errorf("Exepected New() returned %s, got %s", want, got)
|
||
|
|
}
|
||
|
|
|
||
|
|
f = &Buildfile{}
|
||
|
|
f.WriteCmd("echo hi")
|
||
|
|
got, want = f.String(), "echo '#DRONE:6563686f206869'\necho hi\n"
|
||
|
|
if got != want {
|
||
|
|
t.Errorf("Exepected WriteCmd returned %s, got %s", want, got)
|
||
|
|
}
|
||
|
|
|
||
|
|
f = &Buildfile{}
|
||
|
|
f.WriteCmdSilent("echo hi")
|
||
|
|
got, want = f.String(), "echo hi\n"
|
||
|
|
if got != want {
|
||
|
|
t.Errorf("Exepected WriteCmdSilent returned %s, got %s", want, got)
|
||
|
|
}
|
||
|
|
|
||
|
|
f = &Buildfile{}
|
||
|
|
f.WriteComment("this is a comment")
|
||
|
|
got, want = f.String(), "#this is a comment\n"
|
||
|
|
if got != want {
|
||
|
|
t.Errorf("Exepected WriteComment returned %s, got %s", want, got)
|
||
|
|
}
|
||
|
|
|
||
|
|
f = &Buildfile{}
|
||
|
|
f.WriteEnv("FOO", "BAR")
|
||
|
|
got, want = f.String(), "export FOO=BAR\n"
|
||
|
|
if got != want {
|
||
|
|
t.Errorf("Exepected WriteEnv returned %s, got %s", want, got)
|
||
|
|
}
|
||
|
|
|
||
|
|
f = &Buildfile{}
|
||
|
|
f.WriteHost("127.0.0.1")
|
||
|
|
got, want = f.String(), "[ -f /usr/bin/sudo ] || echo \"127.0.0.1\" | tee -a /etc/hosts\n[ -f /usr/bin/sudo ] && echo \"127.0.0.1\" | sudo tee -a /etc/hosts\n"
|
||
|
|
if got != want {
|
||
|
|
t.Errorf("Exepected WriteHost returned %s, got %s", want, got)
|
||
|
|
}
|
||
|
|
}
|