提交 b520e543 编写于 作者: T tristaZero

add assertSelectWithPlainColumn()

上级 42de0789
......@@ -26,6 +26,8 @@ import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Arrays;
import java.util.List;
import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertFalse;
......@@ -131,6 +133,21 @@ public final class EncryptStatementTest extends AbstractEncryptJDBCDatabaseAndTa
}
}
@Test
public void assertSelectWithPlainColumn() throws SQLException {
try (Statement statement = getEncryptConnectionWithProps().createStatement()) {
ResultSet resultSet = statement.executeQuery(SELECT_SQL_WITH_PLAIN);
int count = 1;
List<Object> ids = Arrays.asList((Object) 1, 5);
while (resultSet.next()) {
assertThat(resultSet.getObject("id"), is(ids.get(count - 1)));
assertThat(resultSet.getObject("pwd"), is((Object) "decryptValue"));
count += 1;
}
assertThat(count - 1, is(ids.size()));
}
}
private void assertResultSet(final int resultSetCount, final int id, final Object pwd, final Object plain) throws SQLException {
try (Connection conn = getDatabaseTypeMap().values().iterator().next().values().iterator().next().getConnection();
Statement stmt = conn.createStatement()) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册