36e828448db4a24480341d698566cb6815b31af2
[so.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  * ONAP - SO
4  * ================================================================================
5  * Copyright (C) 2020 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  * 
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  * 
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ============LICENSE_END=========================================================
19  */
20 package org.onap.so.bpmn.common.workflow.service;
21
22 import static org.junit.Assert.assertEquals;
23 import static org.mockito.Mockito.doReturn;
24 import java.util.ArrayList;
25 import java.util.List;
26 import org.camunda.bpm.engine.ProcessEngine;
27 import org.camunda.bpm.engine.RepositoryService;
28 import org.camunda.bpm.engine.repository.ProcessDefinition;
29 import org.camunda.bpm.engine.repository.ProcessDefinitionQuery;
30 import org.junit.Test;
31 import org.junit.runner.RunWith;
32 import org.mockito.InjectMocks;
33 import org.mockito.Mock;
34 import org.mockito.Spy;
35 import org.mockito.junit.MockitoJUnitRunner;
36 import org.springframework.core.env.Environment;
37
38 @RunWith(MockitoJUnitRunner.class)
39 public class AutoProcessInstanceMigrationServiceTest {
40
41     @Mock
42     private ProcessEngine processEngine;
43
44     @Mock
45     private ProcessDefinition outdated;
46
47     @Mock
48     private ProcessDefinition newDef;
49
50     @Mock
51     private ProcessDefinition key;
52
53     @Mock
54     private ProcessDefinition testKey;
55
56     @Mock
57     private ProcessDefinition suspendedDef;
58
59     @Mock
60     private RepositoryService repositoryService;
61
62     @Mock
63     private ProcessDefinitionQuery query;
64
65     @Mock
66     private ProcessDefinitionQuery keyQuery;
67
68     @Mock
69     private Environment env;
70
71     @Spy
72     @InjectMocks
73     private AutoProcessInstanceMigrationService migrationService;
74
75
76     @Test
77     public void getOldProcessDefinitionsTest() {
78         List<ProcessDefinition> expectedList = new ArrayList<>();
79         expectedList.add(outdated);
80
81         List<ProcessDefinition> defList = new ArrayList<>();
82         defList.add(outdated);
83         defList.add(newDef);
84         defList.add(suspendedDef);
85
86
87         doReturn(query).when(repositoryService).createProcessDefinitionQuery();
88         doReturn(query).when(query).processDefinitionKey("test");
89         doReturn(defList).when(query).list();
90         doReturn(3).when(outdated).getVersion();
91         doReturn(4).when(newDef).getVersion();
92         doReturn(true).when(suspendedDef).isSuspended();
93         List<ProcessDefinition> outdatedList = migrationService.getOldProcessDefinitions("test", 4);
94
95         assertEquals(expectedList, outdatedList);
96     }
97
98     @Test
99     public void getProcessDefinitionsTest() {
100         List<ProcessDefinition> expected = new ArrayList<ProcessDefinition>();
101         expected.add(testKey);
102         expected.add(key);
103
104         List<String> processDefinitionKeys = new ArrayList<String>();
105         processDefinitionKeys.add("testKey");
106         processDefinitionKeys.add("key");
107
108         doReturn(processDefinitionKeys).when(env).getProperty("migration.processDefinitionKeys", List.class,
109                 new ArrayList<String>());
110
111         doReturn(query).when(repositoryService).createProcessDefinitionQuery();
112         doReturn(query).when(query).processDefinitionKey("testKey");
113         doReturn(query).when(query).latestVersion();
114         doReturn(testKey).when(query).singleResult();
115
116         doReturn(keyQuery).when(query).processDefinitionKey("key");
117         doReturn(keyQuery).when(keyQuery).latestVersion();
118         doReturn(key).when(keyQuery).singleResult();
119
120         List<ProcessDefinition> actualProcessDefinitions = migrationService.getProcessDefinitions();
121
122         assertEquals(expected, actualProcessDefinitions);
123     }
124 }