import com.google.common.util.concurrent.Futures;
import java.lang.reflect.Field;
import java.net.URI;
+import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
-import org.opendaylight.controller.md.sal.dom.api.DOMDataChangeListener;
import org.opendaylight.controller.md.sal.dom.api.DOMDataReadWriteTransaction;
+import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeChangeService;
import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
import org.opendaylight.controller.md.sal.dom.api.DOMTransactionChain;
import org.opendaylight.restconf.common.context.NormalizedNodeContext;
public void setUp() throws Exception {
MockitoAnnotations.initMocks(this);
- final TransactionChainHandler txHandler = Mockito.mock(TransactionChainHandler.class);
- final DOMTransactionChain domTx = Mockito.mock(DOMTransactionChain.class);
+ final TransactionChainHandler txHandler = mock(TransactionChainHandler.class);
+ final DOMTransactionChain domTx = mock(DOMTransactionChain.class);
Mockito.when(this.transactionHandler.get()).thenReturn(domTx);
Mockito.when(txHandler.get()).thenReturn(domTx);
final DOMDataWriteTransaction wTx = Mockito.mock(DOMDataWriteTransaction.class);
Futures.immediateCheckedFuture(null);
Mockito.when(rwTx.submit()).thenReturn(checkedFutureEmpty);
Mockito.when(domTx.newReadWriteTransaction()).thenReturn(rwTx);
- final CheckedFuture<Void, TransactionCommitFailedException> checked = Mockito.mock(CheckedFuture.class);
+ final CheckedFuture<Void, TransactionCommitFailedException> checked = mock(CheckedFuture.class);
Mockito.when(wTx.submit()).thenReturn(checked);
Mockito.when(checked.checkedGet()).thenReturn(null);
this.schemaHandler = new SchemaContextHandler(txHandler);
final DOMDataBroker dataBroker = mock(DOMDataBroker.class);
- final ListenerRegistration<DOMDataChangeListener> listener = mock(ListenerRegistration.class);
+
+ DOMDataTreeChangeService dataTreeChangeService = mock(DOMDataTreeChangeService.class);
+ doReturn(mock(ListenerRegistration.class)).when(dataTreeChangeService)
+ .registerDataTreeChangeListener(any(), any());
+
+ doReturn(Collections.singletonMap(DOMDataTreeChangeService.class, dataTreeChangeService))
+ .when(dataBroker).getSupportedExtensions();
+
doReturn(dataBroker).when(this.dataBrokerHandler).get();
- doReturn(listener).when(dataBroker).registerDataChangeListener(any(), any(), any(), any());
- final MultivaluedMap<String, String> map = Mockito.mock(MultivaluedMap.class);
+
+ final MultivaluedMap<String, String> map = mock(MultivaluedMap.class);
final Set<Entry<String, List<String>>> set = new HashSet<>();
Mockito.when(map.entrySet()).thenReturn(set);
Mockito.when(this.uriInfo.getQueryParameters()).thenReturn(map);