@Test
public void testPrepareJob() throws Exception {
PrepareJobResponse response =
stub.prepare(
PrepareJobRequest.newBuilder()
.setPipelineOptions(Struct.getDefaultInstance())
.setPipeline(Pipeline.getDefaultInstance())
.setJobName("myJobName")
.build());
ApiServiceDescriptor stagingEndpoint = response.getArtifactStagingEndpoint();
ArtifactServiceStager stager =
ArtifactServiceStager.overChannel(
InProcessChannelBuilder.forName(stagingEndpoint.getUrl()).build());
String stagingSessionToken = "token";
File foo = writeTempFile("foo", "foo, bar, baz".getBytes(UTF_8));
File bar = writeTempFile("spam", "spam, ham, eggs".getBytes(UTF_8));
stager.stage(
stagingSessionToken,
ImmutableList.of(StagedFile.of(foo, foo.getName()), StagedFile.of(bar, bar.getName())));
List<byte[]> tempDirFiles = readFlattenedFiles(runnerTemp.getRoot());
assertThat(
tempDirFiles,
hasItems(
arrayEquals(Files.readAllBytes(foo.toPath())),
arrayEquals(Files.readAllBytes(bar.toPath()))));
}