From 022a3e8314daa59b31fdce1d32e0e74d77f564cc Mon Sep 17 00:00:00 2001 From: dvora-h <67596500+dvora-h@users.noreply.github.com> Date: Wed, 14 Dec 2022 14:04:57 +0200 Subject: Raising NotImplementedError for certain CLUSTER commands (#2504) Co-authored-by: Chayim --- redis/commands/cluster.py | 10 ++++++++++ tests/test_cluster.py | 8 ++++++++ 2 files changed, 18 insertions(+) diff --git a/redis/commands/cluster.py b/redis/commands/cluster.py index f0eaaf7..f434f36 100644 --- a/redis/commands/cluster.py +++ b/redis/commands/cluster.py @@ -644,6 +644,16 @@ class ClusterManagementCommands(ManagementCommands): """ return self.execute_command("CLUSTER LINKS", target_nodes=target_node) + def cluster_flushslots(self, target_nodes: Optional["TargetNodesT"] = None) -> None: + raise NotImplementedError( + "CLUSTER FLUSHSLOTS is intentionally not implemented in the client." + ) + + def cluster_bumpepoch(self, target_nodes: Optional["TargetNodesT"] = None) -> None: + raise NotImplementedError( + "CLUSTER BUMPEPOCH is intentionally not implemented in the client." + ) + def readonly(self, target_nodes: Optional["TargetNodesT"] = None) -> ResponseT: """ Enables read queries. diff --git a/tests/test_cluster.py b/tests/test_cluster.py index e45780d..9866cfc 100644 --- a/tests/test_cluster.py +++ b/tests/test_cluster.py @@ -1289,6 +1289,14 @@ class TestClusterRedisCommands: for i in range(0, len(res) - 1, 2): assert res[i][3] == res[i + 1][3] + def test_cluster_flshslots_not_implemented(self, r): + with pytest.raises(NotImplementedError): + r.cluster_flushslots() + + def test_cluster_bumpepoch_not_implemented(self, r): + with pytest.raises(NotImplementedError): + r.cluster_bumpepoch() + @skip_if_redis_enterprise() def test_readonly(self): r = get_mocked_redis_client(host=default_host, port=default_port) -- cgit v1.2.1