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

Class TestPubsub

java.lang.Object

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

All Implemented Interfaces:

org.junit.rules.TestRule

public class TestPubsub
extends java.lang.Object
implements org.junit.rules.TestRule

Test rule which creates a new topic with randomized name and exposes the APIs to work with it.

Deletes topic on shutdown.

Method Summary

Modifier and Type	Method and Description
org.junit.runners.model.Statement	apply(org.junit.runners.model.Statement base, Description description)
void	checkIfAnySubscriptionExists(java.lang.String project, Duration timeoutDuration) Check if topics exist.
static TestPubsub	create() Creates an instance of this rule.
void	publish(java.util.List<PubsubMessage> messages) Publish messages to topicPath().
PubsubClient.TopicPath	topicPath() Topic path where events will be published to.

Methods inherited from class java.lang.Object

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

Method Detail

create

public static TestPubsub create()

Creates an instance of this rule.

Loads GCP configuration from TestPipelineOptions.

apply

public org.junit.runners.model.Statement apply(org.junit.runners.model.Statement base,
                                               Description description)

Specified by:

apply in interface org.junit.rules.TestRule

topicPath

public PubsubClient.TopicPath topicPath()

Topic path where events will be published to.

publish

public void publish(java.util.List<PubsubMessage> messages)
             throws java.io.IOException

Publish messages to topicPath().

Throws:

java.io.IOException

checkIfAnySubscriptionExists

public void checkIfAnySubscriptionExists(java.lang.String project,
                                         Duration timeoutDuration)
                                  throws java.lang.InterruptedException,
                                         java.lang.IllegalArgumentException,
                                         java.io.IOException,
                                         java.util.concurrent.TimeoutException

Check if topics exist.

Parameters:

project - GCP project identifier.

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

Throws:

java.lang.InterruptedException

java.lang.IllegalArgumentException

java.io.IOException

java.util.concurrent.TimeoutException