Package 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 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	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	Description
Statement	apply(Statement base, Description description)
TestBigQuery.RowsAssertion	assertThatAllRows(Schema rowSchema)
static TestBigQuery	create(Schema tableSchema)	Creates an instance of this rule.
List<Row>	getFlatJsonRows(Schema rowSchema)	Loads rows from BigQuery into Rows with given Schema.
TableDataInsertAllResponse	insertRows(Schema rowSchema, Row... rows)
TableReference	tableReference()
String	tableSpec()

Methods inherited from class java.lang.Object

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

Method Details

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("table")
public String tableSpec()

tableReference

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

insertRows

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

Throws:

IOException

getFlatJsonRows

public 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)