Skip to content

Commit

Permalink
Merge branch 'master' into feature/vertex-pipeline
Browse files Browse the repository at this point in the history
  • Loading branch information
haroldrubio committed Dec 9, 2024
2 parents e6d11b1 + 10f1782 commit b1bb6cf
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,16 +62,16 @@ def await_queue(super_client=None):
assert not super_client.get_process_logs(status='error')

@staticmethod
def await_queue_edit(super_client, edit_id=None, invitation=None):
print('await_queue_edit', edit_id)
def await_queue_edit(super_client, edit_id=None, invitation=None, count=1, error=False):
expected_status = 'error' if error else 'ok'
while True:
process_logs = super_client.get_process_logs(id=edit_id, invitation=invitation)
if process_logs:
if len(process_logs) >= count and all(process_log['status'] == expected_status for process_log in process_logs):
break

time.sleep(0.5)

assert process_logs[0]['status'] == 'ok'
assert process_logs[0]['status'] == (expected_status), process_logs[0]['log']


@staticmethod
Expand Down

0 comments on commit b1bb6cf

Please sign in to comment.