Merge branch 'develop' into feature/sms-auth

Author: zguesmi

src/main/java/com/iexec/worker/docker/ComputationService.java

@@ -153,6 +153,12 @@ public boolean runTeeComputation(TaskDescription taskDescription,
         long maxExecutionTime = taskDescription.getMaxExecutionTime();
         String teePostComputeImage = taskDescription.getTeePostComputeImage();
 
+        if (!customDockerClient.pullImage(chainTaskId, teePostComputeImage)) {
+            String msg = "runTeeComputation failed (pullImage teePostComputeImage)";
+            log.error(msg + " [chainTaskId:{},teePostComputeImage:{}]", chainTaskId, teePostComputeImage);
+            return false;
+        }
+
         String secureSessionId = smsService.createTeeSession(contributionAuth);
         if (secureSessionId.isEmpty()) {
             log.error("Cannot compute TEE task without secure session [chainTaskId:{}]", chainTaskId);

src/test/java/com/iexec/worker/docker/ComputationServiceTests.java

@@ -59,6 +59,7 @@ TaskDescription getStubTaskDescription(boolean isTeeTask) {
                 .maxExecutionTime(500000) // 5min
                 .cmd("ls")
                 .inputFiles(new ArrayList<>())
+                .teePostComputeImage("registry/post-compute-app:tag")
                 .build();
     }
 
@@ -156,6 +157,7 @@ public void shouldComputeTeeTask() {
         when(smsService.createTeeSession(any()))
                 .thenReturn(awesomeSessionId);
         when(sconeTeeService.buildSconeDockerEnv(any(), any(), any())).thenReturn(stubSconeEnv);
+        when(customDockerClient.pullImage(anyString(), anyString())).thenReturn(true);
         when(customDockerClient.execute(any()))
                 .thenReturn(DockerExecutionResult.success("Computed successfully !", "containerName"))
                 .thenReturn(DockerExecutionResult.success("Encrypted successfully !", "containerName"));