Skip to content

Commit

Permalink
tests/migrations: add scale test
Browse files Browse the repository at this point in the history
split existing test into common subclass and test itself; use common in scale
  • Loading branch information
bashtanov committed Sep 20, 2024
1 parent 937dd47 commit d21b75f
Show file tree
Hide file tree
Showing 3 changed files with 499 additions and 451 deletions.
35 changes: 35 additions & 0 deletions tests/rptest/scale_tests/data_migrations_test.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,35 @@
# Copyright 2022 Redpanda Data, Inc.
#
# Use of this software is governed by the Business Source License
# included in the file licenses/BSL.md
#
# As of the Change Date specified in that file, in accordance with
# the Business Source License, use of this software will be governed
# by the Apache License, Version 2.0

from rptest.services.cluster import cluster
from ducktape.mark import parametrize
from rptest.tests.data_migrations import DataMigrationsTest

MIB_PER_PARTITION = 4
PARTITIONS_PER_SHARD = 1000


class DataMigrationsScaleTest(DataMigrationsTest):
def __init__(self, *args, **kwargs):
super().__init__(*args,
**kwargs,
extra_rp_conf={
'topic_partitions_per_shard':
PARTITIONS_PER_SHARD,
'topic_memory_per_partition':
MIB_PER_PARTITION * 1024 * 1024,
})

@cluster(num_nodes=3,
log_allow_list=DataMigrationsTest.MIGRATION_LOG_ALLOW_LIST)
@parametrize(topics_count=10000, partitions_count=3)
@parametrize(topics_count=3, partitions_count=10000)
def test_migrate(self, topics_count, partitions_count):
self.do_test_creating_and_listing_migrations(topics_count,
partitions_count)
Loading

0 comments on commit d21b75f

Please sign in to comment.