@@ -201,10 +201,11 @@ public void tryCopyingTemplateToImageStoreTestReturnsTrueWhenTemplateExistsInAno
201201 }
202202
203203 @ Test
204- public void tryCopyingTemplateToImageStoreTestReturnsTrueWhenTemplateExistsInAnotherZone () throws StorageUnavailableException , ResourceAllocationException {
204+ public void tryCopyingTemplateToImageStoreTestReturnsTrueWhenTemplateExistsInAnotherZone () {
205205 Scope scopeMock = Mockito .mock (Scope .class );
206206 Mockito .doReturn (scopeMock ).when (destStoreMock ).getScope ();
207207 Mockito .doReturn (1L ).when (scopeMock ).getScopeId ();
208+ Mockito .doReturn (100L ).when (tmpltMock ).getId ();
208209 Mockito .doReturn (List .of (sourceStoreMock )).when (dataStoreManagerMock ).getImageStoresByZoneIds (1L );
209210 Mockito .doReturn (null ).when (templateService ).listTemplate (sourceStoreMock );
210211 Mockito .doReturn (List .of (1L , 2L )).when (_dcDao ).listAllIds ();
@@ -216,6 +217,10 @@ public void tryCopyingTemplateToImageStoreTestReturnsTrueWhenTemplateExistsInAno
216217 templatesInOtherZone .put (tmpltMock .getUniqueName (), tmpltPropMock );
217218 Mockito .doReturn (templatesInOtherZone ).when (templateService ).listTemplate (otherZoneStoreMock );
218219
220+ TemplateObject sourceTmplMock = Mockito .mock (TemplateObject .class );
221+ Mockito .doReturn (sourceTmplMock ).when (templateDataFactoryMock ).getTemplate (100L , otherZoneStoreMock );
222+ Mockito .doReturn ("/mnt/secondary/template.qcow2" ).when (sourceTmplMock ).getInstallPath ();
223+
219224 DataCenterVO dstZoneMock = Mockito .mock (DataCenterVO .class );
220225 Mockito .doReturn (dstZoneMock ).when (_dcDao ).findById (1L );
221226
@@ -229,6 +234,7 @@ public void tryCopyingTemplateToImageStoreTestReturnsFalseWhenDestinationZoneIsM
229234 Scope scopeMock = Mockito .mock (Scope .class );
230235 Mockito .doReturn (scopeMock ).when (destStoreMock ).getScope ();
231236 Mockito .doReturn (1L ).when (scopeMock ).getScopeId ();
237+ Mockito .doReturn (100L ).when (tmpltMock ).getId ();
232238 Mockito .doReturn (List .of (1L , 2L )).when (_dcDao ).listAllIds ();
233239 Mockito .doReturn (List .of ()).when (dataStoreManagerMock ).getImageStoresByZoneIds (1L );
234240
@@ -238,6 +244,10 @@ public void tryCopyingTemplateToImageStoreTestReturnsFalseWhenDestinationZoneIsM
238244 Map <String , TemplateProp > templates = new HashMap <>();
239245 templates .put (tmpltMock .getUniqueName (), tmpltPropMock );
240246 Mockito .doReturn (templates ).when (templateService ).listTemplate (otherZoneStoreMock );
247+
248+ TemplateObject sourceTmplMock = Mockito .mock (TemplateObject .class );
249+ Mockito .doReturn (sourceTmplMock ).when (templateDataFactoryMock ).getTemplate (100L , otherZoneStoreMock );
250+ Mockito .doReturn ("/mnt/secondary/template.qcow2" ).when (sourceTmplMock ).getInstallPath ();
241251 Mockito .doReturn (null ).when (_dcDao ).findById (1L );
242252
243253 boolean result = templateService .tryCopyingTemplateToImageStore (tmpltMock , destStoreMock );
@@ -250,23 +260,20 @@ public void tryCopyingTemplateToImageStoreTestReturnsTrueWhenCrossZoneCopyTaskIs
250260 Scope scopeMock = Mockito .mock (Scope .class );
251261 Mockito .doReturn (scopeMock ).when (destStoreMock ).getScope ();
252262 Mockito .doReturn (1L ).when (scopeMock ).getScopeId ();
253-
263+ Mockito . doReturn ( 100L ). when ( tmpltMock ). getId ();
254264 Mockito .doReturn (List .of (1L , 2L )).when (_dcDao ).listAllIds ();
255-
256- Mockito .doReturn (List .of ())
257- .when (dataStoreManagerMock )
258- .getImageStoresByZoneIds (1L );
265+ Mockito .doReturn (List .of ()).when (dataStoreManagerMock ).getImageStoresByZoneIds (1L );
259266
260267 DataStore otherZoneStoreMock = Mockito .mock (DataStore .class );
261- Mockito .doReturn (List .of (otherZoneStoreMock ))
262- .when (dataStoreManagerMock )
263- .getImageStoresByZoneIds (2L );
268+ Mockito .doReturn (List .of (otherZoneStoreMock )).when (dataStoreManagerMock ).getImageStoresByZoneIds (2L );
264269
265270 Map <String , TemplateProp > templates = new HashMap <>();
266271 templates .put (tmpltMock .getUniqueName (), tmpltPropMock );
267- Mockito .doReturn (templates )
268- .when (templateService )
269- .listTemplate (otherZoneStoreMock );
272+ Mockito .doReturn (templates ).when (templateService ).listTemplate (otherZoneStoreMock );
273+
274+ TemplateObject sourceTmplMock = Mockito .mock (TemplateObject .class );
275+ Mockito .doReturn (sourceTmplMock ).when (templateDataFactoryMock ).getTemplate (100L , otherZoneStoreMock );
276+ Mockito .doReturn ("/mnt/secondary/template.qcow2" ).when (sourceTmplMock ).getInstallPath ();
270277
271278 DataCenterVO dstZoneMock = Mockito .mock (DataCenterVO .class );
272279 Mockito .doReturn (dstZoneMock ).when (_dcDao ).findById (1L );
0 commit comments