Package org.apache.beam.sdk.io.gcp.testing

Class FakeJobService

    • 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 specified Job by the given JobReference.
      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.

      • Methods inherited from class java.lang.Object

        clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

    • Constructor Detail

      • FakeJobService

        public FakeJobService()

      • FakeJobService

        public FakeJobService​(int numFailures)

    • Method Detail

      • close

        public void close()
           throws java.lang.Exception
        Specified by:
        close in interface java.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 interface BigQueryServices.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 interface BigQueryServices.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 interface BigQueryServices.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 interface BigQueryServices.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 interface BigQueryServices.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 interface BigQueryServices.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 interface BigQueryServices.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 specified Job by the given JobReference.

        Returns null if the job is not found.

        Specified by:
        getJob in interface BigQueryServices.JobService