Change value from json functions

test/login_logout/test/behave/features/steps/login.py

@@ -32,15 +32,13 @@ def step_impl(context, element, variable):
 
 @then(u'I expect that the "{variable}" in the json output is "{value}"')
 def step_impl(context, variable, value):
-    obj = get_object_from_json_body(context)
-    assert obj[variable] == context.oauth[value]
+    assert context.oauth[value] == get_value_from_json_body(context, variable)
 
 @then(u'I expect that the "{variable}" in the json output contains "{value}"')
 def step_impl(context, variable, value):
-    obj = get_object_from_json_body(context)
-    assert context.oauth[value] in obj[variable]
+    assert context.oauth[value] in get_value_from_json_body(context, variable)
 
-def get_object_from_json_body(context):
+def get_value_from_json_body(context, key):
     obj_serialized = context.behave_driver.get_element("body").text
-    return json.loads(obj_serialized)
+    return json.loads(obj_serialized)[key]