Python - Implement JSON.MGET command

python/python/glide/async_commands/server_modules/json.py

@@ -213,7 +213,7 @@ async def mget(
     Retrieves the JSON values at the specified `path` stored at multiple `keys`.
 
     Note:
-        In cluster mode, if keys in `keyValueMap` map to different hash slots, the command
+        In cluster mode, if keys in `keys` map to different hash slots, the command
         will be split across these slots and executed separately for each. This means the command
         is atomic only at the slot level. If one or more slot-specific requests fail, the entire
         call will return the first encountered error, even though some requests may have succeeded

python/python/tests/tests_server_modules/test_json.py

@@ -178,6 +178,7 @@ async def test_json_mget(self, glide_client: TGlideClient):
             await json.set(glide_client, key2, "$", OuterJson.dumps(json2_value)) == OK
         )
 
+        # Test with root JSONPath
         result = await json.mget(
             glide_client,
             [key1, key2],
@@ -189,6 +190,7 @@ async def test_json_mget(self, glide_client: TGlideClient):
         ]
         assert result == expected_result
 
+        # Retrieves the full JSON objects from multiple keys.
         result = await json.mget(
             glide_client,
             [key1, key2],
@@ -208,6 +210,7 @@ async def test_json_mget(self, glide_client: TGlideClient):
         expected_result = [b"[1.0]", b"[3.0]"]
         assert result == expected_result
 
+        # Retrieves the value of the 'b' field for multiple keys.
         result = await json.mget(
             glide_client,
             [key1, key2],
@@ -216,6 +219,7 @@ async def test_json_mget(self, glide_client: TGlideClient):
         expected_result = [b'[{"a":1,"b":2.5,"c":true}]', b'[{"a":1,"b":4}]']
         assert result == expected_result
 
+        # Retrieves all values of 'b' fields using recursive path for multiple keys
         result = await json.mget(
             glide_client,
             [key1, key2],
@@ -224,6 +228,7 @@ async def test_json_mget(self, glide_client: TGlideClient):
         expected_result = [b'[{"a":1,"b":2.5,"c":true},2.5]', b'[{"a":1,"b":4},4]']
         assert result == expected_result
 
+        # retrieves the value of the nested 'b.b' field for multiple keys
         result = await json.mget(
             glide_client,
             [key1, key2],
@@ -232,6 +237,15 @@ async def test_json_mget(self, glide_client: TGlideClient):
         expected_result = [b"2.5", b"4"]
         assert result == expected_result
 
+        # Legacy path that exists in only one of the keys
+        result = await json.mget(
+            glide_client,
+            [key1, key2],
+            ".b.c",
+        )
+        expected_result = [b"true", None]
+        assert result == expected_result
+
         # JSONPath doesn't exist
         result = await json.mget(
             glide_client,
@@ -249,7 +263,15 @@ async def test_json_mget(self, glide_client: TGlideClient):
         )
         assert result == [None, None]
 
-        # Keys don't exist
+        # One key doesn't exist
+        result = await json.mget(
+            glide_client,
+            [key1, "{non_existing_key}"],
+            "$.a",
+        )
+        assert result == [b"[1.0]", None]
+
+        # Both keys don't exist
         result = await json.mget(
             glide_client,
             ["{non_existing_key}1", "{non_existing_key}2"],