Merge pull request #302 from iExecBlockchainComputing/feature/tee-post-compute-pull

Pull post compute image

Author: james-toussaint

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

@@ -153,6 +153,11 @@ 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 = sconeTeeService.createSconeSecureSession(contributionAuth);
 

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(sconeTeeService.createSconeSecureSession(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"));