Skip to content

samples(test): update exactly once delivery test #628

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
anguillanneuf merged 1 commit into from
Mar 29, 2022
Merged

samples(test): update exactly once delivery test #628

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 16 additions & 20 deletions samples/snippets/subscriber_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,13 @@ def publisher_client() -> Generator[pubsub_v1.PublisherClient, None, None]:
yield pubsub_v1.PublisherClient()


@pytest.fixture(scope="module")
def regional_publisher_client() -> Generator[pubsub_v1.PublisherClient, None, None]:
client_options = {"api_endpoint": "us-east1-pubsub.googleapis.com:443"}
publisher = pubsub_v1.PublisherClient(client_options=client_options)
yield publisher


@pytest.fixture(scope="module")
def topic(publisher_client: pubsub_v1.PublisherClient) -> Generator[str, None, None]:
topic_path = publisher_client.topic_path(PROJECT_ID, TOPIC)
Expand Down Expand Up @@ -697,33 +704,22 @@ def eventually_consistent_test() -> None:


def test_receive_messages_with_exactly_once_delivery_enabled(
publisher_client: pubsub_v1.PublisherClient,
regional_publisher_client: pubsub_v1.PublisherClient,
exactly_once_delivery_topic: str,
subscription_eod: str,
capsys: CaptureFixture[str],
) -> None:

typed_backoff = cast(
Callable[[C], C], backoff.on_exception(backoff.expo, Unknown, max_time=60),
)
message_ids = _publish_messages(regional_publisher_client, exactly_once_delivery_topic)

@typed_backoff
def eventually_consistent_test() -> None:
message_ids = _publish_messages(publisher_client, exactly_once_delivery_topic)

subscriber.receive_messages_with_exactly_once_delivery_enabled(
PROJECT_ID, SUBSCRIPTION_EOD, 10
)

out, _ = capsys.readouterr()
assert "Listening" in out
assert subscription_eod in out
assert "Received" in out
assert "Ack" in out
for message_id in message_ids:
assert message_id in out
subscriber.receive_messages_with_exactly_once_delivery_enabled(
PROJECT_ID, SUBSCRIPTION_EOD, 10
)

eventually_consistent_test()
out, _ = capsys.readouterr()
assert subscription_eod in out
for message_id in message_ids:
assert message_id in out


def test_listen_for_errors(
Expand Down