Add support for ModuleCommands in cluster (#2951)

redis · Sep 19, 2023 · 6207641 · 6207641
1 parent 2df75ad
commit 6207641
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 2 deletions.
diff --git a/redis/cluster.py b/redis/cluster.py
@@ -295,6 +295,10 @@ class AbstractRedisCluster:
                 "LATENCY HISTORY",
                 "LATENCY LATEST",
                 "LATENCY RESET",
+                "MODULE LIST",
+                "MODULE LOAD",
+                "MODULE UNLOAD",
+                "MODULE LOADEX",
             ],
             DEFAULT_NODE,
         ),

diff --git a/redis/commands/cluster.py b/redis/commands/cluster.py
@@ -32,11 +32,13 @@
     AsyncFunctionCommands,
     AsyncGearsCommands,
     AsyncManagementCommands,
+    AsyncModuleCommands,
     AsyncScriptCommands,
     DataAccessCommands,
     FunctionCommands,
     GearsCommands,
     ManagementCommands,
+    ModuleCommands,
     PubSubCommands,
     ResponseT,
     ScriptCommands,
@@ -873,6 +875,7 @@ class RedisClusterCommands(
     ScriptCommands,
     FunctionCommands,
     GearsCommands,
+    ModuleCommands,
     RedisModuleCommands,
 ):
     """
@@ -903,6 +906,7 @@ class AsyncRedisClusterCommands(
     AsyncScriptCommands,
     AsyncFunctionCommands,
     AsyncGearsCommands,
+    AsyncModuleCommands,
 ):
     """
     A class for all Redis Cluster commands

diff --git a/tests/test_asyncio/test_commands.py b/tests/test_asyncio/test_commands.py
@@ -3213,7 +3213,6 @@ async def test_memory_usage(self, r: redis.Redis):
         assert isinstance(await r.memory_usage("foo"), int)
 
     @skip_if_server_version_lt("4.0.0")
-    @pytest.mark.onlynoncluster
     async def test_module_list(self, r: redis.Redis):
         assert isinstance(await r.module_list(), list)
         for x in await r.module_list():

diff --git a/tests/test_commands.py b/tests/test_commands.py
@@ -4909,7 +4909,6 @@ def test_latency_latest(self, r: redis.Redis):
     def test_latency_reset(self, r: redis.Redis):
         assert r.latency_reset() == 0
 
-    @pytest.mark.onlynoncluster
     @skip_if_server_version_lt("4.0.0")
     @skip_if_redis_enterprise()
     def test_module_list(self, r):