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))
}
rateLimiter = make(chan struct{}, 1)
})
Describe("Perform", func() {
var stepErr error
JustBeforeEach(func() {
container, err := gardenClient.Create(garden.ContainerSpec{
Handle: handle,
})
Expect(err).NotTo(HaveOccurred())
step = steps.NewDownload(
container,
downloadAction,
cache,
rateLimiter,
fakeStreamer,
logger,
)
stepErr = step.Perform()
})
var tarReader *tar.Reader
It("downloads via the cache with a tar transformer", func() {
Expect(cache.FetchCallCount()).To(Equal(1))
url, cacheKey, transformer, cancelChan := cache.FetchArgsForCall(0)
Expect(url.Host).To(ContainSubstring("mr_jones"))
Expect(cacheKey).To(Equal("the-cache-key"))