com.google.cloud.dataflow.sdk.testing

Class DataflowAssert

java.lang.Object

com.google.cloud.dataflow.sdk.testing.DataflowAssert

public class DataflowAssert
extends java.lang.Object

An assertion on the contents of a PCollection incorporated into the pipeline. Such an assertion can be checked no matter what kind of PipelineRunner is used.

Note that the DataflowAssert call must precede the call to Pipeline.run().

Examples of use:

Pipeline p = TestPipeline.create();
 ...
 PCollection<String> output =
      input
      .apply(ParDo.of(new TestDoFn()));
 DataflowAssert.that(output)
     .containsInAnyOrder("out1", "out2", "out3");
 ...
 PCollection<Integer> ints = ...
 PCollection<Integer> sum =
     ints
     .apply(Combine.globally(new SumInts()));
 DataflowAssert.that(sum)
     .is(42);
 ...
 p.run();
 

JUnit and Hamcrest must be linked in by any code that uses DataflowAssert.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static interface	DataflowAssert.AssertRelation<Actual,Expected> A serializable function implementing a binary predicate between types Actual and Expected.
static class	DataflowAssert.AssertThat<Actual,Expected> A SerializableFunction that performs an Assert.assertThat() operation using a Matcher operation.
static class	DataflowAssert.IterableAssert<T> An assertion about the contents of a PCollectionView<Iterable<T>>.
static class	DataflowAssert.SingletonAssert<T> An assertion about the single value of type T associated with a PCollectionView<T>.

Method Summary

Methods

Modifier and Type	Method and Description
static <T> DataflowAssert.IterableAssert<T>	that(PCollection<T> actual) Constructs an DataflowAssert.IterableAssert for the elements of the provided PCollection<T>.
static <T> DataflowAssert.IterableAssert<T>	thatIterable(PCollectionView<java.lang.Iterable<T>> actual) Constructs an DataflowAssert.IterableAssert for the value of the provided PCollectionView PCollectionView<Iterable<T>>.
static <T> DataflowAssert.SingletonAssert<T>	thatSingleton(PCollection<T> actual) Constructs a DataflowAssert.SingletonAssert for the value of the provided PCollection PCollection<T>, which must be a singleton.
static <T> DataflowAssert.IterableAssert<T>	thatSingletonIterable(PCollection<java.lang.Iterable<T>> actual) Constructs an DataflowAssert.IterableAssert for the value of the provided PCollection<Iterable<T>>, which must be a singleton.

Methods inherited from class java.lang.Object

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

Method Detail

that

public static <T> DataflowAssert.IterableAssert<T> that(PCollection<T> actual)

Constructs an DataflowAssert.IterableAssert for the elements of the provided PCollection<T>.

thatSingletonIterable

public static <T> DataflowAssert.IterableAssert<T> thatSingletonIterable(PCollection<java.lang.Iterable<T>> actual)

Constructs an DataflowAssert.IterableAssert for the value of the provided PCollection<Iterable<T>>, which must be a singleton.

thatIterable

public static <T> DataflowAssert.IterableAssert<T> thatIterable(PCollectionView<java.lang.Iterable<T>> actual)

Constructs an DataflowAssert.IterableAssert for the value of the provided PCollectionView PCollectionView<Iterable<T>>.

thatSingleton

public static <T> DataflowAssert.SingletonAssert<T> thatSingleton(PCollection<T> actual)

Constructs a DataflowAssert.SingletonAssert for the value of the provided PCollection PCollection<T>, which must be a singleton.