|
19 | 19 | import java.sql.JDBCType;
|
20 | 20 | import java.sql.PreparedStatement;
|
21 | 21 | import java.sql.ResultSet;
|
| 22 | +import java.sql.ResultSetMetaData; |
22 | 23 | import java.sql.SQLException;
|
23 | 24 | import java.sql.Statement;
|
24 | 25 | import java.sql.Timestamp;
|
|
28 | 29 | import java.time.OffsetDateTime;
|
29 | 30 | import java.time.ZoneId;
|
30 | 31 | import java.time.ZonedDateTime;
|
| 32 | +import java.util.ArrayList; |
| 33 | +import java.util.Arrays; |
31 | 34 | import java.util.GregorianCalendar;
|
32 | 35 | import java.util.HashMap;
|
| 36 | +import java.util.List; |
33 | 37 | import java.util.Map;
|
34 | 38 | import java.util.Properties;
|
35 | 39 | import java.util.Random;
|
@@ -220,6 +224,78 @@ public void testIntegerTypes() throws SQLException {
|
220 | 224 | }
|
221 | 225 | }
|
222 | 226 |
|
| 227 | + @Test(groups = { "integration" }) |
| 228 | + public void testUnsignedIntegerTypes() throws Exception { |
| 229 | + Random rand = new Random(); |
| 230 | + runQuery("CREATE TABLE test_unsigned_integers (order Int8, " |
| 231 | + + "uint8 Nullable(UInt8), " |
| 232 | + + "uint16 Nullable(UInt16), " |
| 233 | + + "uint32 Nullable(UInt32), " |
| 234 | + + "uint64 Nullable(UInt64), " |
| 235 | + + "uint128 Nullable(UInt128), " |
| 236 | + + "uint256 Nullable(UInt256)" |
| 237 | + + ") ENGINE = MergeTree ORDER BY ()"); |
| 238 | + |
| 239 | + // Insert null values |
| 240 | + insertData("INSERT INTO test_unsigned_integers VALUES ( 1, " |
| 241 | + + "NULL, NULL, NULL, NULL, NULL, NULL)"); |
| 242 | + |
| 243 | + // Insert minimum values |
| 244 | + insertData("INSERT INTO test_unsigned_integers VALUES ( 2, " |
| 245 | + + "0, 0, 0, 0, 0, 0)"); |
| 246 | + |
| 247 | + // Insert random values |
| 248 | + int uint8 = rand.nextInt(256); |
| 249 | + int uint16 = rand.nextInt(65536); |
| 250 | + long uint32 = rand.nextLong() & 0xFFFFFFFFL; |
| 251 | + long uint64 = rand.nextLong() & 0xFFFFFFFFFFFFL; |
| 252 | + BigInteger uint128 = new BigInteger(38, rand); |
| 253 | + BigInteger uint256 = new BigInteger(77, rand); |
| 254 | + insertData("INSERT INTO test_unsigned_integers VALUES ( 3, " |
| 255 | + + uint8 + ", " + uint16 + ", " + uint32 + ", " + uint64 + ", " + uint128 + ", " + uint256 + ")"); |
| 256 | + |
| 257 | + try (Connection conn = getConnection(); |
| 258 | + Statement stmt = conn.createStatement(); |
| 259 | + ResultSet rs = stmt.executeQuery("SELECT * FROM test_unsigned_integers ORDER BY order")) { |
| 260 | + |
| 261 | + List<Class<?>> expectedTypes = Arrays.asList( |
| 262 | + Short.class, Integer.class, Long.class, BigInteger.class, BigInteger.class, BigInteger.class); |
| 263 | + List<Class<?>> actualTypes = new ArrayList<>(); |
| 264 | + ResultSetMetaData rsmd = rs.getMetaData(); |
| 265 | + for (int i = 2; i <= rsmd.getColumnCount(); i++) { |
| 266 | + actualTypes.add(Class.forName(rsmd.getColumnClassName(i))); |
| 267 | + } |
| 268 | + assertEquals(actualTypes, expectedTypes); |
| 269 | + |
| 270 | + |
| 271 | + assertTrue(rs.next()); |
| 272 | + assertEquals((Short) rs.getObject("uint8"), null); |
| 273 | + assertEquals((Integer) rs.getObject("uint16"), null); |
| 274 | + assertEquals((Long) rs.getObject("uint32"), null); |
| 275 | + assertEquals((BigInteger) rs.getObject("uint64"), null); |
| 276 | + assertEquals((BigInteger) rs.getObject("uint128"), null); |
| 277 | + assertEquals((BigInteger) rs.getObject("uint256"), null); |
| 278 | + |
| 279 | + assertTrue(rs.next()); |
| 280 | + assertEquals((Short) rs.getObject("uint8"), (byte) 0); |
| 281 | + assertEquals((Integer) rs.getObject("uint16"), (short) 0); |
| 282 | + assertEquals((Long) rs.getObject("uint32"), 0); |
| 283 | + assertEquals((BigInteger) rs.getObject("uint64"), BigInteger.ZERO); |
| 284 | + assertEquals((BigInteger) rs.getObject("uint128"), BigInteger.ZERO); |
| 285 | + assertEquals((BigInteger) rs.getObject("uint256"), BigInteger.ZERO); |
| 286 | + |
| 287 | + assertTrue(rs.next()); |
| 288 | + assertEquals(((Short) rs.getObject("uint8")).intValue(), uint8); |
| 289 | + assertEquals((Integer) rs.getObject("uint16"), uint16); |
| 290 | + assertEquals((Long) rs.getObject("uint32"), uint32); |
| 291 | + assertEquals((BigInteger) rs.getObject("uint64"), BigInteger.valueOf(uint64)); |
| 292 | + assertEquals((BigInteger) rs.getObject("uint128"), uint128); |
| 293 | + assertEquals((BigInteger) rs.getObject("uint256"), uint256); |
| 294 | + |
| 295 | + assertFalse(rs.next()); |
| 296 | + } |
| 297 | + } |
| 298 | + |
223 | 299 | @Test(groups = { "integration" })
|
224 | 300 | public void testDecimalTypes() throws SQLException {
|
225 | 301 | runQuery("CREATE TABLE test_decimals (order Int8, "
|
|
0 commit comments