def test_get_waiter_args_missing_awswaitin():
"""Missing awsWaiterIn raises."""
context = Context({'k1': 'v1'})
with pytest.raises(KeyNotInContextError) as err_info:
client_in, service_name, waiter_name = wait.get_waiter_args(context)
assert str(err_info.value) == (
"context['awsWaitIn'] doesn't exist. It must exist for "
"pypyraws.steps.wait.")
def test_get_waiter_args_missing_awswaitin():
"""Missing awsWaiterIn raises"""
context = Context({'k1': 'v1'})
with pytest.raises(KeyNotInContextError) as err_info:
client_in, service_name, waiter_name = wait.get_waiter_args(
context)
assert repr(err_info.value) == (
"KeyNotInContextError(\'awsWaitIn not found in the pypyr "
"context.',)")
def test_get_waiter_args_missing_waitername():
"""Missing waiterName raises."""
context = Context({
'k1': 'v1',
'awsWaitIn': {
'serviceName': 'service name',
'arbKey': 'arb_value'
}
})
with pytest.raises(KeyNotInContextError) as err_info:
client_in, service_name, waiter_name = wait.get_waiter_args(context)
assert str(err_info.value) == ("awsWaitIn missing required key for "
"pypyraws.steps.wait: 'waiterName'")
def test_get_waiter_args_waitername_empty():
"""Whitespace serviceName raises."""
context = Context({
'k1': 'v1',
'awsWaitIn': {
'serviceName': 'service name',
'waiterName': ' ',
'arbKey': 'arb_value'
}
})
with pytest.raises(KeyInContextHasNoValueError) as err_info:
client_in, service_name, waiter_name = wait.get_waiter_args(context)
assert str(err_info.value) == ("waiterName required in awsWaitIn "
"for pypyraws.steps.wait")
def test_get_waiter_args_pass():
"""get_service_args pass."""
context = Context({
'k1': 'v1',
'awsWaitIn': {
'serviceName': 'service name',
'waiterName': 'waiter_name',
'arbKey': 'arb_value'
}
})
client_in, service_name, waiter_name = wait.get_waiter_args(context)
assert client_in == {
'serviceName': 'service name',
'waiterName': 'waiter_name',
'arbKey': 'arb_value'
}
assert service_name == 'service name'
assert waiter_name == 'waiter_name'
