org.apache.beam.it.gcp.bigquery.matchers

Class BigQueryAsserts

java.lang.Object

org.apache.beam.it.gcp.bigquery.matchers.BigQueryAsserts

public class BigQueryAsserts
extends java.lang.Object

Assert utilities for BigQuery tests.

Constructor Summary

Constructors

Constructor and Description
BigQueryAsserts()

Method Summary

Modifier and Type	Method and Description
static org.apache.beam.it.truthmatchers.RecordsSubject	assertThatBigQueryRecords(com.google.cloud.bigquery.TableResult tableResult) Creates a RecordsSubject to assert information within a list of records.
static java.util.List<java.util.Map<java.lang.String,java.lang.Object>>	bigQueryRowsToRecords(java.util.List<com.google.cloud.bigquery.InsertAllRequest.RowToInsert> rows) Convert BigQuery InsertAllRequest.RowToInsert to a list of maps.
static java.util.List<java.util.Map<java.lang.String,java.lang.Object>>	bigQueryRowsToRecords(java.util.List<com.google.cloud.bigquery.InsertAllRequest.RowToInsert> rows, java.util.List<java.lang.String> excludeCols) Convert BigQuery InsertAllRequest.RowToInsert to a list of maps.
static java.util.List<java.util.Map<java.lang.String,java.lang.Object>>	tableResultToRecords(com.google.cloud.bigquery.TableResult tableResult) Convert BigQuery TableResult to a list of maps.

Methods inherited from class java.lang.Object

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

Constructor Detail

BigQueryAsserts

public BigQueryAsserts()

Method Detail

tableResultToRecords

public static java.util.List<java.util.Map<java.lang.String,java.lang.Object>> tableResultToRecords(com.google.cloud.bigquery.TableResult tableResult)

Convert BigQuery TableResult to a list of maps.

Parameters:

tableResult - Table Result to parse

Returns:

List of maps to use in RecordsSubject

bigQueryRowsToRecords

public static java.util.List<java.util.Map<java.lang.String,java.lang.Object>> bigQueryRowsToRecords(java.util.List<com.google.cloud.bigquery.InsertAllRequest.RowToInsert> rows)

Convert BigQuery InsertAllRequest.RowToInsert to a list of maps.

Parameters:

rows - BigQuery rows to parse.

Returns:

List of maps to use in RecordsSubject

bigQueryRowsToRecords

public static java.util.List<java.util.Map<java.lang.String,java.lang.Object>> bigQueryRowsToRecords(java.util.List<com.google.cloud.bigquery.InsertAllRequest.RowToInsert> rows,
                                                                                                     java.util.List<java.lang.String> excludeCols)

Convert BigQuery InsertAllRequest.RowToInsert to a list of maps.

Parameters:

rows - BigQuery rows to parse.

excludeCols - BigQuery columns to filter out of result.

Returns:

List of maps to use in RecordsSubject

assertThatBigQueryRecords

public static org.apache.beam.it.truthmatchers.RecordsSubject assertThatBigQueryRecords(com.google.cloud.bigquery.TableResult tableResult)

Creates a RecordsSubject to assert information within a list of records.

Parameters:

tableResult - Records in BigQuery TableResult format to use in the comparison.

Returns:

Truth Subject to chain assertions.