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()