From e67e4b83b4801c797d9e2080714e78b7b24765d0 Mon Sep 17 00:00:00 2001 From: Alex Willmer Date: Mon, 8 Feb 2021 22:15:18 +0000 Subject: [PATCH] tests: Fix throttling of Docker pulls fixes #809 --- .ci/ansible_install.py | 6 +++--- .ci/ci_lib.py | 14 ++++++++++++++ 2 files changed, 17 insertions(+), 3 deletions(-) diff --git a/.ci/ansible_install.py b/.ci/ansible_install.py index bb659f8a..08f04356 100755 --- a/.ci/ansible_install.py +++ b/.ci/ansible_install.py @@ -16,9 +16,9 @@ batches = [ ] ] -batches.extend( - ['docker pull %s' % (ci_lib.image_for_distro(distro),), 'sleep 1'] +batches.append(ci_lib.throttle( + 'docker pull %s' % (ci_lib.image_for_distro(distro),) for distro in ci_lib.DISTROS -) +)) ci_lib.run_batches(batches) diff --git a/.ci/ci_lib.py b/.ci/ci_lib.py index cf5406d7..524337a9 100644 --- a/.ci/ci_lib.py +++ b/.ci/ci_lib.py @@ -129,6 +129,20 @@ def combine(batch): )) +def throttle(batch, pause=1): + """ + Add pauses between commands in a batch + + >>> throttle(['echo foo', 'echo bar', 'echo baz']) + ['echo foo', 'sleep 1', 'echo bar', 'sleep 1', 'echo baz'] + """ + def _with_pause(batch, pause): + for cmd in batch: + yield cmd + yield 'sleep %i' % (pause,) + return list(_with_pause(batch, pause))[:-1] + + def run_batches(batches): """ Run shell commands grouped into batches, showing an execution trace.