- DOMTransactionChain mockChain = mock(DOMTransactionChain.class);
- DOMDataWriteTransaction mockTx = mock(DOMDataWriteTransaction.class);
- doReturn(Futures.immediateCheckedFuture(null)).when(mockTx).submit();
- doReturn(mockTx).when(mockChain).newWriteOnlyTransaction();
-
- doReturn(mockChain).when(mockDataBroker).createTransactionChain(any());
- SchemaContextHandler schemaContextHandler = SchemaContextHandler.newInstance(
- new TransactionChainHandler(mockDataBroker), Mockito.mock(DOMSchemaService.class));
- schemaContextHandler.onGlobalContextUpdated(schemaContext);
+ DOMDataTreeWriteTransaction mockTx = mock(DOMDataTreeWriteTransaction.class);
+ doReturn(CommitInfo.emptyFluentFuture()).when(mockTx).commit();
+ doReturn(mockTx).when(mockDataBroker).newWriteOnlyTransaction();
+
+ SchemaContextHandler schemaContextHandler = new SchemaContextHandler(mockDataBroker,
+ mock(DOMSchemaService.class));
+ schemaContextHandler.onModelContextUpdated(schemaContext);