Example #1
0
def test_set_parameters_error():
    test_workflow = ArgoWorkflow()
    test_workflow.body = {
        'spec': {
            'arguments': {
                'parameters': [{
                    'name': 'test-param-1',
                    'value': 'test-value-1'
                }, {
                    'name': 'test-param-2'
                }]
            }
        }
    }

    with pytest.raises(KeyError):
        test_workflow.parameters = {'test-param-1': 'new-value'}
Example #2
0
def test_set_parameters():
    test_workflow = ArgoWorkflow()
    test_workflow.body = {
        'spec': {
            'arguments': {
                'parameters': [{
                    'name': 'test-param-1',
                    'value': 'test-value-1'
                }, {
                    'name': 'test-param-2',
                    'value': 'test-value-2'
                }]
            }
        }
    }

    test_workflow.parameters = {'test-param-2': 'new-value'}

    assert test_workflow.parameters == {
        'test-param-1': 'test-value-1',
        'test-param-2': 'new-value'
    }