package packer import ( "bytes" "io" "strings" "testing" "time" ) type TestCommunicator struct { Stderr io.Reader Stdout io.Reader } func (c *TestCommunicator) Start(rc *RemoteCmd) error { go func() { if rc.Stdout != nil && c.Stdout != nil { io.Copy(rc.Stdout, c.Stdout) } if rc.Stderr != nil && c.Stderr != nil { io.Copy(rc.Stderr, c.Stderr) } }() return nil } func (c *TestCommunicator) Upload(string, io.Reader) error { return nil } func (c *TestCommunicator) Download(string, io.Writer) error { return nil } func TestRemoteCmd_StartWithUi(t *testing.T) { data := "hello\nworld\nthere" originalOutput := new(bytes.Buffer) rcOutput := new(bytes.Buffer) uiOutput := new(bytes.Buffer) rcOutput.WriteString(data) testComm := &TestCommunicator{ Stdout: rcOutput, } testUi := &ReaderWriterUi{ Reader: new(bytes.Buffer), Writer: uiOutput, } rc := &RemoteCmd{ Command: "test", Stdout: originalOutput, } go func() { time.Sleep(100 * time.Millisecond) rc.SetExited(0) }() err := rc.StartWithUi(testComm, testUi) if err != nil { t.Fatalf("err: %s", err) } if uiOutput.String() != strings.TrimSpace(data)+"\n" { t.Fatalf("bad output: '%s'", uiOutput.String()) } if originalOutput.String() != data { t.Fatalf("original is bad: '%s'", originalOutput.String()) } } func TestRemoteCmd_Wait(t *testing.T) { var cmd RemoteCmd result := make(chan bool) go func() { cmd.Wait() result <- true }() cmd.SetExited(42) select { case <-result: // Success case <-time.After(500 * time.Millisecond): t.Fatal("never got exit notification") } }