Fixed tests for async

Author: vladvildanov

tests/test_asyncio/test_commands.py

@@ -24,6 +24,7 @@
     assert_resp_response,
     assert_resp_response_in,
     is_resp2_connection,
+    skip_if_resp_version,
     skip_if_server_version_gte,
     skip_if_server_version_lt,
     skip_unless_arch_bits,
@@ -1412,13 +1413,23 @@ async def test_scard(self, r: redis.Redis):
         await r.sadd("a", "1", "2", "3")
         assert await r.scard("a") == 3
 
+    @skip_if_resp_version(3)
     @pytest.mark.onlynoncluster
     async def test_sdiff(self, r: redis.Redis):
         await r.sadd("a", "1", "2", "3")
         assert set(await r.sdiff("a", "b")) == {b"1", b"2", b"3"}
         await r.sadd("b", "2", "3")
         assert await r.sdiff("a", "b") == {b"1"}
 
+    @skip_if_resp_version(2)
+    @pytest.mark.onlynoncluster
+    async def test_sdiff_resp3(self, r: redis.Redis):
+        await r.sadd("a", "1", "2", "3")
+        assert set(await r.sdiff("a", "b")) == {b"1", b"2", b"3"}
+        await r.sadd("b", "2", "3")
+        assert await r.sdiff("a", "b") == [b"1"]
+
+    @skip_if_resp_version(3)
     @pytest.mark.onlynoncluster
     async def test_sdiffstore(self, r: redis.Redis):
         await r.sadd("a", "1", "2", "3")
@@ -1428,13 +1439,33 @@ async def test_sdiffstore(self, r: redis.Redis):
         assert await r.sdiffstore("c", "a", "b") == 1
         assert await r.smembers("c") == {b"1"}
 
+    @skip_if_resp_version(2)
+    @pytest.mark.onlynoncluster
+    async def test_sdiffstore_resp3(self, r: redis.Redis):
+        await r.sadd("a", "1", "2", "3")
+        assert await r.sdiffstore("c", "a", "b") == 3
+        assert set(await r.smembers("c")) == {b"1", b"2", b"3"}
+        await r.sadd("b", "2", "3")
+        assert await r.sdiffstore("c", "a", "b") == 1
+        assert await r.smembers("c") == [b"1"]
+
+    @skip_if_resp_version(3)
     @pytest.mark.onlynoncluster
     async def test_sinter(self, r: redis.Redis):
         await r.sadd("a", "1", "2", "3")
         assert await r.sinter("a", "b") == set()
         await r.sadd("b", "2", "3")
         assert set(await r.sinter("a", "b")) == {b"2", b"3"}
 
+    @skip_if_resp_version(2)
+    @pytest.mark.onlynoncluster
+    async def test_sinter_resp3(self, r: redis.Redis):
+        await r.sadd("a", "1", "2", "3")
+        assert await r.sinter("a", "b") == []
+        await r.sadd("b", "2", "3")
+        assert await r.sinter("a", "b") == [b"2", b"3"]
+
+    @skip_if_resp_version(3)
     @pytest.mark.onlynoncluster
     async def test_sinterstore(self, r: redis.Redis):
         await r.sadd("a", "1", "2", "3")
@@ -1444,6 +1475,16 @@ async def test_sinterstore(self, r: redis.Redis):
         assert await r.sinterstore("c", "a", "b") == 2
         assert set(await r.smembers("c")) == {b"2", b"3"}
 
+    @skip_if_resp_version(2)
+    @pytest.mark.onlynoncluster
+    async def test_sinterstore_resp3(self, r: redis.Redis):
+        await r.sadd("a", "1", "2", "3")
+        assert await r.sinterstore("c", "a", "b") == 0
+        assert await r.smembers("c") == []
+        await r.sadd("b", "2", "3")
+        assert await r.sinterstore("c", "a", "b") == 2
+        assert set(await r.smembers("c")) == {b"2", b"3"}
+
     async def test_sismember(self, r: redis.Redis):
         await r.sadd("a", "1", "2", "3")
         assert await r.sismember("a", "1")
@@ -1455,6 +1496,7 @@ async def test_smembers(self, r: redis.Redis):
         await r.sadd("a", "1", "2", "3")
         assert set(await r.smembers("a")) == {b"1", b"2", b"3"}
 
+    @skip_if_resp_version(3)
     @pytest.mark.onlynoncluster
     async def test_smove(self, r: redis.Redis):
         await r.sadd("a", "a1", "a2")
@@ -1463,6 +1505,15 @@ async def test_smove(self, r: redis.Redis):
         assert await r.smembers("a") == {b"a2"}
         assert set(await r.smembers("b")) == {b"b1", b"b2", b"a1"}
 
+    @skip_if_resp_version(2)
+    @pytest.mark.onlynoncluster
+    async def test_smove_resp3(self, r: redis.Redis):
+        await r.sadd("a", "a1", "a2")
+        await r.sadd("b", "b1", "b2")
+        assert await r.smove("a", "b", "a1")
+        assert await r.smembers("a") == [b"a2"]
+        assert set(await r.smembers("b")) == {b"b1", b"b2", b"a1"}
+
     async def test_spop(self, r: redis.Redis):
         s = [b"1", b"2", b"3"]
         await r.sadd("a", *s)