@Override public Tuple<String, Iterable<com.google.api.services.bigquery.model.Job>> call() { return serviceOptions .getBigQueryRpcV2() .listJobs(serviceOptions.getProjectId(), optionsMap); } },
@Test public void testListJobs() { bigquery = options.getService(); ImmutableList<Job> jobList = ImmutableList.of( new Job(bigquery, new JobInfo.BuilderImpl(COMPLETE_QUERY_JOB)), new Job(bigquery, new JobInfo.BuilderImpl(COMPLETE_LOAD_JOB))); Tuple<String, Iterable<com.google.api.services.bigquery.model.Job>> result = Tuple.of( CURSOR, Iterables.transform( jobList, new Function<Job, com.google.api.services.bigquery.model.Job>() { @Override public com.google.api.services.bigquery.model.Job apply(Job job) { return job.toPb(); } })); EasyMock.expect(bigqueryRpcMock.listJobs(PROJECT, EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(bigqueryRpcMock); Page<Job> page = bigquery.listJobs(); assertEquals(CURSOR, page.getNextPageToken()); assertArrayEquals(jobList.toArray(), Iterables.toArray(page.getValues(), Job.class)); }
@Test public void testListJobsWithOptions() { bigquery = options.getService(); ImmutableList<Job> jobList = ImmutableList.of( new Job(bigquery, new JobInfo.BuilderImpl(COMPLETE_QUERY_JOB)), new Job(bigquery, new JobInfo.BuilderImpl(COMPLETE_LOAD_JOB))); Tuple<String, Iterable<com.google.api.services.bigquery.model.Job>> result = Tuple.of( CURSOR, Iterables.transform( jobList, new Function<Job, com.google.api.services.bigquery.model.Job>() { @Override public com.google.api.services.bigquery.model.Job apply(Job job) { return job.toPb(); } })); EasyMock.expect(bigqueryRpcMock.listJobs(PROJECT, JOB_LIST_OPTIONS)).andReturn(result); EasyMock.replay(bigqueryRpcMock); Page<Job> page = bigquery.listJobs( JOB_LIST_ALL_USERS, JOB_LIST_STATE_FILTER, JOB_LIST_PAGE_TOKEN, JOB_LIST_PAGE_SIZE); assertEquals(CURSOR, page.getNextPageToken()); assertArrayEquals(jobList.toArray(), Iterables.toArray(page.getValues(), Job.class)); }
EasyMock.expect(bigqueryRpcMock.listJobs(eq(PROJECT), capture(capturedOptions))) .andReturn(result); EasyMock.replay(bigqueryRpcMock);
@Override public Tuple<String, Iterable<com.google.api.services.bigquery.model.Job>> call() { return serviceOptions .getBigQueryRpcV2() .listJobs(serviceOptions.getProjectId(), optionsMap); } },