/**
* Copyright 2020 ZTE Corporation.
*
* 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
*
* 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 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 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 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 getJdbiDaoByOnDemand(Class daoClazz) {
return (T) dao;
}
}
class EngineEntityDaoStub implements EngineEntityDao {
private Set 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 getAllEntities() {
return new ArrayList<>(entitySet);
}
@Override
public List 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;
}
}
}
}