org.apache.beam.sdk.io.gcp.bigquery

Class TestBigQuery

java.lang.Object

org.apache.beam.sdk.io.gcp.bigquery.TestBigQuery

All Implemented Interfaces:

TestRule

public class TestBigQuery
extends java.lang.Object
implements TestRule

Test rule which creates a new table with specified schema, with randomized name and exposes few APIs to work with it.

Deletes the table on test shutdown.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static interface	TestBigQuery.PollingAssertion Interface to implement a polling assertion.
class	TestBigQuery.RowsAssertion Interface for creating a polling eventual assertion.

Method Summary

Modifier and Type	Method and Description
Statement	apply(Statement base, Description description)
TestBigQuery.RowsAssertion	assertThatAllRows(Schema rowSchema)
static TestBigQuery	create(Schema tableSchema) Creates an instance of this rule.
java.util.List<Row>	getFlatJsonRows(Schema rowSchema) Loads rows from BigQuery into Rows with given Schema.
TableDataInsertAllResponse	insertRows(Schema rowSchema, Row... rows)
TableReference	tableReference()
java.lang.String	tableSpec()

Methods inherited from class java.lang.Object

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

Method Detail

create

public static TestBigQuery create(Schema tableSchema)

Creates an instance of this rule.

Loads GCP configuration from TestPipelineOptions.

apply

public Statement apply(Statement base,
                       Description description)

Specified by:

apply in interface TestRule

tableSpec

@RequiresNonNull(value="table")
public java.lang.String tableSpec()

tableReference

@RequiresNonNull(value="table")
public TableReference tableReference()

insertRows

@RequiresNonNull(value="table")
public TableDataInsertAllResponse insertRows(Schema rowSchema,
                                                                             Row... rows)
                                                                      throws java.io.IOException

Throws:

java.io.IOException

getFlatJsonRows

public java.util.List<Row> getFlatJsonRows(Schema rowSchema)

Loads rows from BigQuery into Rows with given Schema.

Current implementation only supports flat Rows and target Schemas with Schema.FieldType.STRING fields only.

assertThatAllRows

public TestBigQuery.RowsAssertion assertThatAllRows(Schema rowSchema)