|
3 | 3 | import com.clickhouse.client.api.ClientConfigProperties; |
4 | 4 | import com.clickhouse.client.api.internal.ServerSettings; |
5 | 5 | import com.clickhouse.client.api.query.GenericRecord; |
| 6 | +import com.clickhouse.data.ClickHouseVersion; |
| 7 | +import org.apache.commons.lang3.RandomStringUtils; |
6 | 8 | import org.slf4j.Logger; |
7 | 9 | import org.slf4j.LoggerFactory; |
8 | 10 | import org.testng.Assert; |
9 | | -import com.clickhouse.data.ClickHouseVersion; |
10 | | -import org.apache.commons.lang3.RandomStringUtils; |
11 | 11 | import org.testng.annotations.DataProvider; |
12 | 12 | import org.testng.annotations.Test; |
13 | 13 |
|
@@ -383,50 +383,50 @@ public void testSettingRole() throws SQLException { |
383 | 383 | info.setProperty("password", userPass); |
384 | 384 |
|
385 | 385 | try (ConnectionImpl conn = new ConnectionImpl(getEndpointString(), info)) { |
386 | | - GenericRecord record = conn.getClient().queryAll("SELECT currentRoles()").get(0); |
387 | | - assertEquals(record.getList(1).size(), 0); |
| 386 | + GenericRecord dataRecord = conn.getClient().queryAll("SELECT currentRoles()").get(0); |
| 387 | + assertEquals(dataRecord.getList(1).size(), 0); |
388 | 388 |
|
389 | 389 | try (Statement stmt = conn.createStatement()) { |
390 | 390 | stmt.execute("SET ROLE role1"); |
391 | 391 | } |
392 | 392 |
|
393 | | - record = conn.getClient().queryAll("SELECT currentRoles()").get(0); |
394 | | - assertEquals(record.getList(1).size(), 1); |
395 | | - assertEquals(record.getList(1).get(0), "role1"); |
| 393 | + dataRecord = conn.getClient().queryAll("SELECT currentRoles()").get(0); |
| 394 | + assertEquals(dataRecord.getList(1).size(), 1); |
| 395 | + assertEquals(dataRecord.getList(1).get(0), "role1"); |
396 | 396 |
|
397 | 397 | try (Statement stmt = conn.createStatement()) { |
398 | 398 | stmt.execute("SET ROLE role2"); |
399 | 399 | } |
400 | 400 |
|
401 | | - record = conn.getClient().queryAll("SELECT currentRoles()").get(0); |
402 | | - assertEquals(record.getList(1).size(), 1); |
403 | | - assertEquals(record.getList(1).get(0), "role2"); |
| 401 | + dataRecord = conn.getClient().queryAll("SELECT currentRoles()").get(0); |
| 402 | + assertEquals(dataRecord.getList(1).size(), 1); |
| 403 | + assertEquals(dataRecord.getList(1).get(0), "role2"); |
404 | 404 |
|
405 | 405 | try (Statement stmt = conn.createStatement()) { |
406 | 406 | stmt.execute("SET ROLE NONE"); |
407 | 407 | } |
408 | 408 |
|
409 | | - record = conn.getClient().queryAll("SELECT currentRoles()").get(0); |
410 | | - assertEquals(record.getList(1).size(), 0); |
| 409 | + dataRecord = conn.getClient().queryAll("SELECT currentRoles()").get(0); |
| 410 | + assertEquals(dataRecord.getList(1).size(), 0); |
411 | 411 |
|
412 | 412 | try (Statement stmt = conn.createStatement()) { |
413 | 413 | stmt.execute("SET ROLE \"role1\",\"role2\""); |
414 | 414 | } |
415 | 415 |
|
416 | | - record = conn.getClient().queryAll("SELECT currentRoles()").get(0); |
417 | | - assertEquals(record.getList(1).size(), 2); |
418 | | - assertEquals(record.getList(1).get(0), "role1"); |
419 | | - assertEquals(record.getList(1).get(1), "role2"); |
| 416 | + dataRecord = conn.getClient().queryAll("SELECT currentRoles()").get(0); |
| 417 | + assertEquals(dataRecord.getList(1).size(), 2); |
| 418 | + assertEquals(dataRecord.getList(1).get(0), "role1"); |
| 419 | + assertEquals(dataRecord.getList(1).get(1), "role2"); |
420 | 420 |
|
421 | 421 | try (Statement stmt = conn.createStatement()) { |
422 | 422 | stmt.execute("SET ROLE \"role1\",\"role2\",\"role3\""); |
423 | 423 | } |
424 | 424 |
|
425 | | - record = conn.getClient().queryAll("SELECT currentRoles()").get(0); |
426 | | - assertEquals(record.getList(1).size(), 3); |
427 | | - assertEquals(record.getList(1).get(0), "role1"); |
428 | | - assertEquals(record.getList(1).get(1), "role2"); |
429 | | - assertEquals(record.getList(1).get(2), "role3"); |
| 425 | + dataRecord = conn.getClient().queryAll("SELECT currentRoles()").get(0); |
| 426 | + assertEquals(dataRecord.getList(1).size(), 3); |
| 427 | + assertEquals(dataRecord.getList(1).get(0), "role1"); |
| 428 | + assertEquals(dataRecord.getList(1).get(1), "role2"); |
| 429 | + assertEquals(dataRecord.getList(1).get(2), "role3"); |
430 | 430 | } |
431 | 431 | } |
432 | 432 |
|
|
0 commit comments