Merge pull request #4 from ctomkow/key_chain_hang_bug

fix bug when key not found in key_chain

Author: ctomkow

jsonparse/VERSION

@@ -1 +1 @@
-0.6.1
+0.6.2

jsonparse/parser.py

@@ -116,6 +116,10 @@ def key_chain(self, data: Union[dict, list], keys: list) -> list:
 
         while len(keys) >= 1:
 
+            # if key is not found, return empty list
+            if self._queue_size() == 0 and len(keys) >= 1:
+                return []
+
             queue_size_snapshot = self._queue_size()
             key_found = False
 

tests/test_parser.py

@@ -90,7 +90,6 @@ def complex_json(self):
             }
         ]
 
-    # all_inst_of_key
     def test_key(self, parser, complex_json):
 
         result = parser.key(complex_json, "id")
@@ -101,6 +100,11 @@ def test_key(self, parser, complex_json):
             '1004', '1003', '1002', '1001'
         ]
 
+    def test_key_not_found(self, parser, complex_json):
+
+        result = parser.key(complex_json, "key_not_in_data")
+        assert result == []
+
     def test_key_empty_key(self, parser, complex_json):
 
         try:
@@ -131,7 +135,6 @@ def test_key_not_list_or_dict_data(self, parser):
         except TypeError:
             assert True
 
-    # all_inst_of_key_chain
     def test_key_chain(self, parser, complex_json):
 
         result = parser.key_chain(
@@ -180,6 +183,14 @@ def test_key_chain_not_list_or_dict_data(self, parser):
         except TypeError:
             assert True
 
+    def test_key_chain_key_not_found(self, parser, complex_json):
+
+        result = parser.key_chain(
+                complex_json,
+                ["key_not_in_data"]
+            )
+        assert result == []
+
     def test_key_chain_wildcard(self, parser, complex_json):
 
         result = parser.key_chain(