Skip to content

Commit a3f0a37

Browse files
author
Jérémy James Toussaint
committed
ComputationService Tee unit test updated
1 parent 6345e49 commit a3f0a37

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

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

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@ TaskDescription getStubTaskDescription(boolean isTeeTask) {
5959
.maxExecutionTime(500000) // 5min
6060
.cmd("ls")
6161
.inputFiles(new ArrayList<>())
62+
.teePostComputeImage("registry/post-compute-app:tag")
6263
.build();
6364
}
6465

@@ -156,6 +157,7 @@ public void shouldComputeTeeTask() {
156157
when(sconeTeeService.createSconeSecureSession(any()))
157158
.thenReturn(awesomeSessionId);
158159
when(sconeTeeService.buildSconeDockerEnv(any(), any(), any())).thenReturn(stubSconeEnv);
160+
when(customDockerClient.pullImage(anyString(), anyString())).thenReturn(true);
159161
when(customDockerClient.execute(any()))
160162
.thenReturn(DockerExecutionResult.success("Computed successfully !", "containerName"))
161163
.thenReturn(DockerExecutionResult.success("Encrypted successfully !", "containerName"));

0 commit comments

Comments
 (0)