Docker pull runner

runci/engine/core.py

@@ -113,6 +113,7 @@ def status(self):
     "runci.engine.runner.compose_build",
     "runci.engine.runner.compose_run",
     "runci.engine.runner.docker_build",
+    "runci.engine.runner.docker_pull",
     "runci.engine.runner.target_run",
     "runci.engine.listener.terminal",
 ]

runci/engine/runner/docker_pull.py

@@ -0,0 +1,25 @@
+import asyncio
+
+from .base import RunnerBase
+from runci.entities.context import Context
+
+
+class DockerPullRunner(RunnerBase):
+    _selector = 'docker-pull'
+
+    async def run_internal(self, context: Context):
+        images = self.spec.get("image",
+                               self.spec.get("_", None))
+
+        if images is None:
+            raise Exception("Image name should be specified for docker-pull step")
+
+        base_args = ['docker', 'pull']
+
+        tasks = []
+        for image in images.split(' '):
+            args = list(base_args)
+            args.append(image)
+            tasks.append(asyncio.create_task(self._run_process(args)))
+
+        await asyncio.wait(tasks)

tests/test_runner_docker_pull.py

@@ -0,0 +1,45 @@
+import asyncio
+import unittest
+
+from runci.entities.config import Project, Target, Step
+from runci.entities.parameters import Parameters
+from runci.engine.core import create_context, DependencyTree
+from runci.engine.runner.docker_pull import DockerPullRunner
+from unittest.mock import patch, call
+
+
+class test_runner_docker_build(unittest.TestCase):
+    step = Step("test", "docker-pull", {
+        "image": "busybox alpine"
+    })
+
+    project = Project(
+        services=[],
+        targets=[Target(
+            name="target",
+            dependencies=[],
+            steps=[step]
+        )])
+    parameters = Parameters(dataconnection="runci.yml", targets=["target"], verbosity=0)
+
+    @patch('runci.engine.runner.docker_pull.DockerPullRunner._run_process')
+    def test_command_line_args(self, mock):
+        async def run():
+            runner = DockerPullRunner(lambda e: None, self.step.spec)
+            context = create_context(self.project, self.parameters)
+            await runner.run(context)
+
+        asyncio.run(run())
+        mock.assert_has_calls([
+            call('docker pull busybox'.split(' ')),
+            call('docker pull alpine'.split(' '))])
+
+    @patch('runci.engine.runner.docker_pull.DockerPullRunner.run')
+    def test_integration(self, mock):
+        context = create_context(self.project, self.parameters)
+        DependencyTree(context).run()
+        mock.assert_called_once()
+
+
+if __name__ == '__main__':
+    unittest.main()