@Test
public void testGetDataset() {
Dataset dataset = bigquery.getDataset(DATASET);
assertEquals(bigquery.options().projectId(), dataset.datasetId().project());
assertEquals(DATASET, dataset.datasetId().dataset());
assertEquals(DESCRIPTION, dataset.description());
assertNotNull(dataset.acl());
assertNotNull(dataset.etag());
assertNotNull(dataset.id());
assertNotNull(dataset.lastModified());
assertNotNull(dataset.selfLink());
}
@Test
public void testUpdateDataset() {
Dataset dataset =
bigquery.create(DatasetInfo.builder(OTHER_DATASET).description("Some Description").build());
assertNotNull(dataset);
assertEquals(bigquery.options().projectId(), dataset.datasetId().project());
assertEquals(OTHER_DATASET, dataset.datasetId().dataset());
assertEquals("Some Description", dataset.description());
Dataset updatedDataset =
bigquery.update(dataset.toBuilder().description("Updated Description").build());
assertEquals("Updated Description", updatedDataset.description());
assertTrue(dataset.delete());
}
@Test
public void testGetDatasetWithSelectedFields() {
Dataset dataset =
bigquery.getDataset(DATASET, DatasetOption.fields(DatasetField.CREATION_TIME));
assertEquals(bigquery.options().projectId(), dataset.datasetId().project());
assertEquals(DATASET, dataset.datasetId().dataset());
assertNotNull(dataset.creationTime());
assertNull(dataset.description());
assertNull(dataset.defaultTableLifetime());
assertNull(dataset.acl());
assertNull(dataset.etag());
assertNull(dataset.friendlyName());
assertNull(dataset.id());
assertNull(dataset.lastModified());
assertNull(dataset.location());
assertNull(dataset.selfLink());
}
@Test
public void testUpdateDatasetWithSelectedFields() {
Dataset dataset =
bigquery.create(DatasetInfo.builder(OTHER_DATASET).description("Some Description").build());
assertNotNull(dataset);
assertEquals(bigquery.options().projectId(), dataset.datasetId().project());
assertEquals(OTHER_DATASET, dataset.datasetId().dataset());
assertEquals("Some Description", dataset.description());
Dataset updatedDataset =
bigquery.update(
dataset.toBuilder().description("Updated Description").build(),
DatasetOption.fields(DatasetField.DESCRIPTION));
assertEquals("Updated Description", updatedDataset.description());
assertNull(updatedDataset.creationTime());
assertNull(updatedDataset.defaultTableLifetime());
assertNull(updatedDataset.acl());
assertNull(updatedDataset.etag());
assertNull(updatedDataset.friendlyName());
assertNull(updatedDataset.id());
assertNull(updatedDataset.lastModified());
assertNull(updatedDataset.location());
assertNull(updatedDataset.selfLink());
assertTrue(dataset.delete());
}