@@ -73,7 +73,6 @@ class PreComputeServiceTests {
7373 .datasetUri (datasetUri )
7474 .datasetName ("datasetName" )
7575 .datasetChecksum ("datasetChecksum" )
76- .teePostComputeImage ("teePostComputeImage" )
7776 .teeFramework (TeeFramework .SCONE )
7877 .appEnclaveConfiguration (TeeEnclaveConfiguration .builder ()
7978 .fingerprint ("01ba4719c80b6fe911b091a7c05124b64eeece964e09c058ef8f9805daca546b" )
@@ -169,8 +168,6 @@ void shouldRunTeePreComputeAndPrepareInputDataWhenDatasetAndInputFilesArePresent
169168 void shouldRunTeePreComputeAndPrepareInputDataWhenOnlyDatasetIsPresent () throws TeeSessionGenerationException {
170169 final TaskDescription taskDescription = taskDescriptionBuilder .build ();
171170
172- when (dockerClientInstanceMock .pullImage (taskDescription .getTeePostComputeImage ()))
173- .thenReturn (true );
174171 when (smsService .createTeeSession (workerpoolAuthorization )).thenReturn (secureSession );
175172 when (preComputeProperties .getImage ()).thenReturn (PRE_COMPUTE_IMAGE );
176173 when (preComputeProperties .getHeapSizeInBytes ()).thenReturn (PRE_COMPUTE_HEAP );
@@ -211,8 +208,6 @@ void shouldRunTeePreComputeAndPrepareInputDataWhenOnlyInputFilesArePresent() thr
211208 .inputFiles (List .of ("input-file1" ))
212209 .build ();
213210
214- when (dockerClientInstanceMock .pullImage (taskDescription .getTeePostComputeImage ()))
215- .thenReturn (true );
216211 when (smsService .createTeeSession (workerpoolAuthorization )).thenReturn (secureSession );
217212 when (preComputeProperties .getImage ()).thenReturn (PRE_COMPUTE_IMAGE );
218213 when (preComputeProperties .getHeapSizeInBytes ()).thenReturn (PRE_COMPUTE_HEAP );
@@ -274,9 +269,7 @@ void shouldFailToRunTeePreComputeSinceTooHighComputeHeapSize() throws TeeSession
274269 @ Test
275270 void shouldFailToRunTeePreComputeSinceCantCreateTeeSession () throws TeeSessionGenerationException {
276271 final TaskDescription taskDescription = taskDescriptionBuilder .build ();
277- when (dockerClientInstanceMock
278- .pullImage (taskDescription .getTeePostComputeImage ()))
279- .thenReturn (true );
272+
280273 when (smsService .createTeeSession (workerpoolAuthorization )).thenReturn (secureSession );
281274
282275 Assertions .assertThat (preComputeService .runTeePreCompute (taskDescription , workerpoolAuthorization ).isSuccessful ())
@@ -288,9 +281,6 @@ void shouldFailToRunTeePreComputeSinceCantCreateTeeSession() throws TeeSessionGe
288281 @ Test
289282 void shouldNotRunTeePreComputeSinceDockerImageNotFoundLocally () throws TeeSessionGenerationException {
290283 final TaskDescription taskDescription = taskDescriptionBuilder .build ();
291- when (dockerClientInstanceMock
292- .pullImage (taskDescription .getTeePostComputeImage ()))
293- .thenReturn (true );
294284 when (smsService .createTeeSession (workerpoolAuthorization ))
295285 .thenReturn (secureSession );
296286 when (preComputeProperties .getImage ()).thenReturn (PRE_COMPUTE_IMAGE );
@@ -309,9 +299,6 @@ void shouldNotRunTeePreComputeSinceDockerImageNotFoundLocally() throws TeeSessio
309299 @ MethodSource ("shouldFailToRunTeePreComputeSinceDockerRunFailedArgs" )
310300 void shouldFailToRunTeePreComputeSinceDockerRunFailed (Map .Entry <Integer , ReplicateStatusCause > exitCodeKeyToExpectedCauseValue ) throws TeeSessionGenerationException {
311301 final TaskDescription taskDescription = taskDescriptionBuilder .build ();
312- when (dockerClientInstanceMock
313- .pullImage (taskDescription .getTeePostComputeImage ()))
314- .thenReturn (true );
315302 when (smsService .createTeeSession (workerpoolAuthorization ))
316303 .thenReturn (secureSession );
317304 when (preComputeProperties .getImage ()).thenReturn (PRE_COMPUTE_IMAGE );
@@ -351,9 +338,6 @@ private static Stream<Map.Entry<Integer, ReplicateStatusCause>> shouldFailToRunT
351338 @ Test
352339 void shouldFailToRunTeePreComputeSinceTimeout () throws TeeSessionGenerationException {
353340 final TaskDescription taskDescription = taskDescriptionBuilder .build ();
354- when (dockerClientInstanceMock
355- .pullImage (taskDescription .getTeePostComputeImage ()))
356- .thenReturn (true );
357341 when (smsService .createTeeSession (workerpoolAuthorization ))
358342 .thenReturn (secureSession );
359343 when (preComputeProperties .getImage ()).thenReturn (PRE_COMPUTE_IMAGE );
0 commit comments