func (transformer *Transformer) StepFor(
logStreamer log_streamer.LogStreamer,
action *models.Action,
container garden.Container,
externalIP string,
ports []executor.PortMapping,
logger lager.Logger,
) steps.Step {
a := action.GetValue()
switch actionModel := a.(type) {
case *models.RunAction:
return steps.NewRun(
container,
*actionModel,
logStreamer.WithSource(actionModel.LogSource),
logger,
externalIP,
ports,
transformer.exportNetworkEnvVars,
transformer.clock,
)
case *models.DownloadAction:
return steps.NewDownload(
container,
*actionModel,
transformer.cachedDownloader,
transformer.downloadLimiter,
logStreamer.WithSource(actionModel.LogSource),
logger,
)
case *models.UploadAction:
return steps.NewUpload(
container,
*actionModel,
transformer.uploader,
transformer.compressor,
transformer.tempDir,
logStreamer.WithSource(actionModel.LogSource),
transformer.uploadLimiter,
logger,
)
case *models.EmitProgressAction:
return steps.NewEmitProgress(
transformer.StepFor(
logStreamer,
actionModel.Action,
container,
externalIP,
ports,
logger,
),
actionModel.StartMessage,
actionModel.SuccessMessage,
actionModel.FailureMessagePrefix,
logStreamer.WithSource(actionModel.LogSource),
logger,
)
case *models.TimeoutAction:
return steps.NewTimeout(
transformer.StepFor(
logStreamer.WithSource(actionModel.LogSource),
actionModel.Action,
container,
externalIP,
ports,
logger,
),
time.Duration(actionModel.Timeout),
logger,
)
case *models.TryAction:
return steps.NewTry(
transformer.StepFor(
logStreamer.WithSource(actionModel.LogSource),
actionModel.Action,
container,
externalIP,
ports,
logger,
),
logger,
)
case *models.ParallelAction:
subSteps := make([]steps.Step, len(actionModel.Actions))
for i, action := range actionModel.Actions {
subSteps[i] = transformer.StepFor(
logStreamer.WithSource(actionModel.LogSource),
action,
container,
externalIP,
ports,
logger,
)
}
return steps.NewParallel(subSteps)
case *models.CodependentAction:
subSteps := make([]steps.Step, len(actionModel.Actions))
for i, action := range actionModel.Actions {
subSteps[i] = transformer.StepFor(
logStreamer.WithSource(actionModel.LogSource),
action,
container,
externalIP,
ports,
logger,
)
}
errorOnExit := true
return steps.NewCodependent(subSteps, errorOnExit)
case *models.SerialAction:
subSteps := make([]steps.Step, len(actionModel.Actions))
for i, action := range actionModel.Actions {
subSteps[i] = transformer.StepFor(
logStreamer,
action,
container,
externalIP,
ports,
logger,
)
}
return steps.NewSerial(subSteps)
}
panic(fmt.Sprintf("unknown action: %T", action))
}
uploadTarget.Close()
})
handle := "some-container-handle"
JustBeforeEach(func() {
gardenClient.Connection.CreateReturns(handle, nil)
container, err := gardenClient.Create(garden.ContainerSpec{})
Expect(err).NotTo(HaveOccurred())
step = steps.NewUpload(
container,
*uploadAction,
uploader,
compressor,
tempDir,
fakeStreamer,
make(chan struct{}, 1),
logger,
)
})
Describe("Perform", func() {
Context("when streaming out works", func() {
var buffer *gbytes.Buffer
BeforeEach(func() {
buffer = gbytes.NewBuffer()
gardenClient.Connection.StreamOutStub = func(handle string, spec garden.StreamOutSpec) (io.ReadCloser, error) {
Expect(spec.Path).To(Equal("./expected-src.txt"))