Search in sources :

Example 1 with ComputeEngine

use of com.hartwig.pipeline.execution.vm.ComputeEngine in project pipeline5 by hartwigmedical.

the class StageRunnerTest method setUp.

@Before
public void setUp() throws Exception {
    storage = mock(Storage.class);
    Bucket runtimeBucket = mock(Bucket.class);
    when(runtimeBucket.getName()).thenReturn("run-reference-tumor-test");
    when(storage.get("run-reference-tumor-test")).thenReturn(runtimeBucket);
    computeEngine = mock(ComputeEngine.class);
    // noinspection unchecked
    stage = mock(Stage.class);
    startingPoint = mock(StartingPoint.class);
    victim = new StageRunner<>(storage, ARGUMENTS, computeEngine, ResultsDirectory.defaultDirectory(), startingPoint, Labels.of(ARGUMENTS), InputMode.TUMOR_REFERENCE);
    when(stage.namespace()).thenReturn(NAMESPACE);
    output = mock(StageOutput.class);
    when(stage.output(eq(METADATA), any(), any(), any())).thenReturn(output);
}
Also used : StageOutput(com.hartwig.pipeline.StageOutput) Storage(com.google.cloud.storage.Storage) Bucket(com.google.cloud.storage.Bucket) ComputeEngine(com.hartwig.pipeline.execution.vm.ComputeEngine) StartingPoint(com.hartwig.pipeline.reruns.StartingPoint) Before(org.junit.Before)

Aggregations

Bucket (com.google.cloud.storage.Bucket)1 Storage (com.google.cloud.storage.Storage)1 StageOutput (com.hartwig.pipeline.StageOutput)1 ComputeEngine (com.hartwig.pipeline.execution.vm.ComputeEngine)1 StartingPoint (com.hartwig.pipeline.reruns.StartingPoint)1 Before (org.junit.Before)1