|
30 | 30 | @RunWith(JUnit4.class) |
31 | 31 | public class ITTransactionRetryTest { |
32 | 32 |
|
33 | | - @ClassRule |
34 | | - public static IntegrationTestEnv env = new IntegrationTestEnv(); |
| 33 | + @ClassRule public static IntegrationTestEnv env = new IntegrationTestEnv(); |
35 | 34 |
|
36 | 35 | @Test |
37 | 36 | public void TestRetryInfo() { |
38 | 37 | // Creating a database with the table which contains INT64 columns |
39 | | - Database db = env.getTestHelper().createTestDatabase("CREATE TABLE Test(ID INT64, " |
40 | | - + "EMPID INT64) PRIMARY KEY (ID)"); |
| 38 | + Database db = |
| 39 | + env.getTestHelper() |
| 40 | + .createTestDatabase("CREATE TABLE Test(ID INT64, " + "EMPID INT64) PRIMARY KEY (ID)"); |
41 | 41 | DatabaseClient databaseClient = env.getTestHelper().getClient().getDatabaseClient(db.getId()); |
42 | 42 |
|
43 | 43 | // Inserting one row |
44 | | - databaseClient.readWriteTransaction().run(transaction -> { |
45 | | - transaction.buffer(Mutation.newInsertBuilder("Test") |
46 | | - .set("ID") |
47 | | - .to(1) |
48 | | - .set("EMPID") |
49 | | - .to(1) |
50 | | - .build()); |
51 | | - return null; |
52 | | - }); |
| 44 | + databaseClient |
| 45 | + .readWriteTransaction() |
| 46 | + .run( |
| 47 | + transaction -> { |
| 48 | + transaction.buffer( |
| 49 | + Mutation.newInsertBuilder("Test").set("ID").to(1).set("EMPID").to(1).build()); |
| 50 | + return null; |
| 51 | + }); |
53 | 52 |
|
54 | | - try(TransactionManager transactionManager1 = databaseClient.transactionManager()) { |
55 | | - try(TransactionManager transactionManager2 = databaseClient.transactionManager()) { |
| 53 | + try (TransactionManager transactionManager1 = databaseClient.transactionManager()) { |
| 54 | + try (TransactionManager transactionManager2 = databaseClient.transactionManager()) { |
56 | 55 | TransactionContext transaction1 = transactionManager1.begin(); |
57 | 56 | TransactionContext transaction2 = transactionManager2.begin(); |
58 | 57 | transaction1.executeUpdate(Statement.of("UPDATE Test SET EMPID = EMPID + 1 WHERE ID = 1")); |
59 | 58 | transaction2.executeUpdate(Statement.of("UPDATE Test SET EMPID = EMPID + 1 WHERE ID = 1")); |
60 | 59 | transactionManager1.commit(); |
61 | | - AbortedException abortedException = Assert.assertThrows(AbortedException.class, |
62 | | - transactionManager2::commit); |
| 60 | + AbortedException abortedException = |
| 61 | + Assert.assertThrows(AbortedException.class, transactionManager2::commit); |
63 | 62 | assertThat(abortedException.getErrorCode()).isEqualTo(ErrorCode.ABORTED); |
64 | 63 | assertTrue(abortedException.getRetryDelayInMillis() > 0); |
65 | 64 | } |
|
0 commit comments