mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-19 16:15:07 +00:00
dbb8835fdf
We adjust all of the simple cases of cluster tests so they work with `rf_rack_valid_keyspaces: true`. It boils down to assigning nodes to multiple racks. For most of the changes, we do that by: * Using `pytest.mark.prepare_3_racks_cluster` instead of `pytest.mark.prepare_3_nodes_cluster`. * Using an additional argument -- `auto_rack_dc` -- when calling `ManagerClient::servers_add()`. In some cases, we need to assign the racks manually, which may be less obvious, but in every such situation, the tests didn't rely on that assignment, so that doesn't affect them or what they verify.
42 lines
1.7 KiB
Python
42 lines
1.7 KiB
Python
#
|
|
# Copyright (C) 2023-present ScyllaDB
|
|
#
|
|
# SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
|
|
#
|
|
# This file configures pytest for all tests in this directory, and also
|
|
|
|
import pytest
|
|
|
|
from cassandra.protocol import InvalidRequest # type: ignore
|
|
from cassandra.query import SimpleStatement # type: ignore
|
|
from test.cluster.util import new_test_keyspace
|
|
|
|
from test.pylib.manager_client import ManagerClient
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_sticky_coordinator_enforced(manager: ManagerClient) -> None:
|
|
await manager.servers_add(2, cmdline=['--logger-log-level', 'paging=trace'], auto_rack_dc="dc1")
|
|
|
|
cql = manager.get_cql()
|
|
|
|
async with new_test_keyspace(manager, "with replication = {'class': 'NetworkTopologyStrategy', 'replication_factor': 2}") as ks:
|
|
await cql.run_async(f"create table {ks}.tbl (pk int, ck int, v int, primary key (pk, ck))")
|
|
|
|
num_rows = 43
|
|
expected_num_rows = num_rows + 2 # rows + partition-start + partitione-end
|
|
for ck in range(0, num_rows):
|
|
await cql.run_async(f"INSERT INTO {ks}.tbl (pk, ck, v) VALUES (0, {ck}, 100)")
|
|
|
|
unpaged_res = await cql.run_async(f"SELECT * FROM MUTATION_FRAGMENTS({ks}.tbl) WHERE pk = 0")
|
|
assert len(unpaged_res) == expected_num_rows
|
|
|
|
read_stmt = SimpleStatement(f"SELECT * FROM MUTATION_FRAGMENTS({ks}.tbl) WHERE pk = 0", fetch_size=10)
|
|
|
|
# The default round-robin load-balancing policy will jump between the nodes.
|
|
# This should trigger an exception.
|
|
with pytest.raises(
|
|
InvalidRequest,
|
|
match="Moving between coordinators is not allowed in SELECT FROM MUTATION_FRAGMENTS\\(\\) statements.*"):
|
|
await cql.run_async(read_stmt, all_pages=True)
|