import javax.inject.Inject;
import javax.inject.Singleton;
import org.opendaylight.infrautils.utils.concurrent.LoggingFutures;
-import org.opendaylight.mdsal.binding.api.ClusteredDataTreeChangeListener;
import org.opendaylight.mdsal.binding.api.DataBroker;
import org.opendaylight.mdsal.binding.api.DataObjectModification.ModificationType;
+import org.opendaylight.mdsal.binding.api.DataTreeChangeListener;
import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
import org.opendaylight.mdsal.binding.api.DataTreeModification;
import org.opendaylight.mdsal.binding.api.RpcService;
@Singleton
@Component(service = { })
-public final class LLDPPacketPuntEnforcer implements AutoCloseable, ClusteredDataTreeChangeListener<FlowCapableNode> {
+public final class LLDPPacketPuntEnforcer implements AutoCloseable, DataTreeChangeListener<FlowCapableNode> {
private static final Logger LOG = LoggerFactory.getLogger(LLDPPacketPuntEnforcer.class);
private static final Uint8 TABLE_ID = Uint8.ZERO;
private static final String LLDP_PUNT_WHOLE_PACKET_FLOW = "LLDP_PUNT_WHOLE_PACKET_FLOW";
@Reference final RpcService rpcService) {
this.deviceOwnershipService = requireNonNull(deviceOwnershipService);
addFlow = rpcService.getRpc(AddFlow.class);
- listenerRegistration = dataBroker.registerDataTreeChangeListener(
- DataTreeIdentifier.create(LogicalDatastoreType.OPERATIONAL,
+ listenerRegistration = dataBroker.registerTreeChangeListener(
+ DataTreeIdentifier.of(LogicalDatastoreType.OPERATIONAL,
InstanceIdentifier.create(Nodes.class).child(Node.class).augmentation(FlowCapableNode.class)),
this);
}
import static org.junit.Assert.assertNotNull;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
@Mock
private DataTreeModification<FlowCapableNode> dataTreeModification;
@Mock
+ private DataObjectModification<FlowCapableNode> dataObjectModification;
+ @Mock
private DeviceOwnershipService deviceOwnershipService;
@Captor
private ArgumentCaptor<AddFlowInput> addFlowInputCaptor;
@Before
public void setUp() {
- doReturn(reg).when(dataBroker).registerDataTreeChangeListener(any(), any());
+ doReturn(reg).when(dataBroker).registerTreeChangeListener(any(), any());
doReturn(RpcResultBuilder.success().buildFuture()).when(addFlow).invoke(any());
doReturn(addFlow).when(rpcService).getRpc(any());
lldpPacketPuntEnforcer = new LLDPPacketPuntEnforcer(dataBroker, deviceOwnershipService, rpcService);
- final DataTreeIdentifier<FlowCapableNode> identifier = DataTreeIdentifier.create(
+ final DataTreeIdentifier<FlowCapableNode> identifier = DataTreeIdentifier.of(
LogicalDatastoreType.OPERATIONAL, NODE_IID.augmentation(FlowCapableNode.class));
when(dataTreeModification.getRootPath()).thenReturn(identifier);
- when(dataTreeModification.getRootNode()).thenReturn(mock(DataObjectModification.class));
+ when(dataTreeModification.getRootNode()).thenReturn(dataObjectModification);
when(dataTreeModification.getRootNode().modificationType()).thenReturn(ModificationType.WRITE);
when(deviceOwnershipService.isEntityOwned(any())).thenReturn(true);
}