132 lines
2.7 KiB
Go
132 lines
2.7 KiB
Go
package packer
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"errors"
|
|
"io"
|
|
"os"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
// MockCommunicator is a valid Communicator implementation that can be
|
|
// used for tests.
|
|
type MockCommunicator struct {
|
|
StartCalled bool
|
|
StartCmd *RemoteCmd
|
|
StartStderr string
|
|
StartStdout string
|
|
StartStdin string
|
|
StartExitStatus int
|
|
|
|
UploadCalled bool
|
|
UploadPath string
|
|
UploadData string
|
|
|
|
UploadDirDst string
|
|
UploadDirSrc string
|
|
UploadDirExclude []string
|
|
|
|
DownloadDirDst string
|
|
DownloadDirSrc string
|
|
DownloadDirExclude []string
|
|
|
|
DownloadCalled bool
|
|
DownloadPath string
|
|
DownloadData string
|
|
}
|
|
|
|
func (c *MockCommunicator) Start(ctx context.Context, rc *RemoteCmd) error {
|
|
c.StartCalled = true
|
|
c.StartCmd = rc
|
|
|
|
go func() {
|
|
var wg sync.WaitGroup
|
|
if rc.Stdout != nil && c.StartStdout != "" {
|
|
wg.Add(1)
|
|
go func() {
|
|
io.Copy(rc.Stdout, strings.NewReader(c.StartStdout))
|
|
wg.Done()
|
|
}()
|
|
}
|
|
|
|
if rc.Stderr != nil && c.StartStderr != "" {
|
|
wg.Add(1)
|
|
go func() {
|
|
io.Copy(rc.Stderr, strings.NewReader(c.StartStderr))
|
|
wg.Done()
|
|
}()
|
|
}
|
|
|
|
if rc.Stdin != nil {
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
var data bytes.Buffer
|
|
io.Copy(&data, rc.Stdin)
|
|
c.StartStdin = data.String()
|
|
}()
|
|
}
|
|
|
|
wg.Wait()
|
|
rc.SetExited(c.StartExitStatus)
|
|
}()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *MockCommunicator) Upload(path string, r io.Reader, fi *os.FileInfo) error {
|
|
c.UploadCalled = true
|
|
c.UploadPath = path
|
|
|
|
var data bytes.Buffer
|
|
if _, err := io.Copy(&data, r); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
c.UploadData = data.String()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *MockCommunicator) UploadDir(dst string, src string, excl []string) error {
|
|
c.UploadDirDst = dst
|
|
c.UploadDirSrc = src
|
|
c.UploadDirExclude = excl
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *MockCommunicator) Download(path string, w io.Writer) error {
|
|
c.DownloadCalled = true
|
|
c.DownloadPath = path
|
|
w.Write([]byte(c.DownloadData))
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *MockCommunicator) DownloadDir(src string, dst string, excl []string) error {
|
|
c.DownloadDirDst = dst
|
|
c.DownloadDirSrc = src
|
|
c.DownloadDirExclude = excl
|
|
|
|
return nil
|
|
}
|
|
|
|
// ScriptUploadErrorMockCommunicator returns an error from it's Upload() method
|
|
// when a script is uploaded to test the case where this upload fails.
|
|
type ScriptUploadErrorMockCommunicator struct {
|
|
MockCommunicator
|
|
}
|
|
|
|
var ScriptUploadErrorMockCommunicatorError = errors.New("ScriptUploadErrorMockCommunicator Upload error")
|
|
|
|
func (c *ScriptUploadErrorMockCommunicator) Upload(path string, r io.Reader, fi *os.FileInfo) error {
|
|
// only fail on script uploads, not on environment variable uploads
|
|
if !strings.Contains(path, "packer-ps-env-vars") {
|
|
return ScriptUploadErrorMockCommunicatorError
|
|
}
|
|
return c.MockCommunicator.Upload(path, r, fi)
|
|
}
|