/** Example of updating a dataset. */ // [TARGET update(DatasetOption...)] // [VARIABLE "my_friendly_name"] public Dataset updateDataset(String friendlyName) { // [START ] Builder builder = dataset.toBuilder(); builder.setFriendlyName(friendlyName); Dataset updatedDataset = builder.build().update(); // [END ] return updatedDataset; }
@Test public void testUpdate() { initializeExpectedDataset(4); Dataset expectedUpdatedDataset = expectedDataset.toBuilder().setDescription("Description").build(); expect(bigquery.getOptions()).andReturn(mockOptions); expect(bigquery.update(eq(expectedDataset))).andReturn(expectedUpdatedDataset); replay(bigquery); initializeDataset(); Dataset actualUpdatedDataset = dataset.update(); compareDataset(expectedUpdatedDataset, actualUpdatedDataset); }
@Test public void testUpdateWithOptions() { initializeExpectedDataset(4); Dataset expectedUpdatedDataset = expectedDataset.toBuilder().setDescription("Description").build(); expect(bigquery.getOptions()).andReturn(mockOptions); expect(bigquery.update(eq(expectedDataset), eq(BigQuery.DatasetOption.fields()))) .andReturn(expectedUpdatedDataset); replay(bigquery); initializeDataset(); Dataset actualUpdatedDataset = dataset.update(BigQuery.DatasetOption.fields()); compareDataset(expectedUpdatedDataset, actualUpdatedDataset); }