return fakeRunner
case otherPipelineDB:
return otherFakeRunner
default:
panic("unexpected pipelineDB input received")
return nil
}
return fakeRunner
}
pipelineDBFactory.BuildStub = func(pipeline db.SavedPipeline) db.PipelineDB {
switch pipeline.Name {
case "pipeline":
return pipelineDB
case "other-pipeline":
return otherPipelineDB
default:
panic("unexpected pipeline input received")
return nil
}
}
fakeRunnerExitChan = make(chan error, 1)
// avoid data race
exitChan := fakeRunnerExitChan
fakeRunner.RunStub = func(signals <-chan os.Signal, ready chan<- struct{}) error {
close(ready)
return <-exitChan
}
savedVersions := savedVersionsForEachResource[resourceName]
if len(savedVersions) == 0 {
return db.SavedVersionedResource{}, false, nil
}
return savedVersions[len(savedVersions)-1], true, nil
}
fakePipelineDBs[name] = fakePipelineDB
}
fakeBaggageCollectorDB.GetAllPipelinesReturns(savedPipelines, nil)
fakePipelineDBFactory.BuildStub = func(savedPipeline db.SavedPipeline) db.PipelineDB {
return fakePipelineDBs[savedPipeline.Name]
}
fakeVolumes := map[string]*bcfakes.FakeVolume{}
var savedVolumes []db.SavedVolume
for _, volume := range example.volumeData {
savedVolumes = append(savedVolumes, db.SavedVolume{
Volume: volume,
})
fakeVolumes[volume.Handle] = new(bcfakes.FakeVolume)
}
fakeBaggageCollectorDB.GetVolumesReturns(savedVolumes, nil)
fakeBaggageClaimClient.LookupVolumeStub = func(_ lager.Logger, handle string) (baggageclaim.Volume, bool, error) {