@Override public Tuple<String, Iterable<com.google.api.services.bigquery.model.Dataset>> call() { return serviceOptions.getBigQueryRpcV2().listDatasets(projectId, optionsMap); } },
@Test public void testListEmptyDatasets() { ImmutableList<com.google.api.services.bigquery.model.Dataset> datasets = ImmutableList.of(); Tuple<String, Iterable<com.google.api.services.bigquery.model.Dataset>> result = Tuple.<String, Iterable<com.google.api.services.bigquery.model.Dataset>>of(null, datasets); EasyMock.expect(bigqueryRpcMock.listDatasets(PROJECT, EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(bigqueryRpcMock); bigquery = options.getService(); Page<Dataset> page = bigquery.listDatasets(); assertNull(page.getNextPageToken()); assertArrayEquals( ImmutableList.of().toArray(), Iterables.toArray(page.getValues(), Dataset.class)); }
@Test public void testListDatasetsWithProjects() { bigquery = options.getService(); ImmutableList<Dataset> datasetList = ImmutableList.of( new Dataset( bigquery, new DatasetInfo.BuilderImpl(DATASET_INFO.setProjectId(OTHER_PROJECT)))); Tuple<String, Iterable<com.google.api.services.bigquery.model.Dataset>> result = Tuple.of(CURSOR, Iterables.transform(datasetList, DatasetInfo.TO_PB_FUNCTION)); EasyMock.expect(bigqueryRpcMock.listDatasets(OTHER_PROJECT, EMPTY_RPC_OPTIONS)) .andReturn(result); EasyMock.replay(bigqueryRpcMock); Page<Dataset> page = bigquery.listDatasets(OTHER_PROJECT); assertEquals(CURSOR, page.getNextPageToken()); assertArrayEquals( datasetList.toArray(), Iterables.toArray(page.getValues(), DatasetInfo.class)); }
@Test public void testListDatasetsWithOptions() { bigquery = options.getService(); ImmutableList<Dataset> datasetList = ImmutableList.of( new Dataset(bigquery, new DatasetInfo.BuilderImpl(DATASET_INFO_WITH_PROJECT)), new Dataset(bigquery, new DatasetInfo.BuilderImpl(OTHER_DATASET_INFO))); Tuple<String, Iterable<com.google.api.services.bigquery.model.Dataset>> result = Tuple.of(CURSOR, Iterables.transform(datasetList, DatasetInfo.TO_PB_FUNCTION)); EasyMock.expect(bigqueryRpcMock.listDatasets(PROJECT, DATASET_LIST_OPTIONS)).andReturn(result); EasyMock.replay(bigqueryRpcMock); Page<Dataset> page = bigquery.listDatasets(DATASET_LIST_ALL, DATASET_LIST_PAGE_TOKEN, DATASET_LIST_PAGE_SIZE); assertEquals(CURSOR, page.getNextPageToken()); assertArrayEquals( datasetList.toArray(), Iterables.toArray(page.getValues(), DatasetInfo.class)); }
@Test public void testListDatasets() { bigquery = options.getService(); ImmutableList<Dataset> datasetList = ImmutableList.of( new Dataset(bigquery, new DatasetInfo.BuilderImpl(DATASET_INFO_WITH_PROJECT)), new Dataset(bigquery, new DatasetInfo.BuilderImpl(OTHER_DATASET_INFO))); Tuple<String, Iterable<com.google.api.services.bigquery.model.Dataset>> result = Tuple.of(CURSOR, Iterables.transform(datasetList, DatasetInfo.TO_PB_FUNCTION)); EasyMock.expect(bigqueryRpcMock.listDatasets(PROJECT, EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(bigqueryRpcMock); Page<Dataset> page = bigquery.listDatasets(); assertEquals(CURSOR, page.getNextPageToken()); assertArrayEquals( datasetList.toArray(), Iterables.toArray(page.getValues(), DatasetInfo.class)); }
@Override public Tuple<String, Iterable<com.google.api.services.bigquery.model.Dataset>> call() { return serviceOptions.getBigQueryRpcV2().listDatasets(projectId, optionsMap); } },