Fixed asyncio cluster test cases

Author: vladvildanov

tests/test_asyncio/test_cluster.py

@@ -33,6 +33,7 @@
     assert_resp_response,
     is_resp2_connection,
     skip_if_redis_enterprise,
+    skip_if_resp_version,
     skip_if_server_version_gte,
     skip_if_server_version_lt,
     skip_unless_arch_bits,
@@ -1750,12 +1751,21 @@ async def test_cluster_rpoplpush(self, r: RedisCluster) -> None:
         assert await r.lrange("{foo}a", 0, -1) == [b"a1", b"a2"]
         assert await r.lrange("{foo}b", 0, -1) == [b"a3", b"b1", b"b2", b"b3"]
 
+    @skip_if_resp_version(3)
     async def test_cluster_sdiff(self, r: RedisCluster) -> None:
         await r.sadd("{foo}a", "1", "2", "3")
-        assert set(await r.sdiff("{foo}a", "{foo}b")) == {b"1", b"2", b"3"}
+        assert await r.sdiff("{foo}a", "{foo}b") == {b"1", b"2", b"3"}
         await r.sadd("{foo}b", "2", "3")
         assert await r.sdiff("{foo}a", "{foo}b") == {b"1"}
 
+    @skip_if_resp_version(2)
+    async def test_cluster_sdiff_resp3(self, r: RedisCluster) -> None:
+        await r.sadd("{foo}a", "1", "2", "3")
+        assert await r.sdiff("{foo}a", "{foo}b") == [b"1", b"2", b"3"]
+        await r.sadd("{foo}b", "2", "3")
+        assert await r.sdiff("{foo}a", "{foo}b") == [b"1"]
+
+    @skip_if_resp_version(3)
     async def test_cluster_sdiffstore(self, r: RedisCluster) -> None:
         await r.sadd("{foo}a", "1", "2", "3")
         assert await r.sdiffstore("{foo}c", "{foo}a", "{foo}b") == 3
@@ -1764,12 +1774,30 @@ async def test_cluster_sdiffstore(self, r: RedisCluster) -> None:
         assert await r.sdiffstore("{foo}c", "{foo}a", "{foo}b") == 1
         assert await r.smembers("{foo}c") == {b"1"}
 
+    @skip_if_resp_version(2)
+    async def test_cluster_sdiffstore_resp3(self, r: RedisCluster) -> None:
+        await r.sadd("{foo}a", "1", "2", "3")
+        assert await r.sdiffstore("{foo}c", "{foo}a", "{foo}b") == 3
+        assert await r.smembers("{foo}c") == [b"1", b"2", b"3"]
+        await r.sadd("{foo}b", "2", "3")
+        assert await r.sdiffstore("{foo}c", "{foo}a", "{foo}b") == 1
+        assert await r.smembers("{foo}c") == [b"1"]
+
+    @skip_if_resp_version(3)
     async def test_cluster_sinter(self, r: RedisCluster) -> None:
         await r.sadd("{foo}a", "1", "2", "3")
         assert await r.sinter("{foo}a", "{foo}b") == set()
         await r.sadd("{foo}b", "2", "3")
         assert set(await r.sinter("{foo}a", "{foo}b")) == {b"2", b"3"}
 
+    @skip_if_resp_version(2)
+    async def test_cluster_sinter_resp3(self, r: RedisCluster) -> None:
+        await r.sadd("{foo}a", "1", "2", "3")
+        assert await r.sinter("{foo}a", "{foo}b") == []
+        await r.sadd("{foo}b", "2", "3")
+        assert await r.sinter("{foo}a", "{foo}b") == [b"2", b"3"]
+
+    @skip_if_resp_version(3)
     async def test_cluster_sinterstore(self, r: RedisCluster) -> None:
         await r.sadd("{foo}a", "1", "2", "3")
         assert await r.sinterstore("{foo}c", "{foo}a", "{foo}b") == 0
@@ -1778,13 +1806,31 @@ async def test_cluster_sinterstore(self, r: RedisCluster) -> None:
         assert await r.sinterstore("{foo}c", "{foo}a", "{foo}b") == 2
         assert set(await r.smembers("{foo}c")) == {b"2", b"3"}
 
+    @skip_if_resp_version(2)
+    async def test_cluster_sinterstore_resp3(self, r: RedisCluster) -> None:
+        await r.sadd("{foo}a", "1", "2", "3")
+        assert await r.sinterstore("{foo}c", "{foo}a", "{foo}b") == 0
+        assert await r.smembers("{foo}c") == []
+        await r.sadd("{foo}b", "2", "3")
+        assert await r.sinterstore("{foo}c", "{foo}a", "{foo}b") == 2
+        assert await r.smembers("{foo}c") == [b"2", b"3"]
+
+    @skip_if_resp_version(3)
     async def test_cluster_smove(self, r: RedisCluster) -> None:
         await r.sadd("{foo}a", "a1", "a2")
         await r.sadd("{foo}b", "b1", "b2")
         assert await r.smove("{foo}a", "{foo}b", "a1")
         assert await r.smembers("{foo}a") == {b"a2"}
         assert set(await r.smembers("{foo}b")) == {b"b1", b"b2", b"a1"}
 
+    @skip_if_resp_version(2)
+    async def test_cluster_smove_resp3(self, r: RedisCluster) -> None:
+        await r.sadd("{foo}a", "a1", "a2")
+        await r.sadd("{foo}b", "b1", "b2")
+        assert await r.smove("{foo}a", "{foo}b", "a1")
+        assert await r.smembers("{foo}a") == [b"a2"]
+        assert await r.smembers("{foo}b") == [b"b1", b"b2", b"a1"]
+
     async def test_cluster_sunion(self, r: RedisCluster) -> None:
         await r.sadd("{foo}a", "1", "2")
         await r.sadd("{foo}b", "2", "3")