- final DOMDataWriteTransaction wTx = Mockito.mock(DOMDataWriteTransaction.class);
- Mockito.when(domTx.newWriteOnlyTransaction()).thenReturn(wTx);
- final DOMDataReadWriteTransaction rwTx = Mockito.mock(DOMDataReadWriteTransaction.class);
- final CheckedFuture<Boolean, ReadFailedException> checkedFuture = Futures.immediateCheckedFuture(Boolean.TRUE);
- Mockito.when(rwTx.exists(Mockito.any(), Mockito.any())).thenReturn(checkedFuture);
- final CheckedFuture<Void, TransactionCommitFailedException> checkedFutureEmpty =
- Futures.immediateCheckedFuture(null);
- Mockito.when(rwTx.submit()).thenReturn(checkedFutureEmpty);
- Mockito.when(domTx.newReadWriteTransaction()).thenReturn(rwTx);
- final CheckedFuture<Void, TransactionCommitFailedException> checked = mock(CheckedFuture.class);
- Mockito.when(wTx.submit()).thenReturn(checked);
- Mockito.when(checked.checkedGet()).thenReturn(null);
+ final DOMDataTreeWriteTransaction wTx = mock(DOMDataTreeWriteTransaction.class);
+ when(domTx.newWriteOnlyTransaction()).thenReturn(wTx);
+ final DOMDataTreeReadWriteTransaction rwTx = mock(DOMDataTreeReadWriteTransaction.class);
+ when(rwTx.exists(any(), any())).thenReturn(immediateTrueFluentFuture());
+ doReturn(CommitInfo.emptyFluentFuture()).when(rwTx).commit();
+ when(domTx.newReadWriteTransaction()).thenReturn(rwTx);
+ doReturn(CommitInfo.emptyFluentFuture()).when(wTx).commit();