def test_get_definition():
"""
Test get vars definition
"""
variables = Variables()
result_vars = [{
'name': 'Var1',
'type': 'String',
'values': "",
'default_value': None
}, {
'name': 'Var2',
'type': 'Int',
'values': "",
'default_value': None
}, {
'name': 'Var3',
'type': 'String',
'values': ['v1', 'v2'],
'default_value': None
}]
variables.add('Var1')
variables.add('Var2', 'Int')
variables.add('Var3', values=['v1', 'v2'])
assert variables.vars_definition() == json.dumps(result_vars)
def test_task_call():
"""
Test task call to return var definitions
"""
test_args = ['task.py', '--get_vars_definition']
with patch.object(sys, 'argv', test_args):
caputure_output = io.StringIO()
sys.stdout = caputure_output
variables = Variables()
variables.add('Var1')
variables.add('Var2')
with pytest.raises(SystemExit):
Variables().task_call(variables)
sys.stdout = sys.__stdout__
assert caputure_output.getvalue() == variables.vars_definition()
