private void compareDatasetInfo(DatasetInfo expected, DatasetInfo value) { assertEquals(expected, value); assertEquals(expected.getDatasetId(), value.getDatasetId()); assertEquals(expected.getDescription(), value.getDescription()); assertEquals(expected.getEtag(), value.getEtag()); assertEquals(expected.getFriendlyName(), value.getFriendlyName()); assertEquals(expected.getGeneratedId(), value.getGeneratedId()); assertEquals(expected.getLocation(), value.getLocation()); assertEquals(expected.getSelfLink(), value.getSelfLink()); assertEquals(expected.getAcl(), value.getAcl()); assertEquals(expected.getCreationTime(), value.getCreationTime()); assertEquals(expected.getDefaultTableLifetime(), value.getDefaultTableLifetime()); assertEquals(expected.getLastModified(), value.getLastModified()); } }
private void compareDatasets(DatasetInfo expected, DatasetInfo value) { assertEquals(expected, value); assertEquals(expected.getDatasetId(), value.getDatasetId()); assertEquals(expected.getDescription(), value.getDescription()); assertEquals(expected.getEtag(), value.getEtag()); assertEquals(expected.getFriendlyName(), value.getFriendlyName()); assertEquals(expected.getGeneratedId(), value.getGeneratedId()); assertEquals(expected.getLocation(), value.getLocation()); assertEquals(expected.getSelfLink(), value.getSelfLink()); assertEquals(expected.getAcl(), value.getAcl()); assertEquals(expected.getCreationTime(), value.getCreationTime()); assertEquals(expected.getDefaultTableLifetime(), value.getDefaultTableLifetime()); assertEquals(expected.getLastModified(), value.getLastModified()); assertEquals(expected.getLabels(), value.getLabels()); } }
@Test public void testOf() { DatasetInfo datasetInfo = DatasetInfo.of(DATASET_ID.getDataset()); assertEquals(DATASET_ID, datasetInfo.getDatasetId()); assertNull(datasetInfo.getAcl()); assertNull(datasetInfo.getCreationTime()); assertNull(datasetInfo.getDefaultTableLifetime()); assertNull(datasetInfo.getDescription()); assertNull(datasetInfo.getEtag()); assertNull(datasetInfo.getFriendlyName()); assertNull(datasetInfo.getGeneratedId()); assertNull(datasetInfo.getLastModified()); assertNull(datasetInfo.getLocation()); assertNull(datasetInfo.getSelfLink()); assertTrue(datasetInfo.getLabels().isEmpty()); datasetInfo = DatasetInfo.of(DATASET_ID); assertEquals(DATASET_ID, datasetInfo.getDatasetId()); assertNull(datasetInfo.getAcl()); assertNull(datasetInfo.getCreationTime()); assertNull(datasetInfo.getDefaultTableLifetime()); assertNull(datasetInfo.getDescription()); assertNull(datasetInfo.getEtag()); assertNull(datasetInfo.getFriendlyName()); assertNull(datasetInfo.getGeneratedId()); assertNull(datasetInfo.getLastModified()); assertNull(datasetInfo.getLocation()); assertNull(datasetInfo.getSelfLink()); assertTrue(datasetInfo.getLabels().isEmpty()); }
@Test public void testBuilder() { assertNull(DATASET_INFO.getDatasetId().getProject()); assertEquals(DATASET_ID, DATASET_INFO.getDatasetId()); assertEquals(ACCESS_RULES, DATASET_INFO.getAcl()); assertEquals(CREATION_TIME, DATASET_INFO.getCreationTime()); assertEquals(DEFAULT_TABLE_EXPIRATION, DATASET_INFO.getDefaultTableLifetime()); assertEquals(DESCRIPTION, DATASET_INFO.getDescription()); assertEquals(ETAG, DATASET_INFO.getEtag()); assertEquals(FRIENDLY_NAME, DATASET_INFO.getFriendlyName()); assertEquals(GENERATED_ID, DATASET_INFO.getGeneratedId()); assertEquals(LAST_MODIFIED, DATASET_INFO.getLastModified()); assertEquals(LOCATION, DATASET_INFO.getLocation()); assertEquals(SELF_LINK, DATASET_INFO.getSelfLink()); assertEquals(DATASET_ID_COMPLETE, DATASET_INFO_COMPLETE.getDatasetId()); assertEquals(ACCESS_RULES_COMPLETE, DATASET_INFO_COMPLETE.getAcl()); assertEquals(CREATION_TIME, DATASET_INFO_COMPLETE.getCreationTime()); assertEquals(DEFAULT_TABLE_EXPIRATION, DATASET_INFO_COMPLETE.getDefaultTableLifetime()); assertEquals(DESCRIPTION, DATASET_INFO_COMPLETE.getDescription()); assertEquals(ETAG, DATASET_INFO_COMPLETE.getEtag()); assertEquals(FRIENDLY_NAME, DATASET_INFO_COMPLETE.getFriendlyName()); assertEquals(GENERATED_ID, DATASET_INFO_COMPLETE.getGeneratedId()); assertEquals(LAST_MODIFIED, DATASET_INFO_COMPLETE.getLastModified()); assertEquals(LOCATION, DATASET_INFO_COMPLETE.getLocation()); assertEquals(SELF_LINK, DATASET_INFO_COMPLETE.getSelfLink()); assertEquals(LABELS, DATASET_INFO_COMPLETE.getLabels()); }
@Override public StagingTableId provide(EvalContext evalContext) { final String location = getDatasetOrThrow().getLocation(); final TableId stagingTableId = bigQuery().createStagingTableId(tableId, location); final DatasetId stagingDatasetId = DatasetId.of(stagingTableId.getProject(), stagingTableId.getDataset()); if (bigQuery().getDataset(stagingDatasetId) == null) { bigQuery().create(DatasetInfo .newBuilder(stagingDatasetId) .setLocation(location) .setDefaultTableLifetime(Duration.ofDays(1).toMillis()) .build()); LOG.info("created staging dataset: {}", stagingDatasetId); } return StagingTableId.of(this, stagingTableId); }