test hsetnx

amirreza8002 · Dec 13, 2024 · bc93c19 · bc93c19
1 parent 6f754fd
commit bc93c19
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/tests/test_backend.py b/tests/test_backend.py
@@ -824,6 +824,19 @@ def test_hset(self, cache: ValkeyCache):
         assert cache.hexists("foo_hash1", "foo1")
         assert cache.hexists("foo_hash1", "foo2")
 
+    def test_hsetnx(self, cache: ValkeyCache):
+        if isinstance(cache.client, ShardClient):
+            pytest.skip("ShardClient doesn't support get_client")
+
+        res = cache.hsetnx("bar_hash1", "foo1", "baz1")
+        assert res == 1
+        result = cache.hget("bar_hash1", "foo1")
+        assert result == "baz1"
+        res = cache.hsetnx("bar_hash1", "foo1", "baz2")
+        assert res == 0
+        result = cache.hget("bar_hash1", "foo1")
+        assert result == "baz1"
+
     def test_hdel(self, cache: ValkeyCache):
         if isinstance(cache.client, ShardClient):
             pytest.skip("ShardClient doesn't support get_client")

diff --git a/tests/tests_async/test_backend.py b/tests/tests_async/test_backend.py
@@ -852,6 +852,16 @@ async def test_hset(self, cache: AsyncValkeyCache):
         assert await cache.ahexists("foo_hash1", "foo1")
         assert await cache.ahexists("foo_hash1", "foo2")
 
+    async def test_hsetnx(self, cache: AsyncValkeyCache):
+        res = await cache.ahsetnx("bar_hash1", "foo1", "baz1")
+        assert res == 1
+        result = await cache.ahget("bar_hash1", "foo1")
+        assert result == "baz1"
+        res = await cache.ahsetnx("bar_hash1", "foo1", "baz2")
+        assert res == 0
+        result = await cache.ahget("bar_hash1", "foo1")
+        assert result == "baz1"
+
     async def test_hdel(self, cache: AsyncValkeyCache):
         # if isinstance(cache.client, ShardClient):
         #     pytest.skip("ShardClient doesn't support get_client")