/*
* Copyright 2004-2015 the Seasar Foundation and the Others.
*
* 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.seasar.extension.jdbc.query;
import java.sql.CallableStatement;
import java.sql.SQLException;
import java.util.Date;
import java.util.List;
import javax.persistence.Lob;
import javax.persistence.Temporal;
import javax.persistence.TemporalType;
import junit.framework.TestCase;
import org.seasar.extension.jdbc.JdbcContext;
import org.seasar.extension.jdbc.ParamType;
import org.seasar.extension.jdbc.SqlLog;
import org.seasar.extension.jdbc.SqlLogRegistry;
import org.seasar.extension.jdbc.SqlLogRegistryLocator;
import org.seasar.extension.jdbc.annotation.InOut;
import org.seasar.extension.jdbc.annotation.Out;
import org.seasar.extension.jdbc.annotation.ResultSet;
import org.seasar.extension.jdbc.dialect.StandardDialect;
import org.seasar.extension.jdbc.entity.Aaa;
import org.seasar.extension.jdbc.exception.QueryTwiceExecutionRuntimeException;
import org.seasar.extension.jdbc.manager.JdbcManagerImpl;
import org.seasar.extension.jdbc.types.ValueTypes;
import org.seasar.extension.jta.TransactionManagerImpl;
import org.seasar.extension.jta.TransactionSynchronizationRegistryImpl;
import org.seasar.framework.mock.sql.MockCallableStatement;
import org.seasar.framework.mock.sql.MockDataSource;
/**
* @author koichik
*/
public class AutoProcedureCallImplTest extends TestCase {
private JdbcManagerImpl manager;
@Override
protected void setUp() throws Exception {
manager = new JdbcManagerImpl();
manager.setSyncRegistry(new TransactionSynchronizationRegistryImpl(
new TransactionManagerImpl()));
manager.setDataSource(new MockDataSource());
manager.setDialect(new StandardDialect());
}
@Override
protected void tearDown() throws Exception {
SqlLogRegistry regisry = SqlLogRegistryLocator.getInstance();
regisry.clear();
manager = null;
}
/**
* @throws Exception
*/
public void testPrepareParameter_noArg() throws Exception {
AutoProcedureCallImpl query = new AutoProcedureCallImpl(manager, "hoge");
query.prepare("execute");
assertEquals("{call hoge()}", query.executedSql);
assertEquals(0, query.getParamSize());
}
/**
* @throws Exception
*/
public void testPrepareParameter_simpleType() throws Exception {
AutoProcedureCallImpl query = new AutoProcedureCallImpl(manager,
"hoge", 1);
query.prepare("execute");
assertEquals("{call hoge(?)}", query.executedSql);
assertEquals(1, query.getParamSize());
assertEquals(1, query.getParam(0).value);
assertEquals(Integer.class, query.getParam(0).paramClass);
assertEquals(ValueTypes.INTEGER, query.getParam(0).valueType);
}
/**
* @throws Exception
*/
public void testPrepareParameter_dto() throws Exception {
MyDto dto = new MyDto();
dto.arg2 = "aaa";
dto.arg3 = "bbb";
AutoProcedureCallImpl query = new AutoProcedureCallImpl(manager,
"hoge", dto);
query.prepare("execute");
assertEquals("{call hoge(?, ?, ?)}", query.executedSql);
assertEquals(3, query.getParamSize());
assertEquals(null, query.getParam(0).value);
assertEquals(String.class, query.getParam(0).paramClass);
assertEquals(ValueTypes.STRING, query.getParam(0).valueType);
assertEquals(ParamType.OUT, query.getParam(0).paramType);
assertEquals(MyDto.class.getDeclaredField("arg1"),
query.getParam(0).field);
assertEquals("aaa", query.getParam(1).value);
assertEquals(String.class, query.getParam(1).paramClass);
assertEquals(ValueTypes.STRING, query.getParam(1).valueType);
assertEquals(ParamType.IN_OUT, query.getParam(1).paramType);
assertEquals(MyDto.class.getDeclaredField("arg2"),
query.getParam(1).field);
assertEquals("bbb", query.getParam(2).value);
assertEquals(String.class, query.getParam(2).paramClass);
assertEquals(ValueTypes.STRING, query.getParam(2).valueType);
assertEquals(ParamType.IN, query.getParam(2).paramType);
assertNull(query.getParam(2).field);
}
/**
* @throws Exception
*/
public void testPrepareParameter_resultSet() throws Exception {
MyDto2 dto = new MyDto2();
dto.arg2 = "aaa";
AutoProcedureCallImpl query = new AutoProcedureCallImpl(manager,
"hoge", dto);
query.prepare("execute");
assertEquals("{call hoge(?, ?)}", query.executedSql);
assertEquals(2, query.getParamSize());
assertEquals(null, query.getParam(0).value);
assertEquals(String.class, query.getParam(0).paramClass);
assertEquals(ValueTypes.STRING, query.getParam(0).valueType);
assertEquals(ParamType.OUT, query.getParam(0).paramType);
assertEquals(MyDto2.class.getDeclaredField("arg1"),
query.getParam(0).field);
assertEquals("aaa", query.getParam(1).value);
assertEquals(String.class, query.getParam(1).paramClass);
assertEquals(ValueTypes.STRING, query.getParam(1).valueType);
assertEquals(ParamType.IN_OUT, query.getParam(1).paramType);
assertEquals(MyDto2.class.getDeclaredField("arg2"),
query.getParam(1).field);
assertEquals(1, query.getNonParamSize());
assertEquals(ParamType.OUT, query.getNonParam(0).paramType);
assertEquals(MyDto2.class.getDeclaredField("result"), query
.getNonParam(0).field);
}
/**
* @throws Exception
*/
public void testPrepareParameter_clob() throws Exception {
MyDto3 dto = new MyDto3();
dto.largeName = "aaa";
AutoProcedureCallImpl query = new AutoProcedureCallImpl(manager,
"hoge", dto);
query.prepare("execute");
assertEquals("{call hoge(?)}", query.executedSql);
assertEquals(1, query.getParamSize());
assertEquals("aaa", query.getParam(0).value);
assertEquals(ParamType.IN, query.getParam(0).paramType);
assertEquals(ValueTypes.CLOB, query.getParam(0).valueType);
}
/**
* @throws Exception
*/
public void testPrepareParameter_temporalType() throws Exception {
MyDto4 dto = new MyDto4();
Date date = new Date();
dto.timestamp = date;
AutoProcedureCallImpl query = new AutoProcedureCallImpl(manager,
"hoge", dto);
query.prepare("execute");
assertEquals("{call hoge(?)}", query.executedSql);
assertEquals(1, query.getParamSize());
assertEquals(date, query.getParam(0).value);
assertEquals(ParamType.IN, query.getParam(0).paramType);
assertEquals(ValueTypes.DATE_TIMESTAMP, query.getParam(0).valueType);
}
/**
* @throws Exception
*/
public void testCall() throws Exception {
MyDto dto = new MyDto();
dto.arg2 = "aaa";
dto.arg3 = "bbb";
AutoProcedureCallImpl query = new AutoProcedureCallImpl(manager,
"hoge", dto) {
@Override
protected CallableStatement getCallableStatement(
JdbcContext jdbcContext) {
MockCallableStatement cs = new MockCallableStatement(null, null) {
@Override
public String getString(int parameterIndex)
throws SQLException {
return "aaa" + parameterIndex;
}
};
return cs;
}
};
query.execute();
assertEquals("{call hoge(?, ?, ?)}", query.executedSql);
assertEquals("aaa1", dto.arg1);
assertEquals("aaa2", dto.arg2);
SqlLog sqlLog = SqlLogRegistryLocator.getInstance().getLast();
assertEquals("{call hoge(null, 'aaa', 'bbb')}", sqlLog.getCompleteSql());
try {
query.execute();
fail();
} catch (QueryTwiceExecutionRuntimeException expected) {
}
}
private static final class MyDto {
/** */
@Out
public String arg1;
/** */
@InOut
public String arg2;
/** */
public String arg3;
}
private static final class MyDto2 {
/** */
@Out
public String arg1;
/** */
@InOut
public String arg2;
/** */
@ResultSet
public List<Aaa> result;
}
private static final class MyDto3 {
/** */
@Lob
public String largeName;
}
private static final class MyDto4 {
/** */
@Temporal(TemporalType.TIMESTAMP)
public Date timestamp;
}
}