JobInfo setProjectId(String projectId) { Builder builder = toBuilder(); if (jobId != null) { builder.setJobId(jobId.setProjectId(projectId)); } return builder.setConfiguration(configuration.setProjectId(projectId)).build(); }
@Test public void testToBuilderIncomplete() { JobInfo job = JobInfo.of(COPY_CONFIGURATION); compareJobInfo(job, job.toBuilder().build()); }
@Test public void testToBuilder() { compareJobInfo(COPY_JOB, COPY_JOB.toBuilder().build()); compareJobInfo(EXTRACT_JOB, EXTRACT_JOB.toBuilder().build()); compareJobInfo(LOAD_JOB, LOAD_JOB.toBuilder().build()); compareJobInfo(QUERY_JOB, QUERY_JOB.toBuilder().build()); JobInfo job = COPY_JOB.toBuilder().setUserEmail("newEmail").build(); assertEquals("newEmail", job.getUserEmail()); job = job.toBuilder().setUserEmail(EMAIL).build(); compareJobInfo(COPY_JOB, job); job = EXTRACT_JOB.toBuilder().setUserEmail("newEmail").build(); assertEquals("newEmail", job.getUserEmail()); job = job.toBuilder().setUserEmail(EMAIL).build(); compareJobInfo(EXTRACT_JOB, job); job = LOAD_JOB.toBuilder().setUserEmail("newEmail").build(); assertEquals("newEmail", job.getUserEmail()); job = job.toBuilder().setUserEmail(EMAIL).build(); compareJobInfo(LOAD_JOB, job); job = QUERY_JOB.toBuilder().setUserEmail("newEmail").build(); assertEquals("newEmail", job.getUserEmail()); job = job.toBuilder().setUserEmail(EMAIL).build(); compareJobInfo(QUERY_JOB, job); }
@Test public void testReload() throws Exception { initializeExpectedJob(4); JobInfo updatedInfo = JOB_INFO.toBuilder().setEtag("etag").build(); Job expectedJob = new Job(serviceMockReturnsOptions, new JobInfo.BuilderImpl(updatedInfo)); expect(bigquery.getOptions()).andReturn(mockOptions); expect(bigquery.getJob(JOB_INFO.getJobId())).andReturn(expectedJob); replay(bigquery); initializeJob(); Job updatedJob = job.reload(); compareJob(expectedJob, updatedJob); }
@Test public void testReloadWithOptions() throws Exception { initializeExpectedJob(4); JobInfo updatedInfo = JOB_INFO.toBuilder().setEtag("etag").build(); Job expectedJob = new Job(serviceMockReturnsOptions, new JobInfo.BuilderImpl(updatedInfo)); expect(bigquery.getOptions()).andReturn(mockOptions); expect(bigquery.getJob(JOB_INFO.getJobId(), BigQuery.JobOption.fields())) .andReturn(expectedJob); replay(bigquery); initializeJob(); Job updatedJob = job.reload(BigQuery.JobOption.fields()); compareJob(expectedJob, updatedJob); }
boolean idRandom = false; if (jobInfo.getJobId() == null) { jobInfo = jobInfo.toBuilder().setJobId(idProvider.get()).build(); idRandom = true;
JobInfo setProjectId(String projectId) { return toBuilder().configuration(configuration.setProjectId(projectId)).build(); }
JobInfo setProjectId(String projectId) { Builder builder = toBuilder(); if (jobId != null) { builder.setJobId(jobId.setProjectId(projectId)); } return builder.setConfiguration(configuration.setProjectId(projectId)).build(); }
boolean idRandom = false; if (jobInfo.getJobId() == null) { jobInfo = jobInfo.toBuilder().setJobId(idProvider.get()).build(); idRandom = true;