@Override ExtractJobConfiguration setProjectId(String projectId) { if (Strings.isNullOrEmpty(getSourceTable().getProject())) { return toBuilder().setSourceTable(getSourceTable().setProjectId(projectId)).build(); } return this; }
@Test public void testSetProjectId() { ExtractJobConfiguration configuration = EXTRACT_CONFIGURATION.setProjectId(TEST_PROJECT_ID); assertEquals(TEST_PROJECT_ID, configuration.getSourceTable().getProject()); }
@Test public void testOf() { ExtractJobConfiguration job = ExtractJobConfiguration.of(TABLE_ID, DESTINATION_URIS); assertEquals(TABLE_ID, job.getSourceTable()); assertEquals(DESTINATION_URIS, job.getDestinationUris()); job = ExtractJobConfiguration.of(TABLE_ID, DESTINATION_URI); assertEquals(TABLE_ID, job.getSourceTable()); assertEquals(ImmutableList.of(DESTINATION_URI), job.getDestinationUris()); job = ExtractJobConfiguration.of(TABLE_ID, DESTINATION_URIS, JSON_FORMAT); assertEquals(TABLE_ID, job.getSourceTable()); assertEquals(DESTINATION_URIS, job.getDestinationUris()); assertEquals(JSON_FORMAT, job.getFormat()); job = ExtractJobConfiguration.of(TABLE_ID, DESTINATION_URI, JSON_FORMAT); assertEquals(TABLE_ID, job.getSourceTable()); assertEquals(ImmutableList.of(DESTINATION_URI), job.getDestinationUris()); assertEquals(JSON_FORMAT, job.getFormat()); }
@Test public void testSetProjectIdDoNotOverride() { ExtractJobConfiguration configuration = EXTRACT_CONFIGURATION .toBuilder() .setSourceTable(TABLE_ID.setProjectId(TEST_PROJECT_ID)) .build() .setProjectId("do-not-update"); assertEquals(TEST_PROJECT_ID, configuration.getSourceTable().getProject()); }
@Test public void testBuilder() { assertEquals(TABLE_ID, EXTRACT_CONFIGURATION.getSourceTable()); assertEquals(DESTINATION_URIS, EXTRACT_CONFIGURATION.getDestinationUris()); assertEquals(FIELD_DELIMITER, EXTRACT_CONFIGURATION.getFieldDelimiter()); assertEquals(COMPRESSION, EXTRACT_CONFIGURATION.getCompression()); assertEquals(PRINT_HEADER, EXTRACT_CONFIGURATION.printHeader()); assertEquals(FORMAT, EXTRACT_CONFIGURATION.getFormat()); assertEquals(TABLE_ID, EXTRACT_CONFIGURATION_ONE_URI.getSourceTable()); assertEquals( ImmutableList.of(DESTINATION_URI), EXTRACT_CONFIGURATION_ONE_URI.getDestinationUris()); assertEquals(FIELD_DELIMITER, EXTRACT_CONFIGURATION_ONE_URI.getFieldDelimiter()); assertEquals(COMPRESSION, EXTRACT_CONFIGURATION_ONE_URI.getCompression()); assertEquals(PRINT_HEADER, EXTRACT_CONFIGURATION_ONE_URI.printHeader()); assertEquals(FORMAT, EXTRACT_CONFIGURATION_ONE_URI.getFormat()); }
private void compareExtractJobConfiguration( ExtractJobConfiguration expected, ExtractJobConfiguration value) { assertEquals(expected, value); assertEquals(expected.hashCode(), value.hashCode()); assertEquals(expected.toString(), value.toString()); assertEquals(expected.getSourceTable(), value.getSourceTable()); assertEquals(expected.getDestinationUris(), value.getDestinationUris()); assertEquals(expected.getCompression(), value.getCompression()); assertEquals(expected.printHeader(), value.printHeader()); assertEquals(expected.getFieldDelimiter(), value.getFieldDelimiter()); assertEquals(expected.getFormat(), value.getFormat()); } }
@Test public void testToBuilder() { compareExtractJobConfiguration( EXTRACT_CONFIGURATION, EXTRACT_CONFIGURATION.toBuilder().build()); ExtractJobConfiguration job = EXTRACT_CONFIGURATION.toBuilder().setSourceTable(TableId.of("dataset", "newTable")).build(); assertEquals("newTable", job.getSourceTable().getTable()); job = job.toBuilder().setSourceTable(TABLE_ID).build(); compareExtractJobConfiguration(EXTRACT_CONFIGURATION, job); }
@Test public void testSetProjectId() { JobInfo jobInfo = COPY_JOB.setProjectId("p"); assertEquals("p", jobInfo.getJobId().getProject()); CopyJobConfiguration copyConfiguration = jobInfo.getConfiguration(); assertEquals("p", copyConfiguration.getDestinationTable().getProject()); for (TableId sourceTable : copyConfiguration.getSourceTables()) { assertEquals("p", sourceTable.getProject()); } jobInfo = EXTRACT_JOB.setProjectId("p"); assertEquals("p", jobInfo.getJobId().getProject()); ExtractJobConfiguration extractConfiguration = jobInfo.getConfiguration(); assertEquals("p", extractConfiguration.getSourceTable().getProject()); jobInfo = LOAD_JOB.setProjectId("p"); assertEquals("p", jobInfo.getJobId().getProject()); LoadJobConfiguration loadConfiguration = jobInfo.getConfiguration(); assertEquals("p", loadConfiguration.getDestinationTable().getProject()); jobInfo = QUERY_JOB.setProjectId("p"); assertEquals("p", jobInfo.getJobId().getProject()); QueryJobConfiguration queryConfiguration = jobInfo.getConfiguration(); assertEquals("p", queryConfiguration.getDefaultDataset().getProject()); assertEquals("p", queryConfiguration.getDestinationTable().getProject()); }
@Override ExtractJobConfiguration setProjectId(String projectId) { if (Strings.isNullOrEmpty(getSourceTable().getProject())) { return toBuilder().setSourceTable(getSourceTable().setProjectId(projectId)).build(); } return this; }