123 lines
3.2 KiB
Java
123 lines
3.2 KiB
Java
package com.iluwatar.servicelayer.common;
|
|
|
|
import com.iluwatar.servicelayer.hibernate.HibernateUtil;
|
|
|
|
import org.junit.After;
|
|
import org.junit.Before;
|
|
import org.junit.Test;
|
|
|
|
import java.util.List;
|
|
import java.util.concurrent.atomic.AtomicInteger;
|
|
import java.util.function.Function;
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
import static org.junit.Assert.assertNotNull;
|
|
import static org.junit.Assert.assertNull;
|
|
|
|
/**
|
|
* Date: 12/28/15 - 10:53 PM
|
|
*
|
|
* @author Jeroen Meulemeester
|
|
*/
|
|
public abstract class BaseDaoTest<E extends BaseEntity, D extends DaoBaseImpl<E>> {
|
|
|
|
/**
|
|
* The number of entities stored before each test
|
|
*/
|
|
private static final int INITIAL_COUNT = 5;
|
|
|
|
/**
|
|
* The unique id generator, shared between all entities
|
|
*/
|
|
private static final AtomicInteger ID_GENERATOR = new AtomicInteger();
|
|
|
|
/**
|
|
* Factory, used to create new entity instances with the given name
|
|
*/
|
|
private final Function<String, E> factory;
|
|
|
|
/**
|
|
* The tested data access object
|
|
*/
|
|
private final D dao;
|
|
|
|
/**
|
|
* Create a new test using the given factory and dao
|
|
*
|
|
* @param factory The factory, used to create new entity instances with the given name
|
|
* @param dao The tested data access object
|
|
*/
|
|
public BaseDaoTest(final Function<String, E> factory, final D dao) {
|
|
this.factory = factory;
|
|
this.dao = dao;
|
|
}
|
|
|
|
@Before
|
|
public void setUp() throws Exception {
|
|
for (int i = 0; i < INITIAL_COUNT; i++) {
|
|
final String className = dao.persistentClass.getSimpleName();
|
|
final String entityName = String.format("%s%d", className, ID_GENERATOR.incrementAndGet());
|
|
this.dao.persist(this.factory.apply(entityName));
|
|
}
|
|
}
|
|
|
|
@After
|
|
public void tearDown() throws Exception {
|
|
HibernateUtil.dropSession();
|
|
}
|
|
|
|
protected final D getDao() {
|
|
return this.dao;
|
|
}
|
|
|
|
@Test
|
|
public void testFind() throws Exception {
|
|
final List<E> all = this.dao.findAll();
|
|
for (final E entity : all) {
|
|
final E byId = this.dao.find(entity.getId());
|
|
assertNotNull(byId);
|
|
assertEquals(byId.getId(), byId.getId());
|
|
}
|
|
}
|
|
|
|
@Test
|
|
public void testDelete() throws Exception {
|
|
final List<E> originalEntities = this.dao.findAll();
|
|
this.dao.delete(originalEntities.get(1));
|
|
this.dao.delete(originalEntities.get(2));
|
|
|
|
final List<E> entitiesLeft = this.dao.findAll();
|
|
assertNotNull(entitiesLeft);
|
|
assertEquals(INITIAL_COUNT - 2, entitiesLeft.size());
|
|
}
|
|
|
|
@Test
|
|
public void testFindAll() throws Exception {
|
|
final List<E> all = this.dao.findAll();
|
|
assertNotNull(all);
|
|
assertEquals(INITIAL_COUNT, all.size());
|
|
}
|
|
|
|
@Test
|
|
public void testSetId() throws Exception {
|
|
final E entity = this.factory.apply("name");
|
|
assertNull(entity.getId());
|
|
|
|
final Long expectedId = Long.valueOf(1);
|
|
entity.setId(expectedId);
|
|
assertEquals(expectedId, entity.getId());
|
|
}
|
|
|
|
@Test
|
|
public void testSetName() throws Exception {
|
|
final E entity = this.factory.apply("name");
|
|
assertEquals("name", entity.getName());
|
|
assertEquals("name", entity.toString());
|
|
|
|
final String expectedName = "new name";
|
|
entity.setName(expectedName);
|
|
assertEquals(expectedName, entity.getName());
|
|
assertEquals(expectedName, entity.toString());
|
|
}
|
|
|
|
} |