attributes.put(BigQueryAttributes.JOB_END_TIME_ATTR, Long.toString(job.getStatistics().getEndTime())); attributes.put(BigQueryAttributes.JOB_START_TIME_ATTR, Long.toString(job.getStatistics().getStartTime())); attributes.put(BigQueryAttributes.JOB_LINK_ATTR, job.getSelfLink()); session.transfer(flowFile, REL_FAILURE); } else { session.getProvenanceReporter().send(flowFile, job.getSelfLink(), job.getStatistics().getEndTime() - job.getStatistics().getStartTime()); session.transfer(flowFile, REL_SUCCESS);
@Test public void testBuilder() { initializeExpectedJob(2); replay(bigquery); Job builtJob = new Job.Builder(serviceMockReturnsOptions, COPY_CONFIGURATION) .setJobId(JOB_ID) .setStatistics(COPY_JOB_STATISTICS) .setJobId(JOB_ID) .setEtag(ETAG) .setGeneratedId(GENERATED_ID) .setSelfLink(SELF_LINK) .setUserEmail(EMAIL) .setStatus(JOB_STATUS) .build(); assertEquals(ETAG, builtJob.getEtag()); assertEquals(GENERATED_ID, builtJob.getGeneratedId()); assertEquals(SELF_LINK, builtJob.getSelfLink()); assertEquals(EMAIL, builtJob.getUserEmail()); assertEquals(JOB_ID, builtJob.getJobId()); assertEquals(JOB_STATUS, builtJob.getStatus()); assertEquals(COPY_CONFIGURATION, builtJob.getConfiguration()); assertEquals(COPY_JOB_STATISTICS, builtJob.getStatistics()); assertSame(serviceMockReturnsOptions, builtJob.getBigQuery()); }
assertNotNull(remoteJob.getStatistics()); assertNotNull(remoteJob.getStatus()); assertEquals(createdJob.getSelfLink(), remoteJob.getSelfLink()); assertEquals(createdJob.getUserEmail(), remoteJob.getUserEmail()); assertTrue(createdTable.delete());
assertNull(createdJob.getStatistics()); assertNull(createdJob.getStatus()); assertNull(createdJob.getSelfLink()); assertNull(createdJob.getUserEmail()); Job remoteJob = bigquery.getJob(createdJob.getJobId(), JobOption.fields(JobField.ETAG)); assertNull(remoteJob.getStatistics()); assertNull(remoteJob.getStatus()); assertNull(remoteJob.getSelfLink()); assertNull(remoteJob.getUserEmail()); assertTrue(createdTable.delete());