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
|
/**
* 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 com.google.common.base.CharMatcher;
import org.junit.Before;
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.utils.DbDaoUtil;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import static com.google.common.base.Predicates.notNull;
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());
@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();
@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 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;
}
}
}
}
|