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

Class TestPubsub

java.lang.Object

org.apache.beam.sdk.io.gcp.pubsub.TestPubsub

All Implemented Interfaces:

TestRule

public class TestPubsub
extends Object
implements TestRule

Test rule which creates a new topic and subscription with randomized names and exposes the APIs to work with them.

Deletes topic and subscription on shutdown.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static interface	TestPubsub.PollingAssertion

Method Summary

Modifier and Type	Method	Description
Statement	apply(Statement base, Description description)
void	assertSubscriptionEventuallyCreated(String project, Duration timeoutDuration)	Block until a subscription is created for this test topic in the specified project.
TestPubsub.PollingAssertion	assertThatTopicEventuallyReceives(Matcher<PubsubMessage>... matchers)	Repeatedly pull messages from subscriptionPath() until receiving one for each matcher (or timeout is reached), then assert that the received messages match the expectations.
void	checkIfAnySubscriptionExists(String project, Duration timeoutDuration)	Deprecated. Use assertSubscriptionEventuallyCreated(java.lang.String, org.joda.time.Duration).
static TestPubsub	create()	Creates an instance of this rule using options provided by TestPipeline.testingPipelineOptions().
static TestPubsub	fromOptions(PipelineOptions options)	Creates an instance of this rule using provided options.
void	publish(List<PubsubMessage> messages)	Publish messages to topicPath().
PubsubClient.SubscriptionPath	subscriptionPath()	Subscription path used to listen for messages on topicPath().
PubsubClient.TopicPath	topicPath()	Topic path where events will be published to.
List<PubsubMessage>	waitForNMessages(int n, Duration timeoutDuration)	Repeatedly pull messages from subscriptionPath(), returns after receiving n messages or after waiting for timeoutDuration.

Methods inherited from class java.lang.Object

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

Method Details

create

public static TestPubsub create()

Creates an instance of this rule using options provided by TestPipeline.testingPipelineOptions().

Loads GCP configuration from TestPipelineOptions.

fromOptions

public static TestPubsub fromOptions(PipelineOptions options)

Creates an instance of this rule using provided options.

Loads GCP configuration from TestPipelineOptions.

apply

public Statement apply(Statement base,
 Description description)

Specified by:

apply in interface TestRule

topicPath

public PubsubClient.TopicPath topicPath()

Topic path where events will be published to.

subscriptionPath

public PubsubClient.SubscriptionPath subscriptionPath()

Subscription path used to listen for messages on topicPath().

publish

public void publish(List<PubsubMessage> messages)

Publish messages to topicPath().

waitForNMessages

public List<PubsubMessage> waitForNMessages(int n,
 Duration timeoutDuration)
                                     throws IOException,
InterruptedException

Repeatedly pull messages from subscriptionPath(), returns after receiving n messages or after waiting for timeoutDuration.

Throws:

IOException

InterruptedException

assertThatTopicEventuallyReceives

public TestPubsub.PollingAssertion assertThatTopicEventuallyReceives(Matcher<PubsubMessage>... matchers)

Repeatedly pull messages from subscriptionPath() until receiving one for each matcher (or timeout is reached), then assert that the received messages match the expectations.

Example usage:

invalid input: '{@code
 testTopic
   .assertThatTopicEventuallyReceives(
     hasProperty("payload", equalTo("hello".getBytes(StandardCharsets.US_ASCII))),
     hasProperty("payload", equalTo("world".getBytes(StandardCharsets.US_ASCII))))
   .waitForUpTo(Duration.standardSeconds(20));
 </pre>'

checkIfAnySubscriptionExists

@Deprecated
public void checkIfAnySubscriptionExists(String project,
 Duration timeoutDuration)
                                  throws InterruptedException,
IllegalArgumentException,
IOException,
TimeoutException

Deprecated.

Use assertSubscriptionEventuallyCreated(java.lang.String, org.joda.time.Duration).

Check if topics exist.

Parameters:

project - GCP project identifier.

timeoutDuration - Joda duration that sets a period of time before checking times out.

Throws:

InterruptedException

IllegalArgumentException

IOException

TimeoutException

assertSubscriptionEventuallyCreated

public void assertSubscriptionEventuallyCreated(String project,
 Duration timeoutDuration)
                                         throws InterruptedException,
IllegalArgumentException,
IOException

Block until a subscription is created for this test topic in the specified project. Throws AssertionError if timeoutDuration is reached before a subscription is created.

Parameters:

project - GCP project identifier.

timeoutDuration - Joda duration before timeout occurs.

Throws:

InterruptedException

IllegalArgumentException

IOException