77b3535ce10955c9aad2d58c758492925a0e5a57
[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         doReturn(query).when(repositoryService).createProcessDefinitionQuery();
87         doReturn(query).when(query).processDefinitionKey("test");
88         doReturn(defList).when(query).list();
89         doReturn(3).when(outdated).getVersion();
90         doReturn(4).when(newDef).getVersion();
91         doReturn(true).when(suspendedDef).isSuspended();
92         List<ProcessDefinition> outdatedList = migrationService.getOldProcessDefinitions("test", 4);
93
94         assertEquals(expectedList, outdatedList);
95     }
96
97     @Test
98     public void getProcessDefinitionsTest() {
99         List<ProcessDefinition> expected = new ArrayList<ProcessDefinition>();
100         expected.add(testKey);
101         expected.add(key);
102
103         List<String> processDefinitionKeys = new ArrayList<String>();
104         processDefinitionKeys.add("testKey");
105         processDefinitionKeys.add("key");
106
107         doReturn(processDefinitionKeys).when(env).getProperty("migration.processDefinitionKeys", List.class,
108                 new ArrayList<String>());
109
110         doReturn(query).when(repositoryService).createProcessDefinitionQuery();
111         doReturn(query).when(query).processDefinitionKey("testKey");
112         doReturn(query).when(query).latestVersion();
113         doReturn(testKey).when(query).singleResult();
114
115         doReturn(keyQuery).when(query).processDefinitionKey("key");
116         doReturn(keyQuery).when(keyQuery).latestVersion();
117         doReturn(key).when(keyQuery).singleResult();
118
119         List<ProcessDefinition> actualProcessDefinitions = migrationService.getProcessDefinitions();
120
121         assertEquals(expected, actualProcessDefinitions);
122     }
123 }