|
16 | 16 | // under the License. |
17 | 17 | package com.cloud.api.query.dao; |
18 | 18 |
|
19 | | -import com.cloud.api.query.vo.UserVmJoinVO; |
20 | | -import com.cloud.storage.Storage; |
21 | | -import com.cloud.storage.VnfTemplateDetailVO; |
22 | | -import com.cloud.storage.VnfTemplateNicVO; |
23 | | -import com.cloud.storage.dao.VnfTemplateDetailsDao; |
24 | | -import com.cloud.storage.dao.VnfTemplateNicDao; |
25 | | -import com.cloud.user.Account; |
26 | | -import com.cloud.user.AccountManager; |
27 | | -import com.cloud.user.UserStatisticsVO; |
28 | | -import com.cloud.user.dao.UserDao; |
29 | | -import com.cloud.user.dao.UserStatisticsDao; |
30 | | -import com.cloud.utils.db.SearchBuilder; |
31 | | -import com.cloud.utils.db.SearchCriteria; |
32 | | -import com.cloud.vm.dao.UserVmDetailsDao; |
| 19 | +import static org.mockito.ArgumentMatchers.nullable; |
| 20 | + |
| 21 | +import java.util.Arrays; |
| 22 | +import java.util.EnumSet; |
| 23 | + |
33 | 24 | import org.apache.cloudstack.annotation.dao.AnnotationDao; |
34 | 25 | import org.apache.cloudstack.api.ApiConstants; |
35 | 26 | import org.apache.cloudstack.api.ResponseObject; |
|
44 | 35 | import org.mockito.Mockito; |
45 | 36 | import org.mockito.junit.MockitoJUnitRunner; |
46 | 37 |
|
47 | | -import java.util.Arrays; |
48 | | -import java.util.EnumSet; |
49 | | - |
50 | | -import static org.mockito.ArgumentMatchers.nullable; |
| 38 | +import com.cloud.api.query.vo.UserVmJoinVO; |
| 39 | +import com.cloud.storage.Storage; |
| 40 | +import com.cloud.storage.VnfTemplateDetailVO; |
| 41 | +import com.cloud.storage.VnfTemplateNicVO; |
| 42 | +import com.cloud.storage.dao.VnfTemplateDetailsDao; |
| 43 | +import com.cloud.storage.dao.VnfTemplateNicDao; |
| 44 | +import com.cloud.user.Account; |
| 45 | +import com.cloud.user.AccountManager; |
| 46 | +import com.cloud.user.UserStatisticsVO; |
| 47 | +import com.cloud.user.dao.UserDao; |
| 48 | +import com.cloud.user.dao.UserStatisticsDao; |
| 49 | +import com.cloud.utils.db.SearchBuilder; |
| 50 | +import com.cloud.utils.db.SearchCriteria; |
| 51 | +import com.cloud.vm.dao.UserVmDetailsDao; |
51 | 52 |
|
52 | 53 | @RunWith(MockitoJUnitRunner.class) |
53 | 54 | public class UserVmJoinDaoImplTest extends GenericDaoBaseWithTagInformationBaseTest<UserVmJoinVO, UserVmResponse> { |
@@ -109,6 +110,7 @@ private void prepareNewUserVmResponseForVnfAppliance() { |
109 | 110 | Mockito.when(userVmMock.getId()).thenReturn(vmId); |
110 | 111 | Mockito.when(userVmMock.getTemplateId()).thenReturn(templateId); |
111 | 112 | Mockito.when(userVmMock.getTemplateType()).thenReturn(Storage.TemplateType.VNF); |
| 113 | + Mockito.when(userVmMock.getTemplateFormat()).thenReturn(Storage.ImageFormat.OVA); |
112 | 114 |
|
113 | 115 | Mockito.when(caller.getId()).thenReturn(2L); |
114 | 116 | Mockito.when(accountMgr.isRootAdmin(nullable(Long.class))).thenReturn(true); |
|
0 commit comments