Class FakeJobService
- java.lang.Object
-
- org.apache.beam.sdk.io.gcp.testing.FakeJobService
-
- All Implemented Interfaces:
java.io.Serializable
,java.lang.AutoCloseable
,BigQueryServices.JobService
@Internal public class FakeJobService extends java.lang.Object implements BigQueryServices.JobService, java.io.Serializable
A fake implementation of BigQuery's job service.- See Also:
- Serialized Form
-
-
Constructor Summary
Constructors Constructor Description FakeJobService()
FakeJobService(int numFailures)
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description void
close()
com.google.api.services.bigquery.model.JobStatistics
dryRunQuery(java.lang.String projectId, com.google.api.services.bigquery.model.JobConfigurationQuery query, java.lang.String location)
Dry runs the query in the given project.void
expectDryRunQuery(java.lang.String projectId, java.lang.String query, com.google.api.services.bigquery.model.JobStatistics result)
java.util.Collection<com.google.api.services.bigquery.model.Job>
getAllJobs()
com.google.api.services.bigquery.model.Job
getJob(com.google.api.services.bigquery.model.JobReference jobRef)
Gets the specifiedJob
by the givenJobReference
.int
getNumExtractJobCalls()
com.google.api.services.bigquery.model.Job
pollJob(com.google.api.services.bigquery.model.JobReference jobRef, int maxAttempts)
Waits for the job is Done, and returns the job.void
setNumFailuresExpected(int numFailuresExpected)
static void
setUp()
void
startCopyJob(com.google.api.services.bigquery.model.JobReference jobRef, com.google.api.services.bigquery.model.JobConfigurationTableCopy copyConfig)
Start a BigQuery copy job.void
startExtractJob(com.google.api.services.bigquery.model.JobReference jobRef, com.google.api.services.bigquery.model.JobConfigurationExtract extractConfig)
Start a BigQuery extract job.void
startLoadJob(com.google.api.services.bigquery.model.JobReference jobRef, com.google.api.services.bigquery.model.JobConfigurationLoad loadConfig)
Start a BigQuery load job.void
startLoadJob(com.google.api.services.bigquery.model.JobReference jobRef, com.google.api.services.bigquery.model.JobConfigurationLoad loadConfig, com.google.api.client.http.AbstractInputStreamContent streamContent)
Start a BigQuery load job with stream content.void
startQueryJob(com.google.api.services.bigquery.model.JobReference jobRef, com.google.api.services.bigquery.model.JobConfigurationQuery query)
Start a BigQuery query job.
-
-
-
Method Detail
-
close
public void close() throws java.lang.Exception
- Specified by:
close
in interfacejava.lang.AutoCloseable
- Throws:
java.lang.Exception
-
setNumFailuresExpected
public void setNumFailuresExpected(int numFailuresExpected)
-
setUp
public static void setUp()
-
startLoadJob
public void startLoadJob(com.google.api.services.bigquery.model.JobReference jobRef, com.google.api.services.bigquery.model.JobConfigurationLoad loadConfig) throws java.io.IOException
Description copied from interface:BigQueryServices.JobService
Start a BigQuery load job.- Specified by:
startLoadJob
in interfaceBigQueryServices.JobService
- Throws:
java.io.IOException
-
startLoadJob
public void startLoadJob(com.google.api.services.bigquery.model.JobReference jobRef, com.google.api.services.bigquery.model.JobConfigurationLoad loadConfig, com.google.api.client.http.AbstractInputStreamContent streamContent) throws java.lang.InterruptedException, java.io.IOException
Description copied from interface:BigQueryServices.JobService
Start a BigQuery load job with stream content.- Specified by:
startLoadJob
in interfaceBigQueryServices.JobService
- Throws:
java.lang.InterruptedException
java.io.IOException
-
startExtractJob
public void startExtractJob(com.google.api.services.bigquery.model.JobReference jobRef, com.google.api.services.bigquery.model.JobConfigurationExtract extractConfig) throws java.io.IOException
Description copied from interface:BigQueryServices.JobService
Start a BigQuery extract job.- Specified by:
startExtractJob
in interfaceBigQueryServices.JobService
- Throws:
java.io.IOException
-
getNumExtractJobCalls
public int getNumExtractJobCalls()
-
startQueryJob
public void startQueryJob(com.google.api.services.bigquery.model.JobReference jobRef, com.google.api.services.bigquery.model.JobConfigurationQuery query)
Description copied from interface:BigQueryServices.JobService
Start a BigQuery query job.- Specified by:
startQueryJob
in interfaceBigQueryServices.JobService
-
startCopyJob
public void startCopyJob(com.google.api.services.bigquery.model.JobReference jobRef, com.google.api.services.bigquery.model.JobConfigurationTableCopy copyConfig) throws java.io.IOException
Description copied from interface:BigQueryServices.JobService
Start a BigQuery copy job.- Specified by:
startCopyJob
in interfaceBigQueryServices.JobService
- Throws:
java.io.IOException
-
pollJob
public com.google.api.services.bigquery.model.Job pollJob(com.google.api.services.bigquery.model.JobReference jobRef, int maxAttempts) throws java.lang.InterruptedException
Description copied from interface:BigQueryServices.JobService
Waits for the job is Done, and returns the job.Returns null if the
maxAttempts
retries reached.- Specified by:
pollJob
in interfaceBigQueryServices.JobService
- Throws:
java.lang.InterruptedException
-
expectDryRunQuery
public void expectDryRunQuery(java.lang.String projectId, java.lang.String query, com.google.api.services.bigquery.model.JobStatistics result)
-
dryRunQuery
public com.google.api.services.bigquery.model.JobStatistics dryRunQuery(java.lang.String projectId, com.google.api.services.bigquery.model.JobConfigurationQuery query, java.lang.String location)
Description copied from interface:BigQueryServices.JobService
Dry runs the query in the given project.- Specified by:
dryRunQuery
in interfaceBigQueryServices.JobService
-
getAllJobs
public java.util.Collection<com.google.api.services.bigquery.model.Job> getAllJobs()
-
getJob
public com.google.api.services.bigquery.model.Job getJob(com.google.api.services.bigquery.model.JobReference jobRef)
Description copied from interface:BigQueryServices.JobService
Gets the specifiedJob
by the givenJobReference
.Returns null if the job is not found.
- Specified by:
getJob
in interfaceBigQueryServices.JobService
-
-