@Test public void testDeleteTrue() { initializeExpectedDataset(1); expect(bigquery.getOptions()).andReturn(mockOptions); expect(bigquery.delete(DATASET_INFO.getDatasetId())).andReturn(true); replay(bigquery); initializeDataset(); assertTrue(dataset.delete()); }
@Test public void testDeleteFalse() { initializeExpectedDataset(1); expect(bigquery.getOptions()).andReturn(mockOptions); expect(bigquery.delete(DATASET_INFO.getDatasetId())).andReturn(false); replay(bigquery); initializeDataset(); assertFalse(dataset.delete()); }
@Test public void testExists_False() throws Exception { initializeExpectedDataset(1); BigQuery.DatasetOption[] expectedOptions = {BigQuery.DatasetOption.fields()}; expect(bigquery.getOptions()).andReturn(mockOptions); expect(bigquery.getDataset(DATASET_INFO.getDatasetId(), expectedOptions)).andReturn(null); replay(bigquery); initializeDataset(); assertFalse(dataset.exists()); }
@Test public void testExists_True() throws Exception { initializeExpectedDataset(1); BigQuery.DatasetOption[] expectedOptions = {BigQuery.DatasetOption.fields()}; expect(bigquery.getOptions()).andReturn(mockOptions); expect(bigquery.getDataset(DATASET_INFO.getDatasetId(), expectedOptions)) .andReturn(expectedDataset); replay(bigquery); initializeDataset(); assertTrue(dataset.exists()); }
@Test public void testReloadNull() throws Exception { initializeExpectedDataset(1); expect(bigquery.getOptions()).andReturn(mockOptions); expect(bigquery.getDataset(DATASET_INFO.getDatasetId().getDataset())).andReturn(null); replay(bigquery); initializeDataset(); assertNull(dataset.reload()); }
DatasetInfo setProjectId(String projectId) { Builder builder = toBuilder(); builder.setDatasetId(getDatasetId().setProjectId(projectId)); if (getAcl() != null) { List<Acl> acls = Lists.newArrayListWithCapacity(getAcl().size()); for (Acl acl : getAcl()) { if (acl.getEntity().getType() == Acl.Entity.Type.VIEW) { Dataset.Access accessPb = acl.toPb(); TableReference viewReferencePb = accessPb.getView(); if (viewReferencePb.getProjectId() == null) { viewReferencePb.setProjectId(projectId); } acls.add(Acl.of(new Acl.View(TableId.fromPb(viewReferencePb)))); } else { acls.add(acl); } } builder.setAcl(acls); } return builder.build(); }
@Test public void testReload() throws Exception { initializeExpectedDataset(4); DatasetInfo updatedInfo = DATASET_INFO.toBuilder().setDescription("Description").build(); Dataset expectedDataset = new Dataset(serviceMockReturnsOptions, new DatasetInfo.BuilderImpl(updatedInfo)); expect(bigquery.getOptions()).andReturn(mockOptions); expect(bigquery.getDataset(DATASET_INFO.getDatasetId().getDataset())) .andReturn(expectedDataset); replay(bigquery); initializeDataset(); Dataset updatedDataset = dataset.reload(); compareDataset(expectedDataset, updatedDataset); }
@Test public void testList() throws Exception { initializeExpectedDataset(4); List<Table> tableResults = ImmutableList.of( new Table(serviceMockReturnsOptions, new Table.BuilderImpl(TABLE_INFO1)), new Table(serviceMockReturnsOptions, new Table.BuilderImpl(TABLE_INFO2)), new Table(serviceMockReturnsOptions, new Table.BuilderImpl(TABLE_INFO3))); PageImpl<Table> expectedPage = new PageImpl<>(null, "c", tableResults); expect(bigquery.getOptions()).andReturn(mockOptions); expect(bigquery.listTables(DATASET_INFO.getDatasetId())).andReturn(expectedPage); replay(bigquery); initializeDataset(); Page<Table> tablePage = dataset.list(); assertArrayEquals( tableResults.toArray(), Iterables.toArray(tablePage.getValues(), Table.class)); assertEquals(expectedPage.getNextPageToken(), tablePage.getNextPageToken()); }
@Test public void testReloadWithOptions() throws Exception { initializeExpectedDataset(4); DatasetInfo updatedInfo = DATASET_INFO.toBuilder().setDescription("Description").build(); Dataset expectedDataset = new Dataset(serviceMockReturnsOptions, new DatasetInfo.BuilderImpl(updatedInfo)); expect(bigquery.getOptions()).andReturn(mockOptions); expect( bigquery.getDataset( DATASET_INFO.getDatasetId().getDataset(), BigQuery.DatasetOption.fields())) .andReturn(expectedDataset); replay(bigquery); initializeDataset(); Dataset updatedDataset = dataset.reload(BigQuery.DatasetOption.fields()); compareDataset(expectedDataset, updatedDataset); }
@Test public void testListWithOptions() throws Exception { initializeExpectedDataset(4); List<Table> tableResults = ImmutableList.of( new Table(serviceMockReturnsOptions, new Table.BuilderImpl(TABLE_INFO1)), new Table(serviceMockReturnsOptions, new Table.BuilderImpl(TABLE_INFO2)), new Table(serviceMockReturnsOptions, new Table.BuilderImpl(TABLE_INFO3))); PageImpl<Table> expectedPage = new PageImpl<>(null, "c", tableResults); expect(bigquery.getOptions()).andReturn(mockOptions); expect(bigquery.listTables(DATASET_INFO.getDatasetId(), BigQuery.TableListOption.pageSize(10L))) .andReturn(expectedPage); replay(bigquery); initializeDataset(); Page<Table> tablePage = dataset.list(BigQuery.TableListOption.pageSize(10L)); assertArrayEquals( tableResults.toArray(), Iterables.toArray(tablePage.getValues(), Table.class)); assertEquals(expectedPage.getNextPageToken(), tablePage.getNextPageToken()); }
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()); } }
@Test public void testToBuilder() { compareDatasets(DATASET_INFO, DATASET_INFO.toBuilder().build()); DatasetInfo datasetInfo = DATASET_INFO .toBuilder() .setDatasetId(DatasetId.of("dataset2")) .setDescription("description2") .build(); assertEquals(DatasetId.of("dataset2"), datasetInfo.getDatasetId()); assertEquals("description2", datasetInfo.getDescription()); datasetInfo = datasetInfo.toBuilder().setDatasetId(DATASET_ID).setDescription("description").build(); compareDatasets(DATASET_INFO, datasetInfo); }
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 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()); }
@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()); }
public void deleteDataset(BigQuery bigQuery, File datasetInfo) throws IOException { try (FileReader reader = new FileReader(datasetInfo)) { DatasetInfo info = readDatasetInfo(reader); if (bigQuery == null) { bigQuery = bigQuery(); } Dataset dataset = bigQuery.getDataset(info.getDatasetId()); forceDelete(dataset); } }
DatasetInfo setProjectId(String projectId) { Builder builder = toBuilder(); builder.setDatasetId(getDatasetId().setProjectId(projectId)); if (getAcl() != null) { List<Acl> acls = Lists.newArrayListWithCapacity(getAcl().size()); for (Acl acl : getAcl()) { if (acl.getEntity().getType() == Acl.Entity.Type.VIEW) { Dataset.Access accessPb = acl.toPb(); TableReference viewReferencePb = accessPb.getView(); if (viewReferencePb.getProjectId() == null) { viewReferencePb.setProjectId(projectId); } acls.add(Acl.of(new Acl.View(TableId.fromPb(viewReferencePb)))); } else { acls.add(acl); } } builder.setAcl(acls); } return builder.build(); }