1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
/**
* Copyright 2020 ZTE Corporation.
* <p>
* Licensed 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
* <p>
* http://www.apache.org/licenses/LICENSE-2.0
* <p>
* 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 org.onap.holmes.common.engine.service.impl;
import org.jdbi.v3.core.Jdbi;
import org.junit.Test;
import org.onap.holmes.common.engine.dao.EngineEntityDao;
import org.onap.holmes.common.engine.entity.EngineEntity;
import org.onap.holmes.common.engine.service.EngineEntityService;
import org.onap.holmes.common.database.DbDaoUtil;
import java.util.*;
import static org.hamcrest.core.Is.is;
import static org.hamcrest.core.IsNull.notNullValue;
import static org.junit.Assert.*;
public class EngineEntityServiceImplTest {
private EngineEntityService service = new EngineEntityServiceImpl(new DbDaoUtilStub(null));
@Test
public void getLegacyEngines() {
List<String> legacyEngines = service.getLegacyEngines();
assertThat(legacyEngines.size(), is(2));
}
@Test
public void getEntity() throws Exception {
EngineEntity entity = service.getEntity("org.onap.holmes_9201");
assertThat(entity, notNullValue());
}
@Test
public void getAllEntities() throws Exception {
List<EngineEntity> entities = service.getAllEntities();
assertThat(entities.size(), is(1));
}
@Test
public void updateEntity() throws Exception {
EngineEntity entity = new EngineEntity("org.onap.holmes", 9201);
long time = System.currentTimeMillis();
entity.setLastModified(time);
service.updateEntity(entity);
assertThat(service.getEntity("org.onap.holmes_9201").getLastModified(), is(time));
}
@Test
public void insertEntity() throws Exception {
EngineEntity entity = new EngineEntity("org.onap.holmes.another", 9201);
service.insertEntity(entity);
assertThat(service.getAllEntities().size(), is(2));
}
@Test
public void deleteEntity() throws Exception {
service.deleteEntity("org.onap.holmes.another_9201");
assertThat(service.getAllEntities().size(), is(1));
}
}
class DbDaoUtilStub extends DbDaoUtil {
private EngineEntityDao dao = new EngineEntityDaoStub();
public DbDaoUtilStub(Jdbi jdbi) {
super(jdbi);
}
@Override
public <T> T getJdbiDaoByOnDemand(Class<T> daoClazz) {
return (T) dao;
}
}
class EngineEntityDaoStub implements EngineEntityDao {
private Set<EngineEntity> entitySet = new HashSet(){
{
add(new EngineEntity("org.onap.holmes", 9201));
}
};
@Override
public EngineEntity getEntity(String id) {
return entitySet.stream().filter(e -> e.getId().equals(id)).findFirst().get();
}
@Override
public List<EngineEntity> getAllEntities() {
return new ArrayList<>(entitySet);
}
@Override
public List<String> getLegacyEngines() {
return Arrays.asList("org.onap.holmes", "org.onap.holmes.legacy.1");
}
@Override
public void insertEntity(EngineEntity entity) {
entitySet.add(entity);
}
@Override
public void updateEntity(EngineEntity entity) {
entitySet.add(entity);
}
@Override
public void deleteEntity(String id) {
for (EngineEntity entity : entitySet) {
if (entity.getId().equals(id)) {
entitySet.remove(entity);
break;
}
}
}
}
|
