Support multiple messages from package_update

This updates the workflow wrapper to use the multiple message interface.

Change-Id: I5a04fb057979ca43c725b99900a1607dafe86a81
Partial-Bug: #1646887
This commit is contained in:
Dougal Matthews 2017-03-16 08:43:16 +00:00
parent 0f2b860d86
commit 7378a38928
1 changed files with 6 additions and 6 deletions

View File

@ -32,8 +32,8 @@ def update(clients, **workflow_input):
)
with tripleoclients.messaging_websocket(queue_name) as ws:
message = ws.wait_for_message(execution.id)
assert message['status'] == "SUCCESS", pprint.pformat(message)
for payload in base.wait_for_messages(workflow_client, ws, execution):
assert payload['status'] == "SUCCESS", pprint.pformat(payload)
def update_and_wait(log, clients, stack, plan_name, verbose_level,
@ -81,8 +81,8 @@ def abort_update(clients, **workflow_input):
)
with tripleoclients.messaging_websocket(queue_name) as ws:
message = ws.wait_for_message(execution.id)
assert message['status'] == "SUCCESS", pprint.pformat(message)
for payload in base.wait_for_messages(workflow_client, ws, execution):
assert payload['status'] == "SUCCESS", pprint.pformat(payload)
def clear_breakpoints(clients, **workflow_input):
@ -98,5 +98,5 @@ def clear_breakpoints(clients, **workflow_input):
)
with tripleoclients.messaging_websocket(queue_name) as ws:
message = ws.wait_for_message(execution.id)
assert message['status'] == "SUCCESS", pprint.pformat(message)
for payload in base.wait_for_messages(workflow_client, ws, execution):
assert payload['status'] == "SUCCESS", pprint.pformat(payload)