Merge branch 'feat/improve_storage' of github.com:josemiguel-rodrigue…

…znaranjo/toolium into feat/improve_storage
Telefonica · Jun 26, 2024 · 87eefd3 · 87eefd3
2 parents 4f93209 + f3b8268
commit 87eefd3
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/toolium/utils/dataset.py b/toolium/utils/dataset.py
@@ -17,7 +17,6 @@
 """
 
 import base64
-import collections
 import datetime
 import json
 import logging
@@ -684,8 +683,11 @@ def _get_initial_value_from_context(initial_key, context):
     """
     context_storage = context.storage if hasattr(context, 'storage') else {}
     if hasattr(context, 'feature_storage'):
-        # context.feature_storage is initialized only when before_feature method is called
-        context_storage = collections.ChainMap(context.storage, context.feature_storage)
+        # Merge feature to storage, context.feature_storage is initialized only when before_feature method is called
+        context_storage.update(context.feature_storage)
+    if hasattr(context, 'run_storage'):
+        # Merge run to storage, context.run_storage is initialized only when before_all method is called
+        context_storage.update(context.run_storage)
     if initial_key in context_storage:
         value = context_storage[initial_key]
     elif hasattr(context, initial_key):