tests

Author: suthar26

test/test_cloud_client.py

@@ -306,10 +306,10 @@ def before_hook(context):
             hook_called["before"] = True
             return context
 
-        def after_hook(context, variable):
+        def after_hook(context, variable, variable_metadata):
             hook_called["after"] = True
 
-        def finally_hook(context, variable):
+        def finally_hook(context, variable, variable_metadata):
             hook_called["finally"] = True
 
         def error_hook(context, error):
@@ -346,10 +346,10 @@ def before_hook(context):
             hook_called["before"] = True
             raise Exception("Before hook failed")
 
-        def after_hook(context, variable):
+        def after_hook(context, variable, variable_metadata):
             hook_called["after"] = True
 
-        def finally_hook(context, variable):
+        def finally_hook(context, variable, variable_metadata):
             hook_called["finally"] = True
 
         def error_hook(context, error):
@@ -382,10 +382,10 @@ def before_hook(context):
             context_received = context
             return context
 
-        def after_hook(context, variable):
+        def after_hook(context, variable, variable_metadata):
             pass
 
-        def finally_hook(context, variable):
+        def finally_hook(context, variable, variable_metadata):
             pass
 
         def error_hook(context, error):

test/test_local_client.py

@@ -432,7 +432,7 @@ def before_hook(context):
             hook_called["before"] = True
             raise Exception("Before hook failed")
 
-        def after_hook(context, variable):
+        def after_hook(context, variable, variable_metadata):
             hook_called["after"] = True
 
         def finally_hook(context, variable):