fixing tests and lint

Author: dvora-h

redis/commands/json/commands.py

@@ -1,11 +1,12 @@
+import os
+from json import JSONDecodeError, loads
+
 from deprecated import deprecated
 
 from redis.exceptions import DataError
 
 from .decoders import decode_dict_keys
 from .path import Path
-from json import loads, JSONDecodeError
-import os
 
 
 class JSONCommands:
@@ -217,31 +218,37 @@ def set(self, name, path, obj, nx=False, xx=False, decode_keys=False):
 
     def set_file(self, name, path, file_name, nx=False, xx=False, decode_keys=False):
         """
-        Set the JSON value at key ``name`` under the ``path`` to the contents of the json file ``file_name``.
+        Set the JSON value at key ``name`` under the ``path`` to the content
+        of the json file ``file_name``.
 
         ``nx`` if set to True, set ``value`` only if it does not exist.
         ``xx`` if set to True, set ``value`` only if it exists.
         ``decode_keys`` If set to True, the keys of ``obj`` will be decoded
         with utf-8.
 
         """
-        try:
-            file_content = loads(file_name)
-        except JSONDecodeError:
-            raise JSONDecodeError("Inappropriate file type, set_file() requires json file")
-        
+
+        with open(file_name, "r") as fp:
+            file_content = loads(fp.read())
+
         return self.set(name, path, file_content, nx, xx, decode_keys)
 
+    def set_path(self, json_path, root_folder, nx=False, xx=False, decode_keys=False):
+        """
+        Iterate over ``root_folder`` and set each JSON file to a value
+        under ``json_path`` with the file name as the key.
 
-    def set_path(self, json_path, root_directory , nx=False, xx=False, decode_keys=False):
-        """ 
+        ``nx`` if set to True, set ``value`` only if it does not exist.
+        ``xx`` if set to True, set ``value`` only if it exists.
+        ``decode_keys`` If set to True, the keys of ``obj`` will be decoded
+        with utf-8.
 
         """
         set_files_result = {}
-        for root, dirs, files in os.walk(root_directory):
+        for root, dirs, files in os.walk(root_folder):
             for file in files:
                 try:
-                    file_name = file.rsplit('.')[0]
+                    file_name = os.path.join(root, file).rsplit(".")[0]
                     file_path = os.path.join(root, file)
                     self.set_file(file_name, json_path, file_path, nx, xx, decode_keys)
                     set_files_result[os.path.join(root, file)] = True
@@ -250,8 +257,6 @@ def set_path(self, json_path, root_directory , nx=False, xx=False, decode_keys=F
 
         return set_files_result
 
-
-
     def strlen(self, name, path=None):
         """Return the length of the string JSON value under ``path`` at key
         ``name``.

tests/test_json.py

@@ -1396,11 +1396,38 @@ def test_custom_decoder(client):
 
 @pytest.mark.redismod
 def test_set_file(client):
-    import tempfile
     import json
+    import tempfile
 
+    obj = {"hello": "world"}
     jsonfile = tempfile.NamedTemporaryFile(suffix=".json")
-    with open(jsonfile.name, 'w+') as fp:
+    with open(jsonfile.name, "w+") as fp:
+        fp.write(json.dumps(obj))
+
+    nojsonfile = tempfile.NamedTemporaryFile()
+    nojsonfile.write(b"Hello World")
+
+    assert client.json().set_file("test", Path.rootPath(), jsonfile.name)
+    assert client.json().get("test") == obj
+    with pytest.raises(json.JSONDecodeError):
+        client.json().set_file("test2", Path.rootPath(), nojsonfile.name)
+
+
+@pytest.mark.redismod
+def test_set_path(client):
+    import json
+    import os
+    import tempfile
+
+    root = tempfile.mkdtemp()
+    sub = tempfile.mkdtemp(dir=root)
+    ospointer, jsonfile = tempfile.mkstemp(suffix=".json", dir=sub)
+    ospointer2, nojsonfile = tempfile.mkstemp(dir=root)
+
+    with open(jsonfile, "w+") as fp:
         fp.write(json.dumps({"hello": "world"}))
+    open(nojsonfile, "a+").write("hello")
 
-    assert client.json().set("test", Path.rootPath(), jsonfile.name)
+    result = {"/private" + jsonfile: True, "/private" + nojsonfile: False}
+    assert client.json().set_path(Path.rootPath(), os.path.realpath(root)) == result
+    assert client.json().get("/private" + jsonfile.rsplit(".")[0]) == {"hello": "world"}