func (S) TestScriptIsolatesEnv(c *C) {
p := pipe.Script(
pipe.SetEnvVar("PIPE_VAR", "outer"),
pipe.Script(
pipe.SetEnvVar("PIPE_VAR", "inner"),
),
pipe.System("echo $PIPE_VAR"),
)
output, err := pipe.Output(p)
c.Assert(err, IsNil)
c.Assert(string(output), Equals, "outer\n")
}
func (S) TestScriptIsolatesDir(c *C) {
dir1 := c.MkDir()
dir2 := c.MkDir()
p := pipe.Script(
pipe.ChDir(dir1),
pipe.Script(
pipe.ChDir(dir2),
),
pipe.System("echo $PWD"),
)
output, err := pipe.Output(p)
c.Assert(err, IsNil)
c.Assert(string(output), Equals, dir1+"\n")
}
func (S) TestSetEnvVar(c *C) {
os.Setenv("PIPE_NEW_VAR", "")
os.Setenv("PIPE_OLD_VAR", "old")
defer os.Setenv("PIPE_OLD_VAR", "")
p := pipe.Script(
pipe.SetEnvVar("PIPE_NEW_VAR", "new"),
pipe.System("echo $PIPE_OLD_VAR $PIPE_NEW_VAR"),
pipe.SetEnvVar("PIPE_NEW_VAR", "after"),
func(s *pipe.State) error {
count := 0
prefix := "PIPE_NEW_VAR="
for _, kv := range s.Env {
if strings.HasPrefix(kv, prefix) {
count++
}
}
if count != 1 {
return fmt.Errorf("found %d environment variables", count)
}
return nil
},
)
output, err := pipe.Output(p)
c.Assert(err, IsNil)
c.Assert(string(output), Equals, "old new\n")
c.Assert(os.Getenv("PIPE_NEW_VAR"), Equals, "")
}
func (S) TestKillAbortedExecTask(c *C) {
p := pipe.Script(
pipe.TaskFunc(func(*pipe.State) error { return fmt.Errorf("boom") }),
pipe.Exec("will-not-run"),
)
_, err := pipe.Output(p)
c.Assert(err, ErrorMatches, "boom")
}
func (S) TestScriptCombinedOutput(c *C) {
p := pipe.Script(
pipe.System("echo out1; echo err1 1>&2; echo out2; echo err2 1>&2"),
pipe.System("echo out3; echo err3 1>&2; echo out4; echo err4 1>&2"),
)
output, err := pipe.CombinedOutput(p)
c.Assert(err, IsNil)
c.Assert(string(output), Equals, "out1\nerr1\nout2\nerr2\nout3\nerr3\nout4\nerr4\n")
}
func (S) TestScriptPreservesStreams(c *C) {
p := pipe.Script(
pipe.Line(
pipe.Print("hello\n"),
pipe.Discard(),
),
pipe.Exec("echo", "world"),
)
output, err := pipe.Output(p)
c.Assert(err, IsNil)
c.Assert(string(output), Equals, "world\n")
}
func (S) TestScriptNesting(c *C) {
b := &bytes.Buffer{}
p := pipe.Line(
pipe.Print("hello"),
pipe.Script(
pipe.Print("world"),
pipe.Exec("sed", "s/l/k/g"),
),
pipe.Write(b),
)
err := pipe.Run(p)
c.Assert(err, IsNil)
c.Assert(b.String(), Equals, "worldhekko")
}
func (S) TestRenameFileAbsolute(c *C) {
dir := c.MkDir()
from := filepath.Join(dir, "from")
to := filepath.Join(dir, "to")
p := pipe.Script(
pipe.WriteFile(from, 0644),
pipe.RenameFile(from, to),
)
err := pipe.Run(p)
c.Assert(err, IsNil)
_, err = os.Stat(from)
c.Assert(err, NotNil)
_, err = os.Stat(to)
c.Assert(err, IsNil)
}
func (S) TestWriteFileRelative(c *C) {
dir := c.MkDir()
path := filepath.Join(dir, "file")
p := pipe.Script(
pipe.ChDir(dir),
pipe.Line(
pipe.Print("hello"),
pipe.Exec("sed", "s/l/k/g"),
pipe.WriteFile("file", 0600),
),
)
output, err := pipe.Output(p)
c.Assert(err, IsNil)
c.Assert(string(output), Equals, "")
data, err := ioutil.ReadFile(path)
c.Assert(err, IsNil)
c.Assert(string(data), Equals, "hekko")
}
func (S) TestMkDir(c *C) {
dir := c.MkDir()
subdir := filepath.Join(dir, "subdir")
subsubdir := filepath.Join(subdir, "subsubdir")
p := pipe.Script(
pipe.MkDir(subdir, 0755), // Absolute
pipe.ChDir(subdir),
pipe.MkDir("subsubdir", 0700), // Relative
pipe.ChDir("subsubdir"),
pipe.System("echo $PWD"),
)
output, err := pipe.Output(p)
c.Assert(err, IsNil)
c.Assert(string(output), Equals, subsubdir+"\n")
stat, err := os.Stat(subsubdir)
c.Assert(err, IsNil)
c.Assert(stat.Mode()&os.ModePerm, Equals, os.FileMode(0700))
}
func (S) TestAppendFileAbsolute(c *C) {
path := filepath.Join(c.MkDir(), "file")
p := pipe.Script(
pipe.Line(
pipe.Print("hello "),
pipe.AppendFile(path, 0600),
),
pipe.Line(
pipe.Print("world!"),
pipe.AppendFile(path, 0600),
),
)
output, err := pipe.Output(p)
c.Assert(err, IsNil)
c.Assert(string(output), Equals, "")
data, err := ioutil.ReadFile(path)
c.Assert(err, IsNil)
c.Assert(string(data), Equals, "hello world!")
}
func (S) TestChDir(c *C) {
wd1, err := os.Getwd()
c.Assert(err, IsNil)
dir := c.MkDir()
subdir := filepath.Join(dir, "subdir")
err = os.Mkdir(subdir, 0755)
p := pipe.Script(
pipe.ChDir(dir),
pipe.ChDir("subdir"),
pipe.System("echo $PWD"),
)
output, err := pipe.Output(p)
c.Assert(err, IsNil)
c.Assert(string(output), Equals, subdir+"\n")
wd2, err := os.Getwd()
c.Assert(err, IsNil)
c.Assert(wd2, Equals, wd1)
}
func (S) TestErrorHandling(c *C) {
sync := make(chan bool)
p := pipe.Script(
pipe.Line(
pipe.TaskFunc(func(*pipe.State) error {
sync <- true
return fmt.Errorf("err1")
}),
pipe.TaskFunc(func(*pipe.State) error {
<-sync
return fmt.Errorf("err2")
}),
),
pipe.Print("never happened"),
)
output, err := pipe.Output(p)
if err.Error() != "err1; err2" && err.Error() != "err2; err1" {
c.Fatalf(`want "err1; err2" or "err2; err1"; got %q`, err.Error())
}
c.Assert(string(output), Equals, "")
}
func (S) TestTeeAppendFileRelative(c *C) {
dir := c.MkDir()
path := filepath.Join(dir, "file")
p := pipe.Script(
pipe.ChDir(dir),
pipe.Line(
pipe.Print("hello "),
pipe.TeeAppendFile("file", 0600),
),
pipe.Line(
pipe.Print("world!"),
pipe.TeeAppendFile("file", 0600),
),
)
output, err := pipe.Output(p)
c.Assert(err, IsNil)
c.Assert(string(output), Equals, "hello world!")
data, err := ioutil.ReadFile(path)
c.Assert(err, IsNil)
c.Assert(string(data), Equals, "hello world!")
}
func (S) TestTeeAppendFileAbsolute(c *C) {
path := filepath.Join(c.MkDir(), "file")
p := pipe.Script(
pipe.Line(
pipe.Print("hello "),
pipe.TeeAppendFile(path, 0600),
),
pipe.Line(
pipe.Print("world!"),
pipe.TeeAppendFile(path, 0600),
),
)
output, err := pipe.Output(p)
c.Assert(err, IsNil)
c.Assert(string(output), Equals, "hello world!")
data, err := ioutil.ReadFile(path)
c.Assert(err, IsNil)
c.Assert(string(data), Equals, "hello world!")
stat, err := os.Stat(path)
c.Assert(err, IsNil)
c.Assert(stat.Mode()&os.ModePerm, Equals, os.FileMode(0600))
}
