-
Notifications
You must be signed in to change notification settings - Fork 579
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
ducky: add si test consuming from blocked s3
When trying to consume from blocked S3, time out error should be returned
- Loading branch information
Elena Anyusheva
committed
Apr 29, 2022
1 parent
5e4bba6
commit 4eefd6e
Showing
1 changed file
with
84 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,84 @@ | ||
# Copyright 2021 Redpanda Data, Inc. | ||
# | ||
# Licensed as a Redpanda Enterprise file under the Redpanda Community | ||
# License (the "License"); you may not use this file except in compliance with | ||
# the License. You may obtain a copy of the License at | ||
# | ||
# https://github.com/redpanda-data/redpanda/blob/master/licenses/rcl.md | ||
|
||
from rptest.services.cluster import cluster | ||
from rptest.tests.redpanda_test import RedpandaTest | ||
from rptest.services.redpanda import SISettings | ||
|
||
from rptest.clients.types import TopicSpec | ||
from rptest.clients.rpk import RpkTool, RpkException | ||
from rptest.clients.kafka_cli_tools import KafkaCliTools | ||
from rptest.util import ( | ||
produce_until_segments, | ||
wait_for_segments_removal, | ||
firewall_blocked, | ||
) | ||
|
||
# Log errors expected when connectivity between redpanda and the S3 | ||
# backend is disrupted | ||
CONNECTION_ERROR_LOGS = [ | ||
"archival - .*Failed to create archivers", | ||
|
||
# e.g. archival - [fiber1] - service.cc:484 - Failed to upload 3 segments out of 4 | ||
r"archival - .*Failed to upload \d+ segments" | ||
] | ||
|
||
|
||
class ShadowIndexingFirewallTest(RedpandaTest): | ||
log_segment_size = 1048576 # 1MB | ||
retention_bytes = 1024 # 1 KB | ||
|
||
s3_topic_name = "panda-topic" | ||
topics = (TopicSpec(name=s3_topic_name, | ||
partition_count=1, | ||
replication_factor=3), ) | ||
|
||
def __init__(self, test_context): | ||
si_settings = SISettings(cloud_storage_reconciliation_interval_ms=500, | ||
cloud_storage_max_connections=5, | ||
log_segment_size=self.log_segment_size) | ||
self.s3_bucket_name = si_settings.cloud_storage_bucket | ||
|
||
super(ShadowIndexingFirewallTest, | ||
self).__init__(test_context=test_context, | ||
si_settings=si_settings) | ||
|
||
self._s3_port = si_settings.cloud_storage_api_endpoint_port | ||
|
||
self.kafka_tools = KafkaCliTools(self.redpanda) | ||
self.rpk = RpkTool(self.redpanda) | ||
|
||
def tearDown(self): | ||
super().tearDown() | ||
|
||
@cluster(num_nodes=3, log_allow_list=CONNECTION_ERROR_LOGS) | ||
def test_consume_from_blocked_s3(self): | ||
produce_until_segments(redpanda=self.redpanda, | ||
topic=self.s3_topic_name, | ||
partition_idx=0, | ||
count=5, | ||
acks=-1) | ||
|
||
self.rpk.alter_topic_config(self.s3_topic_name, | ||
TopicSpec.PROPERTY_RETENTION_BYTES, | ||
self.retention_bytes) | ||
|
||
wait_for_segments_removal(redpanda=self.redpanda, | ||
topic=self.s3_topic_name, | ||
partition_idx=0, | ||
count=4) | ||
"""Disconnect redpanda from S3 and try to read starting with offset 0""" | ||
with firewall_blocked(self.redpanda.nodes, self._s3_port): | ||
try: | ||
out = self.rpk.consume(topic=self.s3_topic_name) | ||
except RpkException as e: | ||
assert 'timed out' in e.msg | ||
else: | ||
raise RuntimeError( | ||
f"RPK consume should have timed out, but ran with output: {out}" | ||
) |