This is an automated email from the ASF dual-hosted git repository.
ofuks pushed a commit to branch develop
in repository https://gitbox.apache.org/repos/asf/incubator-dlab.git
The following commit(s) were added to refs/heads/develop by this push:
new ab8eb5e Created test for ProjectServiceImpl
ab8eb5e is described below
commit ab8eb5ee4d0d90352b49ace455245015373d37c2
Author: Oleh Fuks <olegfuk...@gmail.com>
AuthorDate: Tue Aug 4 14:43:29 2020 +0300
Created test for ProjectServiceImpl
---
pom.xml | 5 +-
...rovisioningServiceApplicationConfiguration.java | 1 +
.../backendapi/{ => conf}/CloudConfiguration.java | 2 +-
.../backendapi/core/commands/CommandBuilder.java | 2 +-
.../epam/dlab/backendapi/dao/ProjectDAOImpl.java | 9 +-
.../service/impl/ProjectServiceImpl.java | 3 +
.../backendapi/service/ProjectServiceImplTest.java | 427 +++++++++++++++++++++
7 files changed, 443 insertions(+), 6 deletions(-)
diff --git a/pom.xml b/pom.xml
index c327cfc..47645ff 100644
--- a/pom.xml
+++ b/pom.xml
@@ -65,6 +65,8 @@
**/*Exception.java,
**/*DAO.java,
**/*DAOImpl.java,
+ **/*Configuration.java,
+ **/*ApplicationConfiguration.java,
src/main/java/com/epam/dlab/dto/**,
src/main/java/com/epam/dlab/backendapi/interceptor/**,
src/main/java/com/epam/dlab/backendapi/auth/**,
@@ -80,7 +82,8 @@
src/main/java/com/epam/dlab/backendapi/validation/**,
src/main/java/com/epam/dlab/process/**,
src/main/java/com/epam/dlab/backendapi/modules/**,
- src/main/java/com/epam/dlab/backendapi/validation/**
+ src/main/java/com/epam/dlab/backendapi/validation/**,
+ src/main/java/com/epam/dlab/backendapi/schedulers/**
</sonar.coverage.exclusions>
</properties>
</profile>
diff --git
a/services/provisioning-service/src/main/java/com/epam/dlab/backendapi/ProvisioningServiceApplicationConfiguration.java
b/services/provisioning-service/src/main/java/com/epam/dlab/backendapi/ProvisioningServiceApplicationConfiguration.java
index 1025ad6..4edc8af 100644
---
a/services/provisioning-service/src/main/java/com/epam/dlab/backendapi/ProvisioningServiceApplicationConfiguration.java
+++
b/services/provisioning-service/src/main/java/com/epam/dlab/backendapi/ProvisioningServiceApplicationConfiguration.java
@@ -20,6 +20,7 @@
package com.epam.dlab.backendapi;
import com.epam.dlab.ServiceConfiguration;
+import com.epam.dlab.backendapi.conf.CloudConfiguration;
import com.epam.dlab.backendapi.core.Directories;
import
com.epam.dlab.backendapi.validation.ProvisioningServiceCloudConfigurationSequenceProvider;
import com.epam.dlab.validation.AwsValidation;
diff --git
a/services/provisioning-service/src/main/java/com/epam/dlab/backendapi/CloudConfiguration.java
b/services/provisioning-service/src/main/java/com/epam/dlab/backendapi/conf/CloudConfiguration.java
similarity index 98%
rename from
services/provisioning-service/src/main/java/com/epam/dlab/backendapi/CloudConfiguration.java
rename to
services/provisioning-service/src/main/java/com/epam/dlab/backendapi/conf/CloudConfiguration.java
index 9d61210..c25624c 100644
---
a/services/provisioning-service/src/main/java/com/epam/dlab/backendapi/CloudConfiguration.java
+++
b/services/provisioning-service/src/main/java/com/epam/dlab/backendapi/conf/CloudConfiguration.java
@@ -17,7 +17,7 @@
* under the License.
*/
-package com.epam.dlab.backendapi;
+package com.epam.dlab.backendapi.conf;
import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.Data;
diff --git
a/services/provisioning-service/src/main/java/com/epam/dlab/backendapi/core/commands/CommandBuilder.java
b/services/provisioning-service/src/main/java/com/epam/dlab/backendapi/core/commands/CommandBuilder.java
index a212d0a..f99eca4 100644
---
a/services/provisioning-service/src/main/java/com/epam/dlab/backendapi/core/commands/CommandBuilder.java
+++
b/services/provisioning-service/src/main/java/com/epam/dlab/backendapi/core/commands/CommandBuilder.java
@@ -19,8 +19,8 @@
package com.epam.dlab.backendapi.core.commands;
-import com.epam.dlab.backendapi.CloudConfiguration;
import com.epam.dlab.backendapi.ProvisioningServiceApplicationConfiguration;
+import com.epam.dlab.backendapi.conf.CloudConfiguration;
import com.epam.dlab.cloud.CloudProvider;
import com.epam.dlab.dto.ResourceBaseDTO;
import com.epam.dlab.dto.aws.AwsCloudSettings;
diff --git
a/services/self-service/src/main/java/com/epam/dlab/backendapi/dao/ProjectDAOImpl.java
b/services/self-service/src/main/java/com/epam/dlab/backendapi/dao/ProjectDAOImpl.java
index 99073d1..f08e3a3 100644
---
a/services/self-service/src/main/java/com/epam/dlab/backendapi/dao/ProjectDAOImpl.java
+++
b/services/self-service/src/main/java/com/epam/dlab/backendapi/dao/ProjectDAOImpl.java
@@ -75,15 +75,18 @@ public class ProjectDAOImpl extends BaseDAO implements
ProjectDAO {
@Override
public List<ProjectDTO>
getProjectsWithEndpointStatusNotIn(UserInstanceStatus... statuses) {
final List<String> statusList =
-
Arrays.stream(statuses).map(UserInstanceStatus::name).collect(Collectors.toList());
+ Arrays.stream(statuses)
+ .map(UserInstanceStatus::name)
+ .collect(Collectors.toList());
return find(PROJECTS_COLLECTION, not(in(ENDPOINT_STATUS_FIELD,
statusList)), ProjectDTO.class);
}
@Override
public List<ProjectDTO> getUserProjects(UserInfo userInfo, boolean
active) {
- final Set<String> groups =
Stream.concat(userGroupDao.getUserGroups(userInfo.getName()).stream(),
- userInfo.getRoles().stream())
+ Stream<String> userGroups =
userGroupDao.getUserGroups(userInfo.getName()).stream();
+ Stream<String> roles = userInfo.getRoles().stream();
+ final Set<String> groups = Stream.concat(userGroups, roles)
.collect(Collectors.toSet());
return find(PROJECTS_COLLECTION, userProjectCondition(groups,
active), ProjectDTO.class);
}
diff --git
a/services/self-service/src/main/java/com/epam/dlab/backendapi/service/impl/ProjectServiceImpl.java
b/services/self-service/src/main/java/com/epam/dlab/backendapi/service/impl/ProjectServiceImpl.java
index 78cf324..7ee90ba 100644
---
a/services/self-service/src/main/java/com/epam/dlab/backendapi/service/impl/ProjectServiceImpl.java
+++
b/services/self-service/src/main/java/com/epam/dlab/backendapi/service/impl/ProjectServiceImpl.java
@@ -341,6 +341,9 @@ public class ProjectServiceImpl implements ProjectService {
}
private String getUpdateBudgetAudit(ProjectDTO p) {
+ if (!configuration.isAuditEnabled()) {
+ return null;
+ }
Integer value =
Optional.ofNullable(get(p.getName()).getBudget())
.map(BudgetDTO::getValue)
.orElse(null);
diff --git
a/services/self-service/src/test/java/com/epam/dlab/backendapi/service/ProjectServiceImplTest.java
b/services/self-service/src/test/java/com/epam/dlab/backendapi/service/ProjectServiceImplTest.java
new file mode 100644
index 0000000..e9367ff
--- /dev/null
+++
b/services/self-service/src/test/java/com/epam/dlab/backendapi/service/ProjectServiceImplTest.java
@@ -0,0 +1,427 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package com.epam.dlab.backendapi.service;
+
+import com.epam.dlab.auth.UserInfo;
+import com.epam.dlab.backendapi.conf.SelfServiceApplicationConfiguration;
+import com.epam.dlab.backendapi.dao.ExploratoryDAO;
+import com.epam.dlab.backendapi.dao.ProjectDAO;
+import com.epam.dlab.backendapi.dao.UserGroupDAO;
+import com.epam.dlab.backendapi.domain.EndpointDTO;
+import com.epam.dlab.backendapi.domain.ProjectDTO;
+import com.epam.dlab.backendapi.domain.ProjectEndpointDTO;
+import com.epam.dlab.backendapi.domain.RequestId;
+import com.epam.dlab.backendapi.domain.UpdateProjectBudgetDTO;
+import com.epam.dlab.backendapi.resources.TestBase;
+import com.epam.dlab.backendapi.service.impl.ProjectServiceImpl;
+import com.epam.dlab.backendapi.util.RequestBuilder;
+import com.epam.dlab.dto.UserInstanceStatus;
+import com.epam.dlab.dto.project.ProjectActionDTO;
+import com.epam.dlab.dto.project.ProjectCreateDTO;
+import com.epam.dlab.exceptions.DlabException;
+import com.epam.dlab.exceptions.ResourceConflictException;
+import com.epam.dlab.exceptions.ResourceNotFoundException;
+import com.epam.dlab.rest.client.RESTService;
+import com.google.common.collect.Sets;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.runners.MockitoJUnitRunner;
+
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.List;
+import java.util.Optional;
+import java.util.Set;
+
+import static org.junit.Assert.assertEquals;
+import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.anyBoolean;
+import static org.mockito.Matchers.anyListOf;
+import static org.mockito.Matchers.anySet;
+import static org.mockito.Matchers.anyString;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.verifyNoMoreInteractions;
+import static org.mockito.Mockito.when;
+
+@RunWith(MockitoJUnitRunner.class)
+public class ProjectServiceImplTest extends TestBase {
+ private static final String CREATE_PRJ_API =
"infrastructure/project/create";
+ private static final String TERMINATE_PRJ_API =
"infrastructure/project/terminate";
+ private static final String START_PRJ_API =
"infrastructure/project/start";
+ private static final String STOP_PRJ_API =
"infrastructure/project/stop";
+
+ private static final String NAME1 = "name1";
+ private static final String NAME2 = "name2";
+ private static final String GROUP1 = "group1";
+ private static final String GROUP2 = "group2";
+ private static final String UUID = "uuid";
+
+ private static final List<UserInstanceStatus> notebookStatuses =
Arrays.asList(
+ UserInstanceStatus.CREATING,
UserInstanceStatus.STARTING, UserInstanceStatus.CREATING_IMAGE,
+ UserInstanceStatus.CONFIGURING,
UserInstanceStatus.RECONFIGURING, UserInstanceStatus.STOPPING,
+ UserInstanceStatus.TERMINATING);
+ private static final UserInstanceStatus[] computeStatuses =
{UserInstanceStatus.CREATING, UserInstanceStatus.CONFIGURING,
UserInstanceStatus.STARTING,
+ UserInstanceStatus.RECONFIGURING,
UserInstanceStatus.CREATING_IMAGE, UserInstanceStatus.STOPPING,
+ UserInstanceStatus.TERMINATING};
+
+ @Mock
+ private ProjectDAO projectDAO;
+ @Mock
+ private EndpointService endpointService;
+ @Mock
+ private RESTService provisioningService;
+ @Mock
+ private RequestBuilder requestBuilder;
+ @Mock
+ private RequestId requestId;
+ @Mock
+ private ExploratoryService exploratoryService;
+ @Mock
+ private ExploratoryDAO exploratoryDAO;
+ @Mock
+ private UserGroupDAO userGroupDao;
+ @Mock
+ private SelfServiceApplicationConfiguration configuration;
+ @InjectMocks
+ private ProjectServiceImpl projectService;
+
+ @Test
+ public void getProjects() {
+ List<ProjectDTO> projectsMock = getProjectDTOs();
+ when(projectDAO.getProjects()).thenReturn(projectsMock);
+
+ List<ProjectDTO> projects = projectService.getProjects();
+
+ assertEquals(projects, projectsMock);
+ verify(projectDAO).getProjects();
+ verifyNoMoreInteractions(projectDAO);
+ }
+
+ @Test
+ public void testGetProjects() {
+ List<ProjectDTO> projectsMock = getProjectDTOs();
+ when(projectDAO.getProjects()).thenReturn(projectsMock);
+
+ List<ProjectDTO> projects =
projectService.getProjects(getUserInfo());
+
+ assertEquals(projects, projectsMock);
+ verify(projectDAO).getProjects();
+ verifyNoMoreInteractions(projectDAO);
+ }
+
+ @Test
+ public void getUserProjects() {
+ List<ProjectDTO> projectsMock =
Collections.singletonList(getProjectCreatingDTO());
+ when(projectDAO.getUserProjects(any(UserInfo.class),
anyBoolean())).thenReturn(projectsMock);
+
+ List<ProjectDTO> projects =
projectService.getUserProjects(getUserInfo(), Boolean.TRUE);
+
+ assertEquals(projectsMock, projects);
+ verify(projectDAO).getUserProjects(getUserInfo(), Boolean.TRUE);
+ verifyNoMoreInteractions(projectDAO);
+ }
+
+ @Test
+ public void getProjectsByEndpoint() {
+ List<ProjectDTO> projectsMock =
Collections.singletonList(getProjectCreatingDTO());
+
when(projectDAO.getProjectsByEndpoint(anyString())).thenReturn(projectsMock);
+
+ List<ProjectDTO> projects =
projectService.getProjectsByEndpoint(ENDPOINT_NAME);
+
+ assertEquals(projectsMock, projects);
+ verify(projectDAO).getProjectsByEndpoint(ENDPOINT_NAME);
+ verifyNoMoreInteractions(projectDAO);
+ }
+
+ @Test
+ public void create() {
+ when(projectDAO.get(anyString())).thenReturn(Optional.empty());
+
when(endpointService.get(anyString())).thenReturn(getEndpointDTO());
+ when(provisioningService.post(anyString(), anyString(), any(),
any())).thenReturn(UUID);
+ when(requestBuilder.newProjectCreate(any(UserInfo.class),
any(ProjectDTO.class), any(EndpointDTO.class))).thenReturn(newProjectCreate());
+
+ ProjectDTO projectDTO = getProjectCreatingDTO();
+ projectService.create(getUserInfo(), projectDTO,
projectDTO.getName());
+
+ verify(projectDAO).get(NAME1);
+ verify(projectDAO).create(projectDTO);
+ verify(endpointService).get(ENDPOINT_NAME);
+ verify(requestBuilder).newProjectCreate(getUserInfo(),
projectDTO, getEndpointDTO());
+ verify(provisioningService).post(ENDPOINT_URL + CREATE_PRJ_API,
TOKEN, newProjectCreate(), String.class);
+ verify(requestId).put(USER.toLowerCase(), UUID);
+ verifyNoMoreInteractions(projectDAO, endpointService,
provisioningService, requestBuilder);
+ }
+
+ @Test(expected = ResourceConflictException.class)
+ public void createWithException() {
+
when(projectDAO.get(anyString())).thenReturn(Optional.of(getProjectCreatingDTO()));
+
+ ProjectDTO projectDTO = getProjectCreatingDTO();
+ projectService.create(getUserInfo(), projectDTO,
projectDTO.getName());
+
+ verify(projectDAO).get(NAME1);
+ verifyNoMoreInteractions(projectDAO);
+ }
+
+ @Test
+ public void get() {
+ ProjectDTO projectMock = getProjectCreatingDTO();
+
when(projectDAO.get(anyString())).thenReturn(Optional.of(projectMock));
+
+ ProjectDTO project = projectService.get(NAME1);
+
+ assertEquals(projectMock, project);
+ verify(projectDAO).get(NAME1);
+ verifyNoMoreInteractions(projectDAO);
+ }
+
+ @Test(expected = ResourceNotFoundException.class)
+ public void getWithException() {
+ when(projectDAO.get(anyString())).thenReturn(Optional.empty());
+
+ projectService.get(NAME1);
+
+ verify(projectDAO).get(NAME1);
+ verifyNoMoreInteractions(projectDAO);
+ }
+
+ @Test
+ public void terminateEndpoint() {
+
when(endpointService.get(anyString())).thenReturn(getEndpointDTO());
+ when(provisioningService.post(anyString(), anyString(), any(),
any())).thenReturn(UUID);
+ when(requestBuilder.newProjectAction(any(UserInfo.class),
anyString(), any(EndpointDTO.class))).thenReturn(getProjectActionDTO());
+
+ projectService.terminateEndpoint(getUserInfo(), ENDPOINT_NAME,
NAME1);
+
+
verify(exploratoryService).updateProjectExploratoryStatuses(getUserInfo(),
NAME1, ENDPOINT_NAME, UserInstanceStatus.TERMINATING);
+ verify(projectDAO).updateEdgeStatus(NAME1, ENDPOINT_NAME,
UserInstanceStatus.TERMINATING);
+ verify(endpointService).get(ENDPOINT_NAME);
+ verify(provisioningService).post(ENDPOINT_URL +
TERMINATE_PRJ_API, TOKEN, getProjectActionDTO(), String.class);
+ verify(requestBuilder).newProjectAction(getUserInfo(), NAME1,
getEndpointDTO());
+ verify(requestId).put(USER.toLowerCase(), UUID);
+ verifyNoMoreInteractions(projectDAO, endpointService,
provisioningService, requestBuilder, exploratoryService);
+ }
+
+ @Test
+ public void terminateEndpointWithException() {
+
when(endpointService.get(anyString())).thenReturn(getEndpointDTO());
+ when(requestBuilder.newProjectAction(any(UserInfo.class),
anyString(), any(EndpointDTO.class))).thenReturn(getProjectActionDTO());
+ when(provisioningService.post(anyString(), anyString(), any(),
any())).thenThrow(new DlabException("Exception message"));
+
+ projectService.terminateEndpoint(getUserInfo(), ENDPOINT_NAME,
NAME1);
+
+
verify(exploratoryService).updateProjectExploratoryStatuses(getUserInfo(),
NAME1, ENDPOINT_NAME, UserInstanceStatus.TERMINATING);
+ verify(projectDAO).updateEdgeStatus(NAME1, ENDPOINT_NAME,
UserInstanceStatus.TERMINATING);
+ verify(projectDAO).updateStatus(NAME1,
ProjectDTO.Status.FAILED);
+ verify(endpointService).get(ENDPOINT_NAME);
+ verify(provisioningService).post(ENDPOINT_URL +
TERMINATE_PRJ_API, TOKEN, getProjectActionDTO(), String.class);
+ verify(requestBuilder).newProjectAction(getUserInfo(), NAME1,
getEndpointDTO());
+ verifyNoMoreInteractions(projectDAO, endpointService,
provisioningService, requestBuilder, exploratoryService);
+ }
+
+ @Test
+ public void testTerminateEndpoint() {
+
when(endpointService.get(anyString())).thenReturn(getEndpointDTO());
+ when(provisioningService.post(anyString(), anyString(), any(),
any())).thenReturn(UUID);
+ when(requestBuilder.newProjectAction(any(UserInfo.class),
anyString(), any(EndpointDTO.class))).thenReturn(getProjectActionDTO());
+
when(projectDAO.get(anyString())).thenReturn(Optional.of(getProjectRunningDTO()));
+
when(exploratoryDAO.fetchProjectEndpointExploratoriesWhereStatusIn(anyString(),
anyListOf(String.class), anyListOf(UserInstanceStatus.class)))
+ .thenReturn(Collections.emptyList());
+
+ projectService.terminateEndpoint(getUserInfo(),
Collections.singletonList(ENDPOINT_NAME), NAME1);
+
+ verify(projectDAO).get(NAME1);
+
verify(exploratoryDAO).fetchProjectEndpointExploratoriesWhereStatusIn(NAME1,
Collections.singletonList(ENDPOINT_NAME), notebookStatuses, computeStatuses);
+
verify(exploratoryService).updateProjectExploratoryStatuses(getUserInfo(),
NAME1, ENDPOINT_NAME, UserInstanceStatus.TERMINATING);
+ verify(projectDAO).updateEdgeStatus(NAME1, ENDPOINT_NAME,
UserInstanceStatus.TERMINATING);
+ verify(endpointService).get(ENDPOINT_NAME);
+ verify(provisioningService).post(ENDPOINT_URL +
TERMINATE_PRJ_API, TOKEN, getProjectActionDTO(), String.class);
+ verify(requestBuilder).newProjectAction(getUserInfo(), NAME1,
getEndpointDTO());
+ verify(requestId).put(USER.toLowerCase(), UUID);
+ verifyNoMoreInteractions(projectDAO, endpointService,
provisioningService, requestBuilder, exploratoryDAO);
+ }
+
+ @Test(expected = ResourceConflictException.class)
+ public void testTerminateEndpointWithException1() {
+
when(projectDAO.get(anyString())).thenReturn(Optional.of(getProjectCreatingDTO()));
+
+ projectService.terminateEndpoint(getUserInfo(),
Collections.singletonList(ENDPOINT_NAME), NAME1);
+
+ verify(projectDAO).get(NAME1);
+ verifyNoMoreInteractions(projectDAO);
+ }
+
+ @Test
+ public void start() {
+
when(endpointService.get(anyString())).thenReturn(getEndpointDTO());
+ when(provisioningService.post(anyString(), anyString(), any(),
any())).thenReturn(UUID);
+ when(requestBuilder.newProjectAction(any(UserInfo.class),
anyString(), any(EndpointDTO.class))).thenReturn(getProjectActionDTO());
+
+ projectService.start(getUserInfo(), ENDPOINT_NAME, NAME1);
+
+ verify(projectDAO).updateEdgeStatus(NAME1, ENDPOINT_NAME,
UserInstanceStatus.STARTING);
+ verify(endpointService).get(ENDPOINT_NAME);
+ verify(provisioningService).post(ENDPOINT_URL + START_PRJ_API,
TOKEN, getProjectActionDTO(), String.class);
+ verify(requestBuilder).newProjectAction(getUserInfo(), NAME1,
getEndpointDTO());
+ verify(requestId).put(USER.toLowerCase(), UUID);
+ verifyNoMoreInteractions(projectDAO, endpointService,
provisioningService, requestBuilder, requestId);
+ }
+
+ @Test
+ public void testStart() {
+
when(endpointService.get(anyString())).thenReturn(getEndpointDTO());
+ when(provisioningService.post(anyString(), anyString(), any(),
any())).thenReturn(UUID);
+ when(requestBuilder.newProjectAction(any(UserInfo.class),
anyString(), any(EndpointDTO.class))).thenReturn(getProjectActionDTO());
+
+ projectService.start(getUserInfo(),
Collections.singletonList(ENDPOINT_NAME), NAME1);
+
+ verify(projectDAO).updateEdgeStatus(NAME1, ENDPOINT_NAME,
UserInstanceStatus.STARTING);
+ verify(endpointService).get(ENDPOINT_NAME);
+ verify(provisioningService).post(ENDPOINT_URL + START_PRJ_API,
TOKEN, getProjectActionDTO(), String.class);
+ verify(requestBuilder).newProjectAction(getUserInfo(), NAME1,
getEndpointDTO());
+ verify(requestId).put(USER.toLowerCase(), UUID);
+ verifyNoMoreInteractions(projectDAO, endpointService,
provisioningService, requestBuilder, requestId);
+ }
+
+ @Test
+ public void stop() {
+
when(endpointService.get(anyString())).thenReturn(getEndpointDTO());
+ when(provisioningService.post(anyString(), anyString(), any(),
any())).thenReturn(UUID);
+ when(requestBuilder.newProjectAction(any(UserInfo.class),
anyString(), any(EndpointDTO.class))).thenReturn(getProjectActionDTO());
+
+ projectService.stop(getUserInfo(), ENDPOINT_NAME, NAME1, null);
+
+ verify(projectDAO).updateEdgeStatus(NAME1, ENDPOINT_NAME,
UserInstanceStatus.STOPPING);
+ verify(endpointService).get(ENDPOINT_NAME);
+ verify(provisioningService).post(ENDPOINT_URL + STOP_PRJ_API,
TOKEN, getProjectActionDTO(), String.class);
+ verify(requestBuilder).newProjectAction(getUserInfo(), NAME1,
getEndpointDTO());
+ verify(requestId).put(USER.toLowerCase(), UUID);
+ verifyNoMoreInteractions(projectDAO, endpointService,
provisioningService, requestBuilder, requestId);
+ }
+
+ @Test
+ public void stopWithResources() {
+
when(endpointService.get(anyString())).thenReturn(getEndpointDTO());
+ when(provisioningService.post(anyString(), anyString(), any(),
any())).thenReturn(UUID);
+ when(requestBuilder.newProjectAction(any(UserInfo.class),
anyString(), any(EndpointDTO.class))).thenReturn(getProjectActionDTO());
+
when(projectDAO.get(anyString())).thenReturn(Optional.of(getProjectRunningDTO()));
+
when(exploratoryDAO.fetchProjectEndpointExploratoriesWhereStatusIn(anyString(),
anyListOf(String.class), anyListOf(UserInstanceStatus.class)))
+ .thenReturn(Collections.emptyList());
+
+ projectService.stopWithResources(getUserInfo(),
Collections.singletonList(ENDPOINT_NAME), NAME1);
+
+ verify(projectDAO).get(NAME1);
+
verify(exploratoryDAO).fetchProjectEndpointExploratoriesWhereStatusIn(NAME1,
Collections.singletonList(ENDPOINT_NAME), notebookStatuses, computeStatuses);
+ verify(projectDAO).updateEdgeStatus(NAME1, ENDPOINT_NAME,
UserInstanceStatus.STOPPING);
+ verify(endpointService).get(ENDPOINT_NAME);
+ verify(provisioningService).post(ENDPOINT_URL + STOP_PRJ_API,
TOKEN, getProjectActionDTO(), String.class);
+ verify(requestBuilder).newProjectAction(getUserInfo(), NAME1,
getEndpointDTO());
+ verify(requestId).put(USER.toLowerCase(), UUID);
+ verifyNoMoreInteractions(projectDAO, endpointService,
provisioningService, requestBuilder, requestId);
+ }
+
+ @Test
+ public void update() {
+ }
+
+ @Test
+ public void updateBudget() {
+ projectService.updateBudget(getUserInfo(),
Collections.singletonList(getUpdateProjectBudgetDTO()));
+
+ verify(projectDAO).updateBudget(NAME1, 10, true);
+ verifyNoMoreInteractions(projectDAO);
+ }
+
+ @Test
+ public void isAnyProjectAssigned() {
+
when(userGroupDao.getUserGroups(anyString())).thenReturn(Sets.newHashSet(GROUP1));
+
when(projectDAO.isAnyProjectAssigned(anySet())).thenReturn(Boolean.TRUE);
+
+ final boolean anyProjectAssigned =
projectService.isAnyProjectAssigned(getUserInfo());
+
+ assertEquals(anyProjectAssigned, Boolean.TRUE);
+ verify(userGroupDao).getUserGroups(USER.toLowerCase());
+
verify(projectDAO).isAnyProjectAssigned(Sets.newHashSet(GROUP1));
+ verifyNoMoreInteractions(userGroupDao, projectDAO);
+ }
+
+ @Test
+ public void checkExploratoriesAndComputationalProgress() {
+
when(exploratoryDAO.fetchProjectEndpointExploratoriesWhereStatusIn(anyString(),
anyListOf(String.class), anyListOf(UserInstanceStatus.class)))
+ .thenReturn(Collections.emptyList());
+
+ final boolean b =
projectService.checkExploratoriesAndComputationalProgress(NAME1,
Collections.singletonList(ENDPOINT_NAME));
+
+ assertEquals(b, Boolean.TRUE);
+
verify(exploratoryDAO).fetchProjectEndpointExploratoriesWhereStatusIn(NAME1,
Collections.singletonList(ENDPOINT_NAME), notebookStatuses, computeStatuses);
+ verifyNoMoreInteractions(exploratoryDAO);
+ }
+
+ private List<ProjectDTO> getProjectDTOs() {
+ ProjectDTO project1 = ProjectDTO.builder()
+ .name(NAME1)
+ .groups(getGroup(GROUP1))
+ .build();
+ ProjectDTO project2 = ProjectDTO.builder()
+ .name(NAME2)
+ .groups(getGroup(GROUP2))
+ .build();
+ return Arrays.asList(project1, project2);
+ }
+
+ private ProjectDTO getProjectCreatingDTO() {
+ ProjectEndpointDTO projectEndpointDTO = new
ProjectEndpointDTO(ENDPOINT_NAME, UserInstanceStatus.CREATING, null);
+ return ProjectDTO.builder()
+ .name(NAME1)
+ .groups(getGroup(GROUP1))
+
.endpoints(Collections.singletonList(projectEndpointDTO))
+ .build();
+ }
+
+ private ProjectDTO getProjectRunningDTO() {
+ ProjectEndpointDTO projectEndpointDTO = new
ProjectEndpointDTO(ENDPOINT_NAME, UserInstanceStatus.RUNNING, null);
+ return ProjectDTO.builder()
+ .name(NAME1)
+ .groups(getGroup(GROUP1))
+
.endpoints(Collections.singletonList(projectEndpointDTO))
+ .build();
+ }
+
+ private Set<String> getGroup(String group) {
+ return Collections.singleton(group);
+ }
+
+ private ProjectCreateDTO newProjectCreate() {
+ return ProjectCreateDTO.builder()
+ .name(NAME1)
+ .endpoint(ENDPOINT_NAME)
+ .build();
+ }
+
+ private ProjectActionDTO getProjectActionDTO() {
+ return new ProjectActionDTO(NAME1, ENDPOINT_NAME);
+ }
+
+ private UpdateProjectBudgetDTO getUpdateProjectBudgetDTO() {
+ return new UpdateProjectBudgetDTO(NAME1, 10, true);
+ }
+}
\ No newline at end of file
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@dlab.apache.org
For additional commands, e-mail: commits-h...@dlab.apache.org
