Java tutorial
package org.jtheque.persistence; import org.jtheque.persistence.utils.AbstractEntity; import org.jtheque.persistence.utils.CachedJDBCDao; import org.jtheque.persistence.utils.Query; import org.jtheque.utils.SystemProperty; import org.jtheque.utils.bean.EqualsBuilder; import org.jtheque.utils.bean.HashCodeUtils; import org.jtheque.utils.bean.ReflectionUtils; import org.jtheque.utils.io.FileUtils; import org.jtheque.utils.unit.db.AbstractDBUnitTest; import org.dbunit.dataset.DataSetException; import org.junit.AfterClass; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.slf4j.LoggerFactory; import org.springframework.jdbc.core.RowMapper; import org.springframework.jdbc.core.simple.ParameterizedRowMapper; import org.springframework.test.context.ContextConfiguration; import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; import javax.annotation.PostConstruct; import javax.annotation.Resource; import javax.sql.DataSource; import java.io.File; import java.lang.reflect.Field; import java.sql.ResultSet; import java.sql.SQLException; import ch.qos.logback.classic.Level; import ch.qos.logback.classic.Logger; import static org.junit.Assert.*; /* * Copyright JTheque (Baptiste Wicht) * * 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. */ @RunWith(SpringJUnit4ClassRunner.class) @ContextConfiguration(locations = "jtheque-persistence-test.xml") public class CachedJDBCDaoTest extends AbstractDBUnitTest { private BasicDao dao; @Resource private DataSource dataSource; @Resource private DaoPersistenceContext daoPersistenceContext; private static String userDir; static { ((Logger) LoggerFactory.getLogger("root")).setLevel(Level.ERROR); userDir = SystemProperty.USER_DIR.get(); File folder = new File(SystemProperty.JAVA_IO_TMP_DIR.get(), "jtheque"); folder.mkdirs(); SystemProperty.USER_DIR.set(folder.getAbsolutePath()); } public CachedJDBCDaoTest() { super("datas.xml"); } @AfterClass public static void after() { FileUtils.delete(new File(SystemProperty.JAVA_IO_TMP_DIR.get(), "jtheque")); SystemProperty.USER_DIR.set(userDir); } @PostConstruct public void init() { initDB(dataSource); } @Before public void before() { dao = new BasicDao(); try { Field f = dao.getClass().getSuperclass().getSuperclass().getDeclaredField("persistenceContext"); f.setAccessible(true); f.set(dao, daoPersistenceContext); } catch (NoSuchFieldException e) { LoggerFactory.getLogger(ReflectionUtils.class).error(e.getMessage(), e); } catch (IllegalAccessException e) { LoggerFactory.getLogger(ReflectionUtils.class).error(e.getMessage(), e); } } @Test public void getAll() { assertEquals(5, dao.getAll().size()); } @Test public void getId() { assertEquals("Test 1", dao.get(1).getTitle()); assertEquals("Test 2", dao.get(2).getTitle()); assertEquals("Unit 2", dao.get(3).getTitle()); assertEquals("Unit 4", dao.get(4).getTitle()); assertEquals("GOF 4", dao.get(5).getTitle()); assertNull(dao.get(100)); assertNull(dao.get(0)); } @Test public void exists() { for (int i = 1; i < 6; i++) { assertTrue(dao.exists(i)); } assertFalse(dao.exists(100)); assertFalse(dao.exists(13)); assertFalse(dao.exists(0)); } @Test public void deleteById() { dao.delete(5); assertEquals(4, dao.getAll().size()); assertEquals(4, getTable("T_TEST").getRowCount()); for (int i = 1; i < 5; i++) { assertTrue(dao.exists(i)); } } @Test public void deleteEntity() { dao.delete(dao.get(4)); assertEquals(4, dao.getAll().size()); assertEquals(4, getTable("T_TEST").getRowCount()); assertTrue(dao.exists(1)); assertTrue(dao.exists(2)); assertTrue(dao.exists(3)); assertTrue(dao.exists(5)); } @Test public void save() throws DataSetException { TestEntity entity = dao.get(3); entity.setTitle("Edited title"); dao.save(entity); assertEquals(5, dao.getAll().size()); assertEquals(5, getTable("T_TEST").getRowCount()); assertEquals("Edited title", dao.get(3).getTitle()); assertEquals("Edited title", getTable("T_TEST").getValue(2, "TITLE")); } @Test public void clearAll() { dao.clearAll(); assertEquals(0, dao.getAll().size()); assertEquals(0, getTable("T_TEST").getRowCount()); } private static final class BasicDao extends CachedJDBCDao<TestEntity> { private static final String TABLE = "T_TEST"; private final QueryMapper queryMapper = new TestEntityQueryMapper(); private final RowMapper<TestEntity> rowMapper = new TestEntityRowMapper(); private BasicDao() { super(TABLE); } @Override protected void loadCache() { defaultFillCache(); } @Override protected QueryMapper getQueryMapper() { return queryMapper; } @Override protected RowMapper<TestEntity> getRowMapper() { return rowMapper; } @Override public TestEntity create() { return new TestEntity(); } private final class TestEntityRowMapper implements ParameterizedRowMapper<TestEntity> { @Override public TestEntity mapRow(ResultSet rs, int i) throws SQLException { TestEntity entity = create(); entity.setId(rs.getInt("ID")); entity.setTitle(rs.getString("TITLE")); entity.setTrigger(rs.getInt("TRIGGER")); return entity; } } private static final class TestEntityQueryMapper implements QueryMapper { @Override public Query constructInsertQuery(Entity entity) { String query = "INSERT INTO " + TABLE + " (TITLE, TRIGGER) VALUES(?, ?)"; return new Query(query, fillArray((TestEntity) entity, false)); } @Override public Query constructUpdateQuery(Entity entity) { String query = "UPDATE " + TABLE + " SET TITLE = ?, TRIGGER = ? WHERE ID = ?"; return new Query(query, fillArray((TestEntity) entity, true)); } /** * Fill the array with the informations of the collection. * * @param entity The collection to use to fill the array. * @param id Indicate if we must add the id to the array. * * @return The filled array. */ private static Object[] fillArray(TestEntity entity, boolean id) { Object[] values = new Object[2 + (id ? 1 : 0)]; values[0] = entity.getTitle(); values[1] = entity.getTrigger(); if (id) { values[2] = entity.getId(); } return values; } } } private static class TestEntity extends AbstractEntity { private String title; private int trigger; @Override public boolean equals(Object object) { if (object instanceof TestEntity) { TestEntity other = (TestEntity) object; return EqualsBuilder.newBuilder(this, other).addField(getId(), other.getId()) .addField(title, other.title).addField(trigger, other.trigger).areEquals(); } return false; } @Override public int hashCode() { return HashCodeUtils.hashCodeDirect(getId(), title, trigger); } @Override public String getDisplayableText() { return title; } public String getTitle() { return title; } public void setTitle(String title) { this.title = title; } public int getTrigger() { return trigger; } public void setTrigger(int trigger) { this.trigger = trigger; } } }