github/jooq
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[#3442] Implement MockResultSet.getArray() methods

Browse Source
This commit is contained in:
Lukas Eder 2014-07-23 13:54:29 +02:00
parent d6d24bb829
commit c6e1e0da81
2 changed files with 85 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jOOQ/src/main/java/org/jooq/tools/jdbc/MockResultSet.java
View File

@ -703,12 +703,12 @@ public class MockResultSet extends JDBC41ResultSet implements ResultSet, Seriali
@Override
public Array getArray(int columnIndex) throws SQLException {
throw new SQLFeatureNotSupportedException("Unsupported data type. Use getObject() instead");
return getValue(columnIndex, Array.class);
}
@Override
public Array getArray(String columnLabel) throws SQLException {
throw new SQLFeatureNotSupportedException("Unsupported data type. Use getObject() instead");
return getValue(columnLabel, Array.class);
}
@Override

83
jOOQ/src/test/java/org/jooq/test/MockTest.java
View File

@ -41,6 +41,8 @@
package org.jooq.test;
import static java.util.Arrays.asList;
import static java.util.Collections.nCopies;
import static org.jooq.impl.DSL.field;
import static org.jooq.impl.DSL.val;
import static org.jooq.test.data.Table1.FIELD_ID1;
import static org.jooq.test.data.Table1.FIELD_NAME1;
@ -49,16 +51,21 @@ import static org.jooq.test.data.Table2.FIELD_ID2;
import static org.jooq.test.data.Table2.FIELD_NAME2;
import static org.jooq.test.data.Table2.TABLE2;
import static org.jooq.test.data.Table3.FIELD_NAME3;
import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import java.io.File;
import java.io.RandomAccessFile;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.List;
import org.jooq.Constants;
import org.jooq.DSLContext;
import org.jooq.Field;
import org.jooq.InsertResultStep;
import org.jooq.Query;
import org.jooq.Record;
@ -426,4 +433,80 @@ public class MockTest extends AbstractTest {
assertEquals("y", r2.getValue(0, 1));
assertEquals("z", r2.getValue(0, 2));
}
final Field<String[]> strings = field("STRINGS", String[].class);
final Field<Integer[]> integers = field("INTEGERS", Integer[].class);
class ArrayResult extends AbstractResult {
final int length;
ArrayResult(int length) {
this.length = length;
}
@Override
public MockResult[] execute(MockExecuteContext ctx) throws SQLException {
Result<Record2<String[], Integer[]>> result = create.newResult(strings, integers);
Record2<String[], Integer[]> empty = create.newRecord(strings, integers);
result.add(empty);
for (int i = 0; i < length; i++) {
Record2<String[], Integer[]> record = create.newRecord(strings, integers);
record.setValue(strings, nCopies(i, "" + i).toArray(new String[i]));
record.setValue(integers, nCopies(i, i).toArray(new Integer[i]));
result.add(record);
}
return new MockResult[] {
new MockResult(0, result)
};
}
}
@Test
public void testArrays() {
DSLContext e = DSL.using(new MockConnection(new ArrayResult(3)), SQLDialect.POSTGRES);
Result<Record2<String[], Integer[]>> r = e.select(strings, integers).fetch();
assertEquals(4, r.size());
assertEquals(2, r.fields().length);
assertNull(r.getValue(0, "STRINGS"));
assertNull(r.getValue(0, "INTEGERS"));
assertArrayEquals(new String[0], (String[]) r.getValue(1, "STRINGS"));
assertArrayEquals(new Integer[0], (Integer[]) r.getValue(1, "INTEGERS"));
assertArrayEquals(new String[] { "1" }, (String[]) r.getValue(2, "STRINGS"));
assertArrayEquals(new Integer[] { 1 }, (Integer[]) r.getValue(2, "INTEGERS"));
assertArrayEquals(new String[] { "2", "2" }, (String[]) r.getValue(3, "STRINGS"));
assertArrayEquals(new Integer[] { 2, 2 }, (Integer[]) r.getValue(3, "INTEGERS"));
}
@Test
public void testJDBCArrays() throws SQLException {
DSLContext e = DSL.using(new MockConnection(new ArrayResult(3)), SQLDialect.POSTGRES);
ResultSet r = e.select(strings, integers).fetchResultSet();
assertEquals(2, r.getMetaData().getColumnCount());
assertTrue(r.next());
assertNull(r.getArray("STRINGS"));
assertNull(r.getArray("INTEGERS"));
assertTrue(r.next());
assertArrayEquals(new String[0], (String[]) r.getArray("STRINGS").getArray());
assertArrayEquals(new Integer[0], (Integer[]) r.getArray("INTEGERS").getArray());
assertTrue(r.next());
assertArrayEquals(new String[] { "1" }, (String[]) r.getArray("STRINGS").getArray());
assertArrayEquals(new Integer[] { 1 }, (Integer[]) r.getArray("INTEGERS").getArray());
assertTrue(r.next());
assertArrayEquals(new String[] { "2", "2" }, (String[]) r.getArray("STRINGS").getArray());
assertArrayEquals(new Integer[] { 2, 2 }, (Integer[]) r.getArray("INTEGERS").getArray());
assertFalse(r.next());
}
}