improve parsing when result stack has multiple items

Hecate2 · Sep 1, 2022 · 74d5dc2 · 74d5dc2
1 parent bbac2e0
commit 74d5dc2
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/neo_fairy_client/rpc/test_client.py b/neo_fairy_client/rpc/test_client.py
@@ -269,8 +269,12 @@ def parse_single_item(item: Union[Dict, List]):
             return result
         if not result['stack']:
             return result['stack']
-        result: List = result['stack'][0]
-        return parse_single_item(result)
+        stack: List = result['stack']
+        if len(stack) > 1:  # typically happens when we invoke a script calling a series of methods
+            return [parse_single_item(item) for item in stack]
+        else:  # if the stack has only 1 item, we simply return the item without a wrapping list
+            result: List = stack[0]
+            return parse_single_item(result)
 
     @classmethod
     def parse_params(cls, param: Union[str, int, dict, Hash160Str, UInt160, UInt256, bytes]) -> Dict[str, str]: